Specialty
Who is hiring is shown to signed-in users. Create a free account to see the employer and apply.
The role
The company, an enterprise AI company, is hiring a Training Infrastructure Engineer to support its model training pipelines and bring models from research to production. There are no location restrictions, though the posting suggests checking its core time zone guidance.
What you will do
- Design and write high-performance, scalable software for model training.
- Improve the training setup in terms of infrastructure and codebase performance.
- Build tools that speed up training cycles and improve training infrastructure efficiency.
- Research and experiment with ideas on the supercompute and data infrastructure.
- Work alongside researchers and contribute production code.
What they ask for
- Very strong software engineering skills.
- Proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR.
- Experience with distributed training infrastructure (Kubernetes, Slurm) and frameworks like Ray.
- Experience with large-scale distributed training strategies.
- Hands-on experience training large models at scale and contributing to training tooling or setup.
What they offer
- Remote-friendly with co-working benefit and $500 home office stipend
- Health and dental benefits, plus a mental health budget
- Retirement matching (RRSP, 401K or pension)
- Parental leave top-up to 100% for up to 6 months
- 6 weeks of paid vacation
- Education and learning stipend
This is our own short reading of the posting. The full description, and anything that has changed since, is on the employer's careers page behind the apply button.