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