Back to Jobs

Senior Customer Success Machine Learning Engineer

Remote, USA Full-time Posted 2026-06-17

Description: • 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 Senior Customer Sales Machine Learning Engineer to lead the technical engagement and success of our federal and public sector clients. Reporting to the Head of Field Engineering and partnering closely with our Head of Federal, this role will be highly visible and critical to the growth of Weights & Biases’ burgeoning public sector practice. In this role, you will lead pre- and post-sales motions, working closely with federal machine learning (ML) teams to tackle critical, real-world challenges across computer vision, NLP, robotics, and help government clients understand how Weights & Biases can help with evolving AI assurance requirements. • You’ll play a central role in helping prospective clients evaluate and implement Weights & Biases, guiding them from initial demo to full deployment and customer success. Additionally, you’ll serve as a government vertical’s technical GTM representative for government needs internally across Product and Engineering teams, including but not limited to coordinating the support of post-sale technical and implementation requirements aligned with government standards. Requirements: • Up to 8 years of relevant experience in a similar role working with the public sector or with government clients • Based in the D.C. metro area • Experience using one or more of the following packages: TensorFlow/Keras, PyTorch Lightning • Strong programming proficiency in Python and willingness to work “hands-on keyboard” to help practitioners see the value of W&B’s platform • Ability to adapt machine learning solutions to meet federal compliance standards and an understanding of what it takes to get the technical win in the government market • Excellent communication and presentation skills, both written and verbal, with a focus on explaining technical concepts within the context of federal guidelines • Ability to effectively manage multiple conflicting priorities, respond promptly, and manage time effectively, often impacted by the lengthy timelines and layered approval processes typical of public sector engagements • Ability to break down complex problems and resolve them through customer consultation and execution • Experience with cloud platforms (AWS, GCP, Azure), preferably in environments compliant with FedRAMP or government standards • Experience with Linux/Unix Benefits: • ??? Flexible time off • ?? Medical, Dental, and Vision for employees and Family Coverage • ?? Remote first culture with in-office flexibility in San Francisco • ?? 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 Apply Job!

Similar Jobs

RN UTILIZATION REVIEW SPECIALIST Per Diem

Remote, USA Full-time

Personal Lines Auto Liability Claims Representative

Remote, USA Full-time

Regulatory and Accreditation Compliance Audit Mgr

Remote, USA Full-time

Zillow Flex Buyer and Listing Agent

Remote, USA Full-time

Biomedical Technician I - Atlanta, GA

Remote, USA Full-time

Private Health Insurance - Team Lead

Remote, USA Full-time

Senior Commercial Escrow Officer - Remote

Remote, USA Full-time

R-106311 Lender Placed Insurance Specialist II

Remote, USA Full-time

Mortgage Loan Officer (Remote-Must Be Licensed)

Remote, USA Full-time

QA Engineer 2 - Remote US

Remote, USA Full-time

Contract Management Analyst/ EPIC Contract Builder (Remote Opportunity) in Burr Ridge, IL

Remote, USA Full-time

Strategist, Growth Partnerships and Operations

Remote, USA Full-time

ServiceNow System Administrator (VA ESOM)

Remote, USA Full-time

Apple At Home Advisor Job

Remote, USA Full-time

Patient Safety Project Lead

Remote, USA Full-time

Senior AI Vertical Mini-Series Director (Freelance - Full Remote - Bulgaria)

Remote, USA Full-time

Experienced Entry-Level Remote Live Chat Specialist – Delivering Exceptional Customer Experience at arenaflex

Remote, USA Full-time

Experienced Customer Care Chat Representative | No Prior Experience Required | Offer empathetic and prompt chat-based customer care support in no-prior-experience-required online position at arenaflex

Remote, USA Full-time

Health Program Administrator (FFTL)

Remote, USA Full-time

Experienced Senior Learning Specialist for Innovative Customer Service Training and Development at blithequark

Remote, USA Full-time