Ai Research Scientist, Foundation Models, Autopilot Ai

You will be an integral part of a small, specialized team of deep learning experts, with access to one of the world’s largest model training clusters. Your primary focus will be training and scaling large vision Foundation Models, aiming to advance the state of the art in vision understanding and autonomous decision-making.

• A proven track record of successfully scaling and training large, complex models. You should have experience dealing with infrastructure challenges and possess techniques for optimizing the training of high-capacity models efficiently.
• Proficiency in Python and a strong foundation in software engineering best practices. Your expertise in maintaining clean and efficient code is crucial.
• An in-depth understanding of the fundamentals of deep learning, including but not limited to knowledge of layer details, loss functions, and optimization techniques.
• Expertise in various neural network architectures for computer vision, speech, NLP, and related domains.
• Prior experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks.
• Familiarity with standard scientific computing primitives, including but not limited to batched operations, grid sampling, scatter/gathering, and einsum.
• (bonus) Strong expertise in system-level knowledge, enabling you to profile and optimize training and evaluation processes efficiently.