Osmosis (YC W25) · 2 days ago
Machine Learning Engineer
United States
Full-time
Remote
Mid Level
$180K/yr - $250K/yrOsmosis is a company focused on enhancing open-source language models using advanced reinforcement learning techniques. They are seeking a Machine Learning Engineer to develop high-performance distributed training infrastructure and engage with customers on production deployments.
Computer Software
Responsibilities
You’ll contribute to work in areas like:
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
Qualification
Required
Expertise in RL algorithms
Experience with distributed training
Low-level optimization skills
Proficiency in Python FastAPI and Golang
Experience with React, TypeScript, and Next.js
Familiarity with AWS Fargate, Docker, Kubernetes, and AWS SageMaker
Experience with ML frameworks such as Verl, slime, Megatron-LM, SkyRL, and PyTorch
Knowledge of databases like DynamoDB and S3
Company
Osmosis (YC W25)
Osmosis is a reinforcement fine-tuning platform that helps companies create task-specific models that outperform foundation models at a fraction of the cost.
2-10 employees
Funding
Current Stage
Early StageCompany data provided by crunchbase