Machine Learning Engineer
Hybrid (San Francisco, CA)$180k - $250kFull-timePosted 9mo ago
About the role
Machine Learning Engineer role at Osmosis focused on reinforcement learning (RL) for AI agents, contributing to high-performance distributed training infrastructure for RL at scale and collaborating with the founding team.
Responsibilities
- Distributed Training Infrastructure: implement new RL algorithms and build scalable post-training pipelines
- Resource Management & Optimization: design infrastructure systems for efficient GPU utilization and dynamic resource allocation
- Customer-Facing Work: work directly with customers on production deployments and custom model development
Qualifications
- 1+ years experience in machine learning or related field
- Experience or strong interest in reinforcement learning (RL)
- Willingness to work in San Francisco, CA
- US citizen/visa required
Skills
Required:Reinforcement learning (RL)Python
Nice to have:Distributed trainingGPU/resource managementGolangPyTorchFastAPI
Last updated Aug 26, 2026. Interested? Apply on the company site.
About Osmosis
Online Reinforcement Learning for AI Agents
Osmosis enables AI self-improvement through real-time reinforcement learning.
aib2bbessemer-ai-infrastructure-2026developer-toolsinfrastructurereinforcement-learningreinforcement-learning-platforms