Machine Learning Environment Engineer- Remote- San Francisco Startup
Remote | Full-Time | Contract
Pay rate $40- 65 an hour
Join a fast-moving U.S. startup building critical infrastructure for the world’s leading AI companies. We’re developing the environments in which intelligent coding agents learn Machine Learning — from training neural networks to implementing new optimization algorithms. Our systems enable frontier AI models to reason, experiment, and advance autonomously.
About the Role
As an Environment Engineer (ML) , you’ll build simulation environments that teach coding agents to be proficient in Machine Learning. You’ll design robust testing infrastructure to evaluate AI-generated ML code on real-world engineering tasks — from training reinforcement learning agents to debugging complex model training loops. You’ll also automate key parts of the testing and benchmarking process to accelerate agent learning and performance evaluation.
The tools you build will directly shape how next-generation AI systems learn and execute ML research.
What You’ll Do
- Build testing environments that simulate ML workflows, including model training, data preprocessing, and evaluation.
- Design automated pipelines to benchmark model training performance and stability.
- Implement, test, and validate new algorithms from ML research papers to verify coding agents’ understanding and execution.
- Develop robust, reproducible experiments for reinforcement learning, deep learning, and optimization tasks.
- Collaborate with an elite engineering team that has built core AI infrastructure at leading labs and startups.
What We’re Looking For
3+ years of hands-on experience in Machine Learning engineering .Strong proficiency with PyTorch , JAX , or similar frameworks.Experience training large models on distributed systems or multi-GPU setups.Solid understanding of ML fundamentals (optimization, model architecture, evaluation metrics).Ability to work across complex codebases and quickly pick up new research and tooling.Bonus : experience with reinforcement learning or ML systems engineering.Why Join
Build core ML training infrastructure for top AI companies and research labs.Work remotely with world-class engineers and researchers.Own end-to-end projects that directly impact how AI models learn.Be part of a small, fast-paced, and well-funded U.S. team backed by leading investors and founders in AI.Interview Process
We move quickly and value your time :10 Minute Zoom with FounderTake Home AssignmentTwo short technical interviewsOffer within 7–10 days.#J-18808-Ljbffr