Senior Software Engineer — MLOps Platform

About the Role

Weights & Biases is looking for a Senior Software Engineer to build the next generation of our ML experiment tracking and model management platform.

What You'll Work On

• Core experiment tracking infrastructure (runs, metrics, artifacts)
• Distributed training integrations (PyTorch, JAX, TensorFlow)
• Model registry and lineage tracking
• Real-time collaboration features for ML teams

• 5+ years software engineering experience
• Strong Python and distributed systems background
• Experience building developer tools or ML platforms
• Familiarity with ML training workflows

• Fully remote with annual team offsite
• Competitive salary + equity
• Top-tier health benefits
• $2,000 home office setup budget