Make a game-changing next move.

Learn more about the opportunities in Coatue's portfolio.

Post-Sales Machine Learning Engineer, Customer Success - US (Remote)

Weights & Biases

Weights & Biases

Software Engineering, Customer Service, Sales & Business Development
San Francisco, CA, USA
Posted on Wednesday, March 13, 2024
At Weights & Biases, our mission is to build the best tools for AI developers. We founded our company on the insight that while there were excellent tools for developers to build better code, there were no similarly great tools to help ML practitioners build better models. Starting with our first experiment tracking product, we have since expanded our solution into a comprehensive AI developer platform for organizations focused on building their own deep learning models and generative AI applications.
Weights & Biases is a Series C company with $250M in funding and over 200 employees. We proudly serve over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.
We're hiring a Machine Learning Engineer, Customer Success to help our customers solve difficult, real-world problems and engage in ground-breaking research by using our developer tools in their machine learning pipelines.
In this role, you'll be collaborating closely with the most advanced ML teams worldwide, tackling some of the most challenging ML issues in areas like computer vision, robotics, natural language processing, and beyond. This specific role will be centered on W&B’s engagement with new customers who have recently adopted W&B for their projects. You'll have the chance to work with ML teams across various industries to understand their ML requirements, enhance their ML workflow, integrate W&B into their environment effectively, participate in projects, and educate them on the best practices of our product. In particular, the person in this role will be tasked with guiding new customers through their onboarding process, understanding their technical needs, showcasing how W&B enhances their capabilities, designing and implementing customer adoption strategies, and ensuring a seamless transition and successful use of our product post-sales.
Machine Learning Engineers on our customer success teams play a vital role in the success of our customers at Weights & Biases. You'll collaborate with Sales, Support, Product, and Engineering teams to lead the technical success of our customers after the sales process, acting as the primary source of knowledge and representative to our customers. You will play a key role in customer adoption, partnering with our product team to steer product development based on customer feedback and usage patterns.
This is an ideal opportunity for anyone with machine learning experience who is customer-focused and eager to work with the top ML companies globally, focusing on ensuring customer success and adoption post-sales.


  • Master the implementation of efficient, robust, and reproducible machine learning pipelines for engineering teams, leveraging Weights & Biases tools, to ensure smooth customer onboarding and adoption processes.
  • Serve as a trusted advisor to new customers by identifying their desired outcomes and clearly communicating W&B’s product best practices, ensuring their success through the effective integration and optimization in the post-sales adoption phase.
  • Conduct product training sessions and workshops that demonstrate best practices and the diverse solutions W&B offers, aimed at facilitating customer success and adoption post-sales.
  • Serve as a pivotal link between customers and internal teams by continuously providing customer feedback and advocating for customer needs, thereby directly influencing the product roadmap and enhancements to reflect real user experiences, trends, challenges, and successes.


  • 3-5 years of relevant experience in a similar role
  • Experience using one or more of the following packages: TensorFlow/Keras, PyTorch Lightning
  • Strong programming proficiency in Python and eagerness to help customers who are primarily users of Python deep learning frameworks and tools be successful
  • Excellent communication and presentation skills, both written and verbal
  • Ability to effectively manage multiple conflicting priorities, respond promptly and manage time effectively in a fast-paced, dynamic team environment
  • Ability to break down complex problems and resolve them through customer consultation and execution.
  • Experience with cloud platforms (AWS, GCP, Azure)
  • Experience with Linux/Unix

Strong Plus:

  • Proficiency with one or more of the following packages: HuggingFace, Fastai, scikit-learn, XGBoost, LightGBM, Ray, any of the GenAI frameworks
  • Experience with data engineering, MLOps and tools such as Docker and Kubernetes
  • Experience with data pipeline tools
  • Experience as an ML educator and/or building and executing customer training sessions, product demos and/or workshops at a SaaS company

Our Benefits:

  • 🏝️ Flexible time off
  • 🩺 Medical, Dental, and Vision for employees and Family Coverage
  • 💵 Home office budget with a new high-powered laptop
  • 🥇 Truly competitive salary and equity
  • 🚼 12 weeks of Parental leave (U.S. specific)
  • 📈 401(k) (U.S. specific)
  • Supplemental benefits may be available depending on your location
  • Explore benefits by country
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