Remote Member of Technical Staff, Modeling

Summary

Join us on our mission to scale intelligence and shape the future! As a Member of Technical Staff, you will design, build, and scale AI systems for serving users, research, implement, and experiment with ideas on supercompute and data infrastructure.

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

Responsibilities

• Design, build and scale AI systems for serving our users
• Research, implement, and experiment with ideas on our supercompute and data infrastructure

Benefits

• Weekly lunch stipend
• Full health and dental benefits
• 100% Parental Leave top-up for 6 months
• Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
• Remote-flexible work arrangement with offices in Toronto, New York, San Francisco, and London
• 6 weeks of vacation