Ai Solution Engineer

RTX; Warminster; Hybrid; GBP 60,000 - 90,000; Full time

Benefits offered by this job

Ground-breaking mission

Collaborative culture

Job summary

OMNIA Training is seeking an AI Solution Engineer to join its UK-based team. You will contribute to designing, delivering, and transforming collective training for the British Army by developing AI/ML solutions and integrating them into secure, DevSecOps-enabled environments.

You will collaborate with cross-functional teams, prepare data, evaluate models, and translate complex AI concepts for stakeholders, while upholding ethical AI practices in defence contexts.

Qualifications

• Experience with Python for AI/ML development.
• Experience with major ML frameworks (PyTorch, TensorFlow, scikit-learn).
• Proven ability to build and maintain end-to-end ML pipelines and deploy to production.
• Ability to document technical solutions for maintainability and compliance.
• Experience with APIs/microservices integration and CI/CD pipelines.

Responsibilities

• Support design, development, and deployment of AI/ML solutions for Army training objectives.
• Ensure alignment with enterprise architecture and DevSecOps pipelines.
• Monitor model performance; apply tuning/retraining strategies.
• Collaborate with data engineers and software teams to build scalable AI components.
• Produce high-quality technical documentation and risk/ethics guidance.

Skills

Python

ML frameworks

ML pipelines

Feature engineering

APIs

Git

CI/CD

Docker

Kubernetes

Cloud

Secure coding

Responsible AI

Analytical

Communication

Cross-functional

Security clearance

Education

Degree in CS/AI

Tools

MLflow

Kubeflow

Model monitoring

OpenShift

Spark

Airflow

Databricks

Job description

AI Solution Engineer

Be at the forefront of transforming British Army training.

About us

At OMNIA Training, we’ve brought together some of the UK’s most innovative defence training organisations under one powerful mission: to transform the British Army’s training system and create the best-trained Army in the world.

OMNIA are redefining the British Army’s collective training. To do that, we are looking for the best and brightest minds from across the UK. We are backed by British innovation and powered by world-class experts, like you. OMNIA is at the heart of the UK’s bold Land Industrial Strategy.

The role

This is more than a job — it’s a mission. You will be part of a high-impact, collaborative environment, where we expect everyone to live the values and standards of the British Army. Every person in our team plays a critical role in delivering OMNIA’s vision; designing, delivering, and transforming collective training so the British Army is ready to fight and win.

You’ll work in a matrix organisation and report operationally through OMNIA Training and functionally through the OMNIA AI Solution Architect. Ultimately, you’ll work for the British Army, championing innovation, and helping shape the future of military collective training.

Key Responsibilities

• Support the development and implementation of AI/ML solutions ensuring alignment with Army training objectives and Omnia architecture principles.
• Contribute to the design, development, and deployment of AI/ML solutions aligned to Army training objectives and enterprise architecture principles.
• Support the development and maintenance of ML pipelines including data pre-processing, feature engineering, model training, validation, and performance evaluation.
• Support integration of AI capabilities into secure operational environments, ensuring compatibility with DevSecOps pipelines and platform constraints.
• Monitor deployed models, analyse performance metrics, and implement tuning or retraining strategies to maintain operational effectiveness.
• Collaborate with data engineers and software teams to support scalable, maintainable, and secure AI components.
• Review and apply ethical AI principles including bias mitigation, data governance, and regulatory compliance in defence contexts.
• Support the production of high-quality technical documentation to support maintainability, auditability, and knowledge transfer.
• Support stakeholder engagement through demonstrations, technical briefings, and translation of complex AI concepts into accessible insights.
• Investigate emerging AI tools and platforms for applicability to Army training use cases.
• Support the development of technical capability through structured learning.
• All other tasks required to deliver the programme.

Who we are looking for

