Software Engineer Ai Model Training Jobs (NOW HIRING)

At Gallatin, we are rebuilding logistics infrastructure for the national security missions of the United States and allied partners. We build AI systems that determine how logistics decisions are made—not just how they're executed. From factory to foxhole, we operate at the layer where data becomes decisions, and decisions make the advantage.

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.

Mission and Identity

We are building the system that enables faster, smarter logistics decisions in contested environments, and we're doing it with a team of seasoned entrepreneurs, operators, and technologists who have built and scaled solutions in this space before. We hold ourselves to an extremely high standard. We value clear thinking, direct communication, and the kind of ownership that doesn't stop until something actually works.

Our mission is to create decision advantage when the stakes are the highest. If we succeed, the system doesn't just run; it gets smarter. We're not building AI for its own sake. We're building it because faster, smarter decisions in the most demanding environments on earth can't wait. If you want to work somewhere the stakes are real and the mission is urgent—you'll fit in here.

Why Gallatin?

The logistics infrastructure that supports America's warfighters and humanitarian disaster responders is overdue for transformation, and we are building it. From defense operations to disaster response, we're solving the hardest problems that keep missions moving when it matters most. Join a team where the mission is the point.

Compensation: Gallatin offers competitive compensation commensurate with experience. Actual compensation may vary based on experience, skills, and location. In addition to base salary, we offer a generous equity grant, full healthcare coverage, 401k, unlimited PTO, and the perks of working in a high-caliber, mission-driven environment.

Gallatin is an equal opportunity employer. We do not discriminate on the basis of race, color, religion, sex, national origin, age, disability, veteran status, sexual orientation, gender identity, or any other characteristic protected by applicable federal, state, or local law.

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.

We comply with the United States Department of Labor's Pay Transparency provision.