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

W&B; Washington (District of Columbia); Remote; USD 137,000 - 186,000; Full time

Job summary

An established industry player is seeking a Senior AI Solutions Engineer to spearhead technical engagements with federal clients. This role is pivotal in enhancing the public sector practice by guiding clients from initial demos to successful deployments of AI solutions. Collaborating closely with internal teams, you will ensure that products meet government standards while addressing critical challenges in machine learning, computer vision, and NLP. If you are passionate about leveraging AI to solve real-world problems and thrive in a dynamic environment, this opportunity is for you.

Benefits

Flexible time off

Medical, Dental, and Vision coverage

Home office budget with a new laptop

12 weeks of Parental leave

401(k) plan

Qualifications

• 5+ years of experience in a similar role with public sector clients.
• Strong programming skills in Python with hands-on experience.

Responsibilities

• Lead pre- and post-sales engagements with federal clients.
• Coordinate with internal teams to meet federal compliance standards.

Skills

Python

Machine Learning

NLP

Computer Vision

Cloud Platforms (AWS, GCP, Azure)

Communication Skills

Problem Solving

Presentation Skills

Education

Bachelor's Degree in Computer Science or related field

Tools

LangChain

LlamaIndex

TensorFlow/Keras

PyTorch Lightning

Docker

Kubernetes

Job 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 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.

Responsibilities:

• 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

Requirements:

• 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 standards
• Experience with Linux/Unix

Strong plus:

• 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

Our 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
• Explore benefits bycountry

This position has an annual estimated salary of $137,000 - $186,000 per year. This position has an additional variable compensation in the form of a bonus or commission component, which is dependent on personal or company performance. Weights & Biases is committed to providing competitive salary, equity and benefits packages for all full-time employees. The actual pay may vary depending on your skills, qualifications, experience, and work location.

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 [e-mail].

#LI-Remote