Member of Technical Staff, Agentic Environments
Core
Design and write high-performing, scalable software for training frontier AI models and LLMs; develop tools to accelerate research and training pipelines.
Role type
Senior IC machine-learning engineer (LLM training infrastructure)
Builds
Scalable production-ready training systems and research tooling for frontier AI models
Domain
Enterprise AI / Large Language Models / Distributed Systems
Deliverable
production ML models
Required skills
Python, JAX, PyTorch, XLA/MLIR, distributed training strategies, Kubernetes, Ray, large-scale debugging, clean code practices, test-driven development
Preferred skills
ML/LLM academic research experience, tuning and optimizing large LLM models, experience with complex ML codebases
Technologies
JAX, PyTorch, XLA, MLIR, Kubernetes, Ray
Responsibilities
Design and write scalable software for model training; Develop new tools to support and accelerate research and LLM training; Research, implement, and experiment with ideas on cluster and data infrastructure; Collaborate with engineering and scientific teams on production systems
Seniority
Senior, hands-on IC
