Software Engineer, Machine Learning Workflows

Software Engineer, Machine Learning Workflows

This job is no longer open
At Weights & Biases, our mission is to build the best developer tools for machine learning. Weights & Biases is a series C company with $200 million in funding and a rapidly growing user base. Our platform is an essential piece of the daily work for machine learning engineers, from academic research institutions like FAIR and UC Berkeley to massive enterprise teams including iRobot, OpenAI, Toyota Research Institute, Samsung, NVIDIA, Salesforce, Blue Cross Blue Shield, Lyft, and more.

In this role, you will own building new products that connect various stages of the machine learning lifecycle. You will partner closely with both internal teams and enterprise customers to improve the velocity of ML workloads. This opportunity means directly working with state-of-the-art ML organizations; shipping features that will have a direct impact on users’ day-to-day lives. You will also tackle key problems in ML systems today such as dataset management, model understanding, and production orchestration. If you are interested in shaping the future of ML tooling, then this is the right role for you!

What you'll achieve

    • Design and build new products that reshape how ML engineers interact with their datasets and models
    • Scale APIs and systems to support enterprise-level ML workloads and datasets
    • Implement automation to assist ML engineers in finding optimal hyperparameters or run an inference job when training is finished
    • Focus on our end users and listen to customer feedback in order to deliver value
    • Keep up with the latest trends in the ML world and leverage existing tools and frameworks whenever necessary

What's needed in this role

    • Ability to synthesize complex requirements into concrete features and milestones
    • Strong software engineering fundamentals and knowledge of at least 1 modern programming language (Java, Go, Python, C++, etc)
    • Experience working with ML or data pipelines (Kubeflow, Spark, Sagemaker, etc)
    • Basic understanding of ML algorithms or familiarity with some ML framework (PyTorch, Tensorflow, XGBoost, etc)
    • Experience with cloud platforms and technologies (AWS, GCP, Kubernetes, etc)

Why join us?

    • Top-tier machine learning teams rely on our tools for their daily work at companies including OpenAI, Toyota Research Institute, Lyft, Samsung, and Pandora.
    • You'll never stop learning. This role gives you first-hand experience talking with leading researchers in the field, understanding their problems, and directly shaping the product direction.
    • Our experienced founding team has successfully built and sold ML tools in the past at Figure Eight, and their deep knowledge of our industry, empathy for our users, and skillful management is driving W&B to success.
    • Customers genuinely benefit from our tool. Here's a quote from Wojciech Zaremba, Cofounder and Robotics Lead, OpenAI: "W&B allows to scale up insights from a single researcher to the entire team, and from a single machine to hundreds of them."


    • 🏝️ Unlimited vacation time; we ensure everyone takes at least 2 weeks in addition to 2 more weeks of holidays throughout the year
    • 🩺 100% Medical, Dental, and Vision for employees and Family Coverage
    • 🏠 Remote first culture with in-office flexibility in San Francisco
    • 💵 $1000 home office budget with new high-powered laptop
    • 🥇 Truly competitive salary and equity
    • 🚼 12 weeks of Parental leave
    • 📈 401(k)
We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out at
This job is no longer open
Logos/outerjoin logo full

Outer Join is the premier job board for remote jobs in data science, analytics, and engineering.