Tech Stack

Description

You will work closely with ML research scientists to solve challenging training and inference problems using very large models, accelerating research and implementing novel models and simulated environments. Your role involves designing workflows, developing tools for computing resource optimization, and establishing best practices for ML model development.

Requirements

• Degree in computer science or related field
• Advanced degree (MSc+) in ML or equivalent experience
• 3+ years industry experience with PyTorch, TensorFlow, or JAX
• Experience with cloud platforms (AWS, GCP, Azure)
• Familiarity with Docker, Kubernetes, and data infrastructures

Responsibilities

• Collaborate with researchers to accelerate model training and inference
• Design and implement workflows for research experiments
• Develop tools to optimize computing resource use
• Establish and document best practices for ML workflows
• Implement novel models and simulated environments