AI Software Engineer
$80k - $120k USD
About Gallatin
At Gallatin, we are rebuilding defense logistics for the warfighters of the United States and allied forces. We take an AI-first approach to improve defense readiness through software products that streamline and modernize logistics operations from factory to foxhole and result in better decisions and outcomes.
We believe that you won’t change the world by phoning it in and as such we work from our office in El Segundo, California or Austin, Texas.
About the Role
We’re looking for an AI/ML Software Engineer to play a foundational role in developing and deploying AI-powered solutions for our core product. You’ll work across the full AI/ML lifecycle—from defining and building models to evaluating and deploying large scale ML pipelines and real-time inference systems—while collaborating with cross-functional teams to deliver impactful, real-world AI applications.
What You'll Do
Build and Deploy AI/ML Models
Own and develop an AI model aligned with our core product needs within the first three months.
Validate AI-driven product features through real-world testing and feedback from warfighters.
Establish scalable MLOps pipelines and real-time inference services to streamline model training, deployment, runtime, and monitoring.
Own high model reliability and uptime by implementing monitoring across your work.
Data Collection & Processing
Identify and integrate structured, unstructured, real-time, and batch data sources.
Work with internal logs, APIs, user interactions, and third-party datasets to improve model training quality.
Continuous Improvement & Best Practices
Stay ahead of AI trends and emerging technologies to improve model performance.
Document and share best practices for AI/ML development.
Contribute to the hiring and mentoring of AI/ML talent as we grow our team.
Teach the wider team about the latest trends and significance in novel approaches in AI.
What We’re Looking For
Strong Programming Skills
Expertise in at least one: Python, C++, or C.
ML Framework Proficiency
Hands-on experience with AI/ML training and fine-tuning frameworks, including:
PyTorch, TensorFlow, CUDA, Jupyter Notebooks
Large Language Models (LLMs) and Open-Source Models (Llama, Anthropic, Mistral)
LangChain, Retrieval-Augmented Generation (RAG), Hugging Face, Exo Labs
MLOps & Data Engineering Knowledge
Experience with data processing and pipeline frameworks such as Apache Kafka, Apache Airflow, AWS Kinesis, pandas, and dbt.
Understanding of AI model performance monitoring, data drift detection, and observability tools like Grafana or Kibana.
AI Fundamentals & Security Awareness
Deep understanding of deep neural networks (DNNs), LLMs, over/underfitting, prompt engineering, and LLM security (jailbreaking risks and protections).
Growth Mindset
Passion for continuous learning and staying current with the latest advancements in AI/ML, as well as teaching others
Bonus Points
Experience mentoring and upskilling junior engineers.
Familiarity with legacy systems and working with public sector customers.
Contributions to open-source (AI/ML) projects.
Why Gallatin?
Join a mission-driven, high-impact, and fast-moving startup where your work directly improves defense logistics readiness for the warfighters of the United States and allied forces.
Work alongside a team of passionate engineers, designers, and industry experts.
Competitive compensation incl. generous options grant, 100% employer-paid health insurance premiums, 401k, opportunities for rapid career growth, unlimited PTO, free lunches and snacks in the office.
This position requires the ability to obtain and maintain relevant security clearances. The successful candidate must be able to work in a classified environment when necessary.
If you’re not excited about working hard, and solving the hard problems of building AI decisions-support systems that enhance our warfighters, don’t apply.
The following ranges are based on the cost of labor across US geographic areas. The base ranges from our lowest geographic market to our highest geographic market.
Software Engineer I $80,000-$120,000
Software Engineer II $84,000-$175,000
SR Software Engineer $134,000-$210,000