Member of Technical Staff, Modeling
Core
Design, build, and scale AI systems for serving users; research, implement, and experiment with ideas on supercompute and data infrastructure.
Role type
Member of Technical Staff (Modeling)
Builds
AI systems for serving users, state-of-the-art models to production
Domain
Artificial Intelligence / Large Language Models / Distributed Systems
Deliverable
production ML models
Required skills
software engineering, Python, ML frameworks (TensorFlow, JAX, XLA/MLIR), GPU kernel development (CUDA), large-scale distributed training, autoregressive sequence models (Transformers)
Preferred skills
publications at top-tier venues (NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)
Technologies
TensorFlow, TF-Serving, JAX, XLA/MLIR, CUDA
Responsibilities
Design and implement novel research ideas; ship state-of-the-art models to production; maintain deep connections to academia; research, implement, and experiment with ideas on supercompute and data infrastructure
Seniority
Individual Contributor
