AI/ML Engineer – Computer Vision & Robotics

Somos la compañía de recolección de datos para el entrenamiento de robots humanoides más grande de Latinoamérica. Los datos que capturamos ayudan a entrenar y mejorar los robots del futuro.

Rol.
Serás responsable de construir y evolucionar nuestro pipeline de IA para transformar datos reales de robótica en información estructurada y utilizable. Trabajarás con Computer Vision, Deep Learning y modelos multimodales, llevando soluciones desde la integración de modelos existentes hasta su despliegue, automatización y mejora continua. Tendrás ownership técnico sobre el pipeline y trabajarás de cerca con Arquitectura, Backend y Operaciones.

¿Qué harás en este rol?

• Investigar, adaptar e implementar soluciones open-source de Visual-Inertial Odometry (VIO) y estimación de pose 3D.
• Construir y automatizar pipelines para procesar datos de robótica de forma eficiente y reproducible.
• Dockerizar soluciones de Computer Vision y resolver dependencias complejas de despliegue.
• Diseñar los outputs y estructuras de datos necesarios para integrarlos con nuestros sistemas Backend.
• Desplegar modelos multimodales Vision-Language-Action (VLA) para generar pre-anotaciones como bounding boxes y segmentaciones.
• Construir y mantener APIs de inferencia para conectar nuestros modelos con herramientas internas.
• Optimizar el proceso de inferencia de modelos de Deep Learning.
• Trabajar con Operaciones y QA para utilizar Ground Truth y feedback real en la mejora de nuestros modelos.
• Ejecutar ciclos de fine-tuning y evaluación para incrementar continuamente el desempeño de la IA.
• Investigar nuevas herramientas y soluciones que permitan procesar nuestros datos de forma más eficiente.

Requisitos indispensables.

• 3+ años de experiencia profesional en Machine Learning, AI, Computer Vision o roles relacionados.
• Dominio avanzado de Python y experiencia escribiendo código estructurado para producción.
• Experiencia sólida con PyTorch.
• Experiencia práctica con Computer Vision y procesamiento de imágenes, video y tensores.
• Experiencia llevando modelos de Machine Learning a producción y sirviendo modelos para inferencia.
• Experiencia sólida con Docker y contenedores.
• Experiencia construyendo APIs o microservicios, idealmente con FastAPI.
• Capacidad para investigar documentación técnica, resolver problemas de dependencias y trabajar de manera autónoma.
• Mentalidad orientada a construir, probar, medir e iterar soluciones.

Deseables.

• Experiencia con Hugging Face, Transformers o Datasets.
• Experiencia con TensorRT, vLLM, TGI, Ollama u otras herramientas de serving y optimización de modelos.
• Experiencia con Visual-Inertial Odometry (VIO) o estimación de pose 3D.
• Experiencia con modelos Vision-Language o Vision-Language-Action.
• Experiencia con ROS / ROS 2.
• Conocimiento de formatos de datos de robótica como .mcap, .bag o .hdf5.
• Conocimiento de cinemática y sincronización de sensores como cámaras e IMU.
• Experiencia con S3 / MinIO u otras soluciones de Object Storage.
• Experiencia con CUDA y despliegues de modelos con alto consumo computacional.

Qué ofrecemos.

• $40,000 – $45,000 MXN mensuales brutos.
• Lunes a viernes de 8:00 a 17:00.
• Prestaciones de ley.
• Participación en proyectos de IA, Computer Vision y robótica.
• Ownership técnico sobre un pipeline clave para el procesamiento de datos.
• Trabajo cercano con equipos de Arquitectura, Backend y Operaciones.
• La oportunidad de llevar soluciones de Machine Learning desde la experimentación hasta producción.

Sueldo: $40,000.00 - $45,000.00 al mes

Lugar de trabajo: Empleo presencial