On-device ML Engineer - US Remote

Únase a Hugging Face en un viaje innovador para avanzar en el Aprendizaje Automático y hacerlo más accesible. Contribuimos al desarrollo de tecnología para el mejoramiento de la sociedad.

Hemos construido la biblioteca de código abierto de modelos pre-entrenados de más rápido crecimiento a nivel mundial. Con más de 1 millón de modelos y más de 320K estrellas en GitHub, nuestra tecnología es confiable para más de 15,000 empresas, incluidas organizaciones líderes en IA como Google, Elastic, Salesforce, Grammarly y NASA.

Como Ingeniero de ML en dispositivo, explorará métodos de vanguardia para ejecutar modelos en plataformas de consumo, centrándose en las tecnologías de Apple. Sus responsabilidades incluirán:

• Optimizar, cuantificar y convertir los mejores modelos para una ejecución eficiente en iPhones y Macs.
• Diseñar, construir y contribuir a software de código abierto que demuestre el uso de modelos y desarrolle bibliotecas para minimizar la fricción para los desarrolladores no familiarizados con ML.
• Difundir métodos, facilitar su adopción y crear herramientas para la comunidad.

Las tareas diarias pueden incluir:

• Evaluar modelos en función de calidad, latencia, memoria y necesidades de almacenamiento.
• Esforzarse por hacer que los modelos SOTA funcionen eficientemente en plataformas Apple convirtiéndolos a formatos nativos como Core ML o MLX.
• Optimizar arquitecturas de modelos para plataformas Apple Silicon, depurar problemas y desarrollar soluciones.
• Escribir código Swift para implementar u optimizar tareas de ML, incluidas las tuberías de pre y post procesamiento.
• Producir documentación técnica de alta calidad, como publicaciones de blog, tutoriales, guías, hilos en redes sociales y aplicaciones de demostración.
• Contribuir a proyectos de código abierto como coremltools para mejorar la cobertura de operaciones en PyTorch.
• Crear herramientas para que los desarrolladores conviertan, ejecuten y compartan modelos fácilmente.
• Escribir o comprender código de bajo nivel, como kernels de GPU paralelos cuando sea necesario.

Prosperará en esta posición si:

• Es un Desarrollador Experto en Swift: Tiene una sólida experiencia en desarrollo en Swift, una mentalidad práctica de constructor y un buen sentido de diseño de software y aplicaciones.
• Le Apasiona el ML: Tiene una comprensión profunda de las arquitecturas de modelos y una pasión por el aprendizaje automático.
• Es Proficiente en Core ML: Tiene experiencia en el uso de Core ML y comprende sus ventajas y limitaciones.
• Es un Contribuyente de Código Abierto: Está deseoso de publicar y contribuir a bibliotecas de código abierto para ayudar a los desarrolladores a adoptar ML.
• Es un Ingeniero Versátil: Puede moverse a través de diferentes niveles de abstracción, desde la UI hasta los kernels de Metal.
• Escribe Código Legible: Escribe código que sea fácil de entender pero también optimiza la ruta crítica para el rendimiento.
• Comprende Técnicas de Optimización: Tiene un conocimiento de técnicas de optimización, desde kv-caching en transformadores hasta cuantización post-entrenamiento.
• Comprende el Sistema: Puede identificar cuellos de botella en el rendimiento.
• Es Proficiente en Varios Marcos: Tiene experiencia con marcos como llama.cpp, MLX, PyTorch y CoreNet.
• Es un buen depurador.
• Puede escribir excelente documentación técnica.
• Participa en foros de discusión y comunidades sobre estos temas.

Incluso si no cumple con todos los requisitos, lo alentamos a postularse. Valoramos la diversidad y estamos formando un equipo cuyas habilidades, experiencias y antecedentes se complementen entre sí.

Valoramos la diversidad, la equidad y la inclusión. Estamos construyendo un lugar de trabajo donde todos se sientan respetados y apoyados, sin importar quién sea o de dónde venga. Hugging Face es un