Senior/ Staff Research Engineer

Candidates must possess a Bachelor's degree in Computer Science, Engineering, or a related field, coupled with over 7 years of industry experience in AI, machine learning, or similar roles, demonstrating a history of successfully implementing and deploying AI models in production. Proficiency in Python and C++, deep expertise in frameworks like PyTorch/Jax, a strong grasp of deep learning fundamentals including neural network architectures, optimization techniques, and evaluation metrics, and experience with distributed training and inference of large-scale AI models on thousands of GPUs using DDP, FSDP, etc., are essential. Experience with performance optimization of large-scale AI models, effective communication and collaboration skills for fast-paced, multidisciplinary team environments, and experience in solving complex problems by comparing alternative solutions, tradeoffs, and diverse viewpoints are also required. Preferred qualifications include a publication track record at top AI conferences, experience working cross-functionally, familiarity with different GPU architectures, cloud environments, low-level hardware optimization, and custom CUDA kernels, as well as contributions to open-source projects.