Software Engineer for AI Models

Job Summary

A company is looking for a Member of Engineering focused on pre-training and inference software.

Key Responsibilities

• Propose and evaluate innovations in the training development experience and reliability
• Enhance and maintain training and inference codebases
• Write high-quality code in Python (PyTorch), Cython, and C/C++, and improve CI/CD processes

Required Qualifications

• Basic understanding of Large Language Models (LLM) and Transformers
• Strong engineering background with programming experience in Linux
• Proficiency in Python with libraries such as numpy, PyTorch, or Jax
• Experience with C/C++ and CI/CD practices
• Strong algorithmic skills and a commitment to code quality