Remote Member of Technical Staff

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 our supercomputer and data infrastructure. We have offices in Toronto, London, San Francisco, and New York but also embrace being remote-friendly!

Requirements

• Extremely strong software engineering skills
• Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR
• Deep experience in AI data evaluation and management
• Strong track record of creating and curating large-scale datasets
• Experience using large-scale distributed training strategies
• Familiarity with autoregressive sequence models, such as Transformers
• Ability to collaborate effectively with human annotators and cross-functional teams
• Paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

Responsibilities

• Design, build and scale AI systems for serving our users
• Research, implement, and experiment with ideas on our supercomputer and data infrastructure
• Learn from and work with the best researchers, engineers and annotators in the field
• Drive agent development in RAG, tool use, and language agents
• Work with internal annotators in creating and evaluating data for post-training LLMs, ensuring quality and performance

Benefits

• An open and inclusive culture and work environment
• Work closely with a team on the cutting edge of AI research
• Free daily lunch
• Full health and dental benefits, including a separate budget to take care of your mental health
• Personal enrichment benefits towards arts and culture, fitness and well-being
• Remote-flexible, offices in Toronto, New York, San Francisco and London and coworking stipends
• Paid vacation