Machine Learning Tools Engineer, SIML

What you'll do

• Design and maintain core machine learning infrastructure components like training pipelines and deployment tooling.
• Develop Python-based SDKs and CLIs to simplify ML engineers' interactions with compute resources and models.
• Identify and address pain points for ML practitioners, translating them into productized solutions that enhance productivity.
• Ensure tools are robust, performant, and user-friendly, with strong observability features and comprehensive documentation.
• Collaborate with infrastructure teams to integrate end-to-end systems and support smooth scaling of ML development environments.

What we're looking for

• 3+ years of software development experience with strong Python proficiency
• Familiarity with machine learning fundamentals and frameworks like PyTorch, TensorFlow, JAX
• Experience working with Linux systems, containers (Docker), and version control (Git)
• Strong debugging, analytical, and problem-solving skills
• Comfortable collaborating across teams in research and product development environments
• Prior experience in ML platform, infrastructure, or productivity tools team preferred
• Exposure to distributed training, experiment tracking, or model serving infrastructure