AI/ML Engineer

This is a U.S. based position. All of the programs we support require U.S. citizenship to be eligible for employment. All work must be conducted within the continental U.S. About the role As an AI/ML Engineer embedded in Rome, NY, you will work directly on model development efforts while leveraging and extending [R]AIMS platform capabilities to accelerate experimentation, evaluation, deployment, and operational transition. You will work closely with platform engineers, AI leadership, and mission stakeholders to move models from experimentation through production. The work sits at the intersection of applied machine learning, model training and evaluation, AI platform engineering, and operational AI deployment. What You'll Do Build and evaluate machine learning models for mission-relevant use cases working directly with government researchers and program stakeholders to understand requirements and translate them into executable technical solutions Develop and maintain model training, fine-tuning, and benchmarking workflows that are reproducible, well-documented, and usable by teammates without hand-holding Build and improve evaluation pipelines for repeatable, rigorous performance measurement across model architectures, datasets, and operational scenarios Integrate models into production-ready [R]AIMS platform infrastructure, working with platform engineers to ensure deployments are containerized, observable, and operationally sustainable Support experimentation across model architectures and datasets, maintaining clear records of results and surfacing actionable findings to AI leadership and mission stakeholders What we are looking for 3 to 6 years of hands‑on experience building and shipping production software or AI/ML systems Strong Python software engineering skills; writes clean, maintainable, production-quality code rather than notebook‑only scripts Demonstrated experience developing and evaluating machine learning models, with a clear understanding of what makes an evaluation rigorous versus misleading Hands‑on familiarity with modern ML frameworks such as PyTorch, TensorFlow, JAX, or Hugging Face Experience building and managing model training pipelines and experimentation workflows at a level beyond tutorial projects Experience working with distributed systems or cloud‑native environments; comfortable in infrastructure that isn't fully managed for you Strong debugging instincts; able to diagnose failure modes in complex pipelines and explain findings clearly to both technical and non‑technical audiences Ability to work independently and manage workstreams without close supervision while staying well‑integrated with a distributed team Strong written and verbal communication skills; able to produce clear technical documentation, status updates, and evaluation summaries Ability to obtain Security+ certification within the first 90 days of employment U.S. citizenship required; ability to obtain and maintain a Top Secret/SCI clearance Highly Preferred Experience fine‑tuning foundation models, LLMs, or multimodal models for specific domain tasks or constrained operational environments Experience designing or operating model evaluation frameworks and benchmarking pipelines at scale Experience with Kubernetes and containerized ML workloads, including deploying and debugging GPU‑enabled inference services Experience with distributed training or large‑scale inference systems Familiarity with streaming or event‑driven architectures such as Kafka or Flink, particularly as they relate to real‑time model inputs or outputs Experience building secure, compliant AI systems for regulated or mission‑critical environments, including familiarity with RMF or IL requirements Prior defense, national security, or government R&D experience, particularly with AFRL or Air Force programs Experience working in prototype‑to‑production environments where research artifacts need to become operational systems Active Secret or Top Secret clearance strongly preferred What Success Looks Like Models developed and evaluated at AFRL are delivered with clear, rigorous documentation of performance, limitations, and operational considerations—not handed off as black boxes Evaluation pipelines are repeatable and trusted by the broader team; results are reproducible and traceable Model integrations into [R]AIMS are clean, containerized, and maintainable by platform engineers without needing the original model developer in the loop AFRL stakeholders view Raft as a technically credible, reliable partner; your presence in Rome strengthens that relationship over time The gap between experimentation and operational deployment shortens with each program cycle because of the infrastructure and workflows you helped build Clearance Requirements No clearance required to start Must be eligible for and willing to obtain a Top Secret/SCI clearance; active clearance strongly preferred Salary Range: $170,000.00 - $220,000.00 Work Type Hybrid in Rome, NY; candidates must be based in or willing to relocate to the Rome, NY area to support a hybrid schedule Up to 25% travel What we will offer you Highly competitive salary Fully covered healthcare, dental, and vision coverage 401(k) and company match Take as you need PTO + 11 paid holidays Education & training benefits Generous Referral Bonuses And More! We're an equal opportunity employer. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status. #J-18808-Ljbffr Raft