Machine Learning Engineer - ML Training Platform

We're looking for Senior/Staff engineers with 5+ years of experience in distributed systems and ML large-scale training to implement a novel substrate for training distributed ML models. Responsibilities include designing and implementing large-scale distributed training systems, building monitoring and metrics systems, and architecting resilient training systems.

Requirements

• Strong experience building and operating distributed systems in production
• Hands-on expertise with distributed training frameworks (FSDP, DeepSpeed, Megatron, or similar)
• Deep understanding of model parallelism (data, tensor, pipeline parallelism)
• Expert-level Python with production experience (concurrency, error handling, retry logic, clean architecture)
• Strong networking fundamentals: P2P systems, gRPC, routing, NAT traversal, distributed coordination
• Experience optimizing GPU workloads, memory management, and large-scale compute efficiency

Benefits

• Equity-heavy compensation with meaningful ownership in a mission-driven company
• Competitive base salary for senior engineering roles in Australia
• Visa sponsorship available for exceptional candidates
• Remote-first with optional access to our Melbourne hub