Machine Learning Engineer, RL Environments
Preference Model4 days ago
Toronto, Canada or San Francisco, CA, USAMid Level / Senior
Responsibilities
- Design and build RL environments and reward functions for ML tasks.
- Develop expertise in ML research, training, and inference infrastructure.
- Collaborate with team members to brainstorm and create new tools for environment building.
Requirements
- Strong ML fundamentals and broad research interests.
- Proficiency in Python and systems programming, and at least one of PyTorch or JAX.
- Problem-solving skills with a focus on ownership and end-to-end solutions.
- Passion for staying current with ML infrastructure advancements.
- Ability to meet throughput expectations and respond quickly to feedback.
Benefits
- Competitive cash and equity compensation (>90th percentile).
- Ownership and autonomy in a fast-moving startup environment.
- Opportunity to work with top machine learning engineers.
- Health, vision, dental benefits.
- 401K match.
- Visa sponsorship and relocation support available.