Builds and maintains sandboxed, reproducible RL environments for AI agent training, including terminal, browser, and tool-augmented setups. Requires 2+ years Python/systems engineering, containerization, and RL concepts understanding.
140k – 200k/yr
Hybrid2+ YOEML Engineering
About the role
What You’ll Do
Design, build, and maintain sandboxed RL environments for agentic AI training—including terminal emulators, browser automation harnesses, computer-use simulators, and tool-augmented workspaces (e.g., environments built on frameworks like TerminalBench, OSWorld, and Tau-bench)
Develop reproducible, containerized execution environments (Docker, VMs, lightweight sandboxes) that support deterministic task rollouts and reward signal collection
Integrate with and extend open-source agentic tooling and custom CLI/API harnesses to enable multi-step agent interaction
Build instrumentation and observability layers—structured logging, trajectory capture, state snapshotting—so training runs and human annotation sessions produce clean, auditable data
Collaborate with data operations to design task curricula and evaluation protocols that stress-test model capabilities across environment types
Own environment deployment and reliability: CI/CD pipelines, automated testing of environment configurations, and monitoring for drift or breakage across versions
Rapidly prototype new environment types as client and internal requirements evolve, moving from spec to working system in days, not weeks
What We’re Looking For
Required
2+ years of professional software engineering experience, with strong fundamentals in Python and at least one systems-level language (Go, Rust, C++)
Demonstrated experience with containerization and sandboxing (Docker, Podman, Firecracker, or similar) in production or near-production contexts
Familiarity with RL concepts: MDPs, reward shaping, episode structure, observation/action spaces. You don’t need to have trained models, but you need to understand what an environment must provide to an RL training loop
Experience building or maintaining developer tooling, CLI tools, or infrastructure automation
Comfort working with browser automation frameworks or terminal interaction tooling
Strong debugging instincts—you can trace failures across process boundaries, container layers, and network calls
Ability to read and implement from academic papers and open-source benchmark repositories without extensive hand-holding
Preferred
Direct experience building or contributing to RL environments (Gymnasium/Gym, PettingZoo, or custom environment implementations)
Experience with agentic AI evaluation frameworks (SWE-bench, WebArena, OSWorld, TerminalBench, or similar)
Familiarity with GCP or AWS infrastructure (Compute Engine, ECS/EKS, Cloud Build)
Prior work at an AI data company, ML platform company, or AI research lab
Contributions to open-source projects in the RL, agents, or dev-tools space
Build and own production AI agent systems (harnesses, evals, orchestration) on frontier LLMs for industrial supply chain workflows at Traba. Requires 1+ years shipping LLM/agent features to production, strong Python/TS skills, and high-agency in ambiguous customer environments.
140k – 200k/yr
On-site1+ YOEML Engineering
Machine Learning Engineer, Search Quality
GleanUnited States
Build advanced search quality systems using machine learning, including personalization signals, ranking models, and domain-adapted LLMs for enterprise search. Requires 2+ years experience in ML, search/NLP, strong coding in Python/Go/Java/C++, and bachelor's in CS/math.
140k – 265k/yr
Hybrid2+ YOEML Engineering
Software Engineer
Neon HealthUnited States
Builds and debugs AI agent infrastructure for healthcare automation, including prompt engineering, runtime issue tracing, evaluation datasets, simulation tooling, data pipelines, and observability dashboards. Requires 2-7 years experience with production LLMs/AI agents and TypeScript proficiency.
140k – 200k/yr
Remote2+ YOEML Engineering
Machine Learning Engineer II, Computer Vision Applied Science
PinterestSan Francisco, CA
Build and fine-tune vision-centric VLMs and generative models using Pinterest's visual-text datasets. Requires 2+ years industry computer vision experience and an M.S. or Ph.D.
139k – 286k/yr
Remote2+ YOEML Engineering
Machine Learning Engineer, Core Engineering
PinterestSan Francisco, CA +2
Build and improve machine learning models for Pinterest's recommendation systems across Homefeed, Ads, Search, and more. Requires 2+ years experience in ML methods like personalization and recommender systems, plus hands-on work with large-scale data pipelines.