You’ll have a mission focus, and the enthusiasm and drive to ‘get things done’. You’ll want to work in collaboration with other defence training organisations, and the British Army. You won’t let bureaucracy get in the way of what needs to be done, you’ll learn lessons and share these lessons across the team. You’ll understand what it means to put the mission first.

The successful candidate will support the development and integration of AI/ML models aligned with Army training goals and OMNIA architecture, assisting with data preparation, model evaluation, and workflow automation. They will collaborate with technical teams, document processes, monitor model performance, and contribute to ethical AI practices. The role also involves learning emerging technologies, supporting stakeholder communications, and working within DevSecOps engineers to ensure smooth deployment and continuous improvement.

Essential Skills and Experience

• Python programming skills for AI/ML development.
• Experience with at least one major ML framework (e.g. PyTorch, TensorFlow, scikit-learn).
• Experience building and maintaining end-to-end ML pipelines (data ingestion, pre-processing, training, evaluation, deployment) and deployment to production environments.
• Understanding of feature engineering, model validation, and performance evaluation techniques.
• Experience integrating ML components into software systems via APIs or microservices.
• Experience working with version control (Git) and collaborative engineering practices
• Experience contributing to CI/CD pipelines and containerised deployments (e.g. Docker, Kubernetes)
• Experience working with cloud or controlled compute environments
• Ability to document technical solutions for maintainability, audit, and operational support
• Understanding of secure coding principles and data protection requirements
• Awareness of responsible AI considerations (bias mitigation, explainability, privacy)
• Analytical and problem-solving capability
• Ability to communicate technical concepts to non-technical stakeholders
• Experience collaborating within cross-functional engineering teams
• Security cleared or ability to obtain (SC or above).

Desirable Skills and Experience

• Degree in Computer Science, Data Science, Artificial Intelligence, Engineering, or related discipline (or equivalent experience)
• Postgraduate qualification in AI, Data Science, or related field
• Ability to work within regulated or governed environments and frameworks
• Ability to document technical solutions for maintainability, audit, and operational support
• Relevant cloud, AI, or security certifications
• Experience deploying ML models into production environments
• Experience with Kubernetes, OpenShift, or cloud-native deployment architectures
• Experience with MLOps tooling (e.g. MLflow, Kubeflow, model monitoring platforms)
• Experience applying DevSecOps practices within cloud or controlled environments
• Exposure to large language models or generative AI systems
• Experience with data engineering tools (e.g. Spark, Airflow, Databricks)
• Experience applying AI and automation to operational or productivity-focused use cases
• Experience working in secure, restricted, or air-gapped environments
• Experience supporting systems subject to accreditation or compliance processes

What we offer

• Be Part of a Ground-breaking Mission
• Join the UK’s most advanced defence training alliance, where industry experts converge to push boundaries and deliver excellence.
• A Culture Built on Collaboration
• Work in an inclusive, high-performing team that celebrates ideas, supports growth, and is laser-focused on delivering results.
• Team dynamics
• You will be part of a team that works together, across multiple organisations united in one mission.
• Work Across Borders, United in Purpose
• Operate across organisations with one shared goal - improving capability across all domains.
• Drive Innovation
• Your voice matters. We champion new ideas, welcome change, and believe in continuous improvement with a growth mind-set.
• Shape the Future
• Play a crucial role in a long-term, future-facing programme with real-world impact on national defence.
• Career Growth & Benefits
• Enjoy a strong benefits package, ongoing development opportunities, and clear pathways to progress within a trusted defence industry partner.

OMNIA Training and its partners are proud to be an equal opportunity employer. We welcome applicants of all backgrounds and identities — what matters is your passion, your expertise, and your dedication to the mission.

This position is classified as

Hybrid: Employees who are working in Hybrid roles will work regularly both onsite and offsite. Ratio of time working onsite will be determined in partnership with your leader.

Candidates will learn more about role type and current site status throughout the recruiting process. For onsite and hybrid roles, commuting to and from the assigned site is the employee’s personal responsibility.