Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance

Job expired!

Únete a Nuestro Equipo de Compiladores PyTorch en Meta

Visión General:

El equipo de Compiladores PyTorch en Meta tiene la misión de mejorar el rendimiento de PyTorch, enfocándose en la eficiencia y la velocidad sin comprometer su reconocida flexibilidad. Nuestra revolucionaria tecnología PT2 ha establecido un nuevo punto de referencia en el panorama de marcos de ML desde su creación en marzo de 2023, impulsando a PyTorch hacia el liderazgo en compiladores de ML en la industria. Estamos buscando un Ingeniero de Software para contribuir al compilador PT2 y ampliar los límites de la tecnología de aprendizaje automático.

Responsabilidades Clave:

  • Desarrollar y mejorar los componentes del compilador PT2 como TorchDynamo, TorchInductor y PyTorch Core.
  • Impulsar mejoras en el rendimiento de PyTorch, proporcionando soluciones sistemáticas que beneficien a la comunidad global de PyTorch.
  • Investigar y optimizar modelos de IA generativa, enfocándose en las etapas de pre-entrenamiento, ajuste fino e inferencia.
  • Explorar sinergias entre el compilador PyTorch y los sistemas distribuidos de PyTorch.
  • Colaborar activamente con los usuarios de PyTorch para fomentar la adopción y usos innovadores de las tecnologías PT2.

Calificaciones Mínimas:

  • Licenciatura/Maestría en Ciencias de la Computación o un campo estrechamente relacionado.
  • Experiencia práctica en compiladores, sistemas de ML, computación de alto rendimiento y optimización de GPU.
  • Proficiencia en programación con Python o CUDA.
  • En proceso de obtener o haber obtenido un título de Licenciatura en Ciencias de la Computación o un campo técnico relacionado.

Calificaciones Preferidas:

  • Doctorado en Ciencias de la Computación, Matemáticas Aplicadas o campos relacionados.
  • Experiencia con tecnologías PT2, Triton, MLIR o contribuciones directas a PyTorch.
  • Amplia experiencia en rendimiento de GPU y desarrollo de kernels CUDA de alto rendimiento.
  • Historial comprobado de excelencia en investigación e ingeniería de software, evidenciado por publicaciones, patentes o participación en competencias relevantes.

Acerca de Meta:

Meta continúa redefiniendo el paisaje de la conectividad y la interacción social mediante tecnologías innovadoras. Desde el lanzamiento de Facebook en 2004 y las aplicaciones subsiguientes como Instagram y WhatsApp, ahora estamos incursionando en la realidad aumentada y virtual para ingeniar la próxima era de la tecnología social. En Meta, tu trabajo trasciende los límites digitales tradicionales, creando nuevas formas para que las personas se conecten, las comunidades prosperen y los negocios crezcan.

Compensación y Beneficios:

El rol ofrece un salario competitivo que va desde $117,000 hasta $173,000 por año, además de bonificaciones, equidad y beneficios integrales. Estamos comprometidos a ofrecer soporte y alojamientos inclusivos en nuestro proceso de contratación para candidatos diversos.

Si estás listo para dar forma al futuro de la tecnología con nosotros en Meta, avanzando uno de los campos más emocionantes del mundo, ¡aplica para ser parte de nuestro equipo de Compiladores PyTorch hoy!