We're looking for a Member of Technical Staff to join our team and contribute to the development of AI systems. As a key member of our team, you'll design, build, and scale AI systems, research and implement new ideas, and learn from and work with top researchers in the field.
Requirements
- Extremely strong software engineering skills
- Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR
- Experience writing kernels for GPUs using CUDA
- Experience using large-scale distributed training strategies
- Familiarity with autoregressive sequence models, such as Transformers
Benefits
- Open and inclusive culture and work environment
- Work closely with a team on the cutting edge of AI research
- Weekly lunch stipend, in-office lunches & snacks
- Full health and dental benefits, including a separate budget to take care of your mental health
- 100% Parental Leave top-up for 6 months for employees based in Canada, the US, and the UK
- Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
- Remote-flexible, offices in Toronto, New York, San Francisco and London and co-working stipend
- 6 weeks of vacation
Originally posted on Himalayas
Apply Now