GDIT is seeking an AI/ML Architect (Technical Team Lead) to lead AI and Agentic AI solutions for the Department of Veterans Affairs (VA) Veterans Benefits Administration (VBA). In this role, you will focus on transforming claims processing by defining data design patterns for building and maturing large language models (LLMs). You will lead a team to design, build, and deploy AI/ML solutions supporting mission-critical workflows.
What You'll Do
- Lead a team of ~6 direct reports, including Data Scientists, Machine Learning Engineers, and Analysts, to design, build, and deploy AI/ML solutions, including Agentic AI capabilities.
- Oversee the end-to-end development and operationalization of machine learning models, including data preprocessing, feature engineering, training, validation, deployment, and monitoring.
- Apply advanced machine learning techniques and tools to evaluate model performance, interpret results, and deliver actionable insights.
- Assess and validate the effectiveness, accuracy, and quality of new data sources, data gathering methods, and model inputs.
- Develop data-driven strategies and solutions for complex business and operational challenges in a mission-focused environment.
- Collaborate closely with multiple Scrum teams to integrate AI and GenAI workflows into functionality.
- Review test results, model outputs, and workflow performance to verify compliance with VBA objectives.
- Leverage professional judgment and critical thinking to continuously improve AI/ML processes and resolve technical issues.
What We're Looking For
- Bachelor’s Degree in Computer Science, Engineering, Data Science, or a related technical discipline.
- 5+ years of professional experience in data science, machine learning engineering, or AI solution development.
- Minimum 5 years of experience applying machine learning and statistical analysis, including predictive analytics, supervised and unsupervised learning, and model validation.
- 4+ years of experience supporting or developing production-grade data pipelines and machine learning models.
- 2+ years leading technical teams (Data Scientists, ML Engineers, or Software Engineers) in the design, development, and deployment of AI/ML solutions.
- Current or previous people management/direct reports.
- Proficiency in programming languages such as Python, R, Java, or Scala, with practical experience implementing machine learning frameworks (e.g., TensorFlow, PyTorch, scikit-learn).
- Experience with Agile and Lean principles and methodologies for iterative development and deployment.
- Must be able to obtain a Position of Trust and successfully pass a government background investigation, including fingerprinting and detailed forms submission.
Nice to Have
- Master’s Degree preferred.
- Strong knowledge of AI and Agentic AI.
- AWS Certified Machine Learning Engineer – Associate.
- AWS Certified Machine Learning - Specialty.
Technical Stack
- Languages: Python, R, Java, Scala
- Frameworks: TensorFlow, PyTorch, scikit-learn
- Domains: AI Architecture, Generative AI, DevSecOps
Team & Environment
You will lead a team with ~6 direct reports, including Data Scientists, Machine Learning Engineers, and Analysts.
Benefits & Compensation
- Compensation range: $166,600 - $225,400
- Variety of medical plan options, some with Health Savings Accounts
- Dental plan options
- Vision plan
- 401(k) plan with company match
- Full flex work weeks where possible
- Paid time off plans including vacation, sick, personal time, holidays, paid parental, military, bereavement and jury duty leave
- 15 days of paid leave per calendar year (prorated)
- 10 paid holidays per year (prorated)
- GDIT Paid Family Leave program provides up to 160 hours of paid leave in a rolling 12 month period
- Short and long-term disability benefits, life, accidental death and dismemberment, personal accident, critical illness and business travel and accident insurance
Work Mode
This is a remote role open to candidates in any location.
GDIT is an Equal Opportunity Employer / Individuals with Disabilities / Protected Veterans.




