Key skills
PyTorchMachine LearningMLJAXGlueGitHub
About this role
Role Overview
- Build and maintain robust multi-stage asynchronous workflows for running data generation, training, and evaluations for our machine learning stack.
- Rationalize machine learning systems design and software architecture.
- Identify blockers and build solutions that scale to the size of foundation models.
- Operate as the glue between research scientists and the infrastructure team.
Requirements
- At least two years relevant industry experience.
- Highly fluent in and enthusiastic about PyTorch and JAX.
- Used to thinking in asynchronous primitives.
- Strong views on library design: clean abstractions, minimal surface area, consistency.
- Solid track record of observable artifacts (e.g., GitHub) showing clear, well-documented code.
- ML generalist who knows what scalable, reliable ML systems look like.
Tech Stack
- PyTorch
Benefits
- Offers Equity
- Offers Bonus