Senior AI Solutions Engineer, Federal - Washington, D.C.

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 AI Solutions Engineer to lead the technical engagement and success of our federal and public sector clients. 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.

• Pre-Sales: Partner with the Head of Federal to serve as the technical lead for pre-sale engagements designing and executing public sector-oriented initiatives to demo and show the capabilities of Weights & Biases to meet government requirements
• Pre-Sales: Provide in-depth demonstrations, articulating best practices for implementing ML pipelines tailored to federal use cases
• Pre-Sales: Partner with GTM team to manage pre-sales requirements from each stage and tactically drive those motions
• Post Sales: Lead coordination in partners with W&B’s solution architecture team to help federal customers in deploying, configuring, and optimizing Weights & Biases solutions, ensuring success in their unique environments
• Cross-Functional: Act as a bridge between federal customer needs and internal teams, providing insights and feedback for product development and prioritization.
• Cross-Functional: Engage with Product, Engineering, and Support teams to ensure that federal-specific requirements are incorporated into our product roadmap.
• Cross-Functional: Guide customers in adapting Weights & Biases solutions to meet federal regulations, including FedRAMP and DoD standards.
• Cross-Functional: Stay informed on relevant federal compliance standards, enabling federal teams to maintain adherence to regulations across deployments

• Up to 5+ years of relevant experience in a similar role working with the public sector or with government clients
• Experience using one or more of the following packages: LangChain, LlamaIndex,  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 standardsExperience with Linux/Unix

• Active US Government Security Clearance
• Based in the D.C. metro area
• A current Secret or TS/SCI clearance
• A current CI-poly (Counterintelligence polygraph)- Helping customers move on-prem to cloud
• Proficiency with one or more of the following packages / LLMs / frameworks: HuggingFace, Fastai, MCP, OpenAI Finetuning API, Microsoft Azure OpenAI, Gemini, Bedrock, Claude. 
• Experience with hyperparameter optimization solutions
• Experience with data engineering, MLOps and tools such as Docker and Kubernetes
• Experience with data pipeline tools

• 🏝️ 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
• 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 careers@wandb.com.

#LI-Remote