Member of Technical Staff: Machine Learning Engineer

What You’ll Do

• Translate cutting-edge research into production-ready machine learning systems
• Design, build, and deploy end-to-end ML models and pipelines
• Develop and optimize models for image and video processing
• Own the full ML lifecycle: experimentation, training/fine-tuning, evaluation, and deployment
• Rapidly prototype using open-source models and adapt them for product needs
• Conduct experiments, analyze results, and iterate to improve performance
• Collaborate with researchers and cross-functional teams (product, engineering, design) to deliver ML solutions at scale
• Participate with advancements in machine learning and apply them to continuously improve productsWhat We’re Looking ForRequired Qualifications
• MS/PhD in Computer Science, Electrical Engineering, or related field
• Strong research experience with familiarity in top conferences (e.g., CVPR, ICCV, NeurIPS)
• 5+ years of experience in Python and proficiency in Java, C++, or Scala
• Strong understanding of diffusion models
• Strong understanding of multi-threading and memory management
• Solid knowledge of ML architectures: CNNs and Transformers
• Experience with PyTorch or TensorFlow
• Experience building end-to-end ML deployment and inference systems, especially for low-latency, real-time applications
• Experience deploying ML models in cloud environments (AWS preferred)
• Experience with experiment tracking systems and ML workflowsNice to Have
• Experience in low level optimisation, cuda etc.
• Experience productionizing and scaling ML models in real-world systems
• Contributions to open-source projects
• Experience with MLOps tools or distributed training systems
• Familiarity with relational databases (Postgres/MySQL)
• Experience handling large-scale data using tools like Spark