AI Software Engineer

The role

You will build and deploy AI-powered solutions across the full AI/ML lifecycle. This includes developing models, establishing MLOps pipelines, and integrating various data sources for core products.

What you will do

• Own and develop an AI model aligned with 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.
• Implement monitoring to ensure high model reliability and uptime.
• Identify and integrate structured, unstructured, batch, and real-time data sources.
• Work with internal logs, APIs, user interactions, and third-party datasets.
• Stay ahead of emerging AI trends and technologies.
• Document and share AI/ML development best practices.
• Contribute to the hiring and mentoring of AI/ML talent.
• Help the wider team understand latest trends and novel AI approaches.

Required qualifications

• Expertise in at least one: Python, C++, or C.
• Hands-on experience with AI/ML training and fine-tuning frameworks, including: PyTorch, TensorFlow, CUDA, Jupyter Notebooks
• Hands-on experience with Large Language Models (LLMs) and Open-Source Models (Llama, Anthropic, Mistral)
• Hands-on experience with LangChain, Retrieval-Augmented Generation (RAG), Hugging Face, Exo Labs
• 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.
• Deep understanding of deep neural networks (DNNs), LLMs, over/underfitting, prompt engineering, and LLM security (jailbreaking risks and protections).
• A passion for continuous learning, staying current with the latest advancements in AI/ML, and teaching others

Nice to have

• Experience mentoring and upskilling junior engineers.
• Familiarity with legacy systems and working with public sector customers.
• Contributions to open-source (AI/ML) projects.

Benefits

• generous equity grant
• full healthcare coverage
• 401k
• unlimited PTO
• perks of working in a high-caliber, mission-driven environment

Before you apply

• This position may require the ability to obtain and maintain a U.S. government security clearance.
• The successful candidate must be able to work in a classified environment when necessary.
• U.S. citizenship is a requirement for all positions at Gallatin.
• Proof of citizenship will be required prior to employment if selected.

About the company

Gallatin is rebuilding logistics infrastructure for the national security missions of the United States and allied partners. They build AI systems that determine how logistics decisions are made.