2 days ago
Mountain View, CA, USAMid Level / Senior
Base Salary
$220k - $300k/yr
Responsibilities
- Design, train, and evaluate machine learning models, including LLMs, diffusion models, and domain-specific architectures.
- Develop scalable experimentation pipelines covering data processing, model training, and evaluation workflows.
- Collaborate with data and infrastructure teams to optimize training throughput and data quality.
- Contribute to internal benchmarks, open research, and emerging multimodal and generative AI techniques.
- Prototype research insights and productionize them into usable tools and models.
- Set standards for research rigor, documentation, and reproducibility across the team.
Requirements
- 3–10 years of experience in ML research, applied ML, or ML systems engineering.
- Deep familiarity with PyTorch, JAX, or TensorFlow and architectures such as Transformers, diffusion models, or RLHF.
- Strong foundations in data processing, distributed training, and evaluation metrics.
- Ability to turn research papers into working prototypes and production-ready code.
- Interest in multimodality, self-learning, synthetic data, and agentic systems.
- Comfort building from zero to one in a fast-moving, unstructured environment.
Benefits
- Annual salary of $220,000–$300,000 USD.
- On-site role in Mountain View, California.
- Visa sponsorship is not available for this role.
Tech Stack
PyTorchTensorFlow
Categories
AI ResearchML Engineering