Software Engineer - ML

Responsibilities:

• Play a critical role in developing and deploying self-evolving AI systems that transform organizational workflows

• Adapt standard machine learning methods to best exploit modern parallel environments

• Design and implement scalable ML pipelines for training, evaluation, and deployment

• Lead technical initiatives in ML system design and excellence

• Develop highly scalable classifiers and tools leveraging machine learning

• Suggest, collect and synthesize requirements and create effective feature roadmaps

• Code deliverables in tandem with the engineering team

• Ensure robust monitoring and maintenance of production ML systems

Qualifications:

• MS/PhD in Computer Science, Machine Learning, or related technical field

• 3+ years of experience in machine learning or related technical field

• Strong programming skills in Python and PyTorch/TensorFlow

• Experience with large language models and transformer architectures

• Proficiency in ML ops and deployment workflows

• Experience with distributed computing and GPU optimization

• Experience building and shipping high-quality work with proven reliability

• Knowledge of ML systems design and architecture

• Experience with cloud platforms (AWS, GCP, Azure)

• Experience with reinforcement learning and self-supervised learning is a plus

• Open-source contributions are valued

• Experience leading technical initiatives is appreciated

Benefits:

• Competitive salary and equity package

• Flexible paid time off

• Flexible working hours

• Comprehensive health insurance

• Education stipend

• Home office setup allowance

• Wellness benefits

• Global in-person team events and retreats

• Learning and development opportunities

ExtensityAI is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, or disability.