Software Engineer, ML Platform

📋 Description

• Develop and improve the model training system across clouds.
• Deploy storage subsystems for dataset management and throughput.
• Build evaluation infrastructure for easy execution and tracking.
• Build tooling to integrate training with telemetry, tracking, and checkpointing.
• Biology experience not required; lab work optional perk.

🎯 Requirements

• Degree in Computer Science, Machine Learning, Computational Biology, or related field.
• 5+ years building and deploying ML systems in production.
• Experience leading technical projects and cross-functional execution.
• Strong Python programming skills.
• Experience with Terraform and Ansible.
• Experience with Torch and Jax deep learning frameworks.
• Solid understanding of machine learning fundamentals.
• Experience with infrastructure for large-scale model training.

🎁 Benefits

• Compensation: base pay range $140,000-$215,000 annually.