AI/ML Software Engineer

Job Number: R0237289The OpportunityTo achieve an organization's mission, leaders need strong team members who can create and analyze processes, communicate requirements, and develop innovative solutions throughout the execution of the mission. Whether reviewing program-wide technical architecture or providing AI/ML expertise, we need someone who combines deep technical understanding of software engineering with strong architectural judgment. That is why we need you, an experienced AI/ML Software Engineer who can operate at a system-of-systems level to support clients in advancing AI-enabled systems within an R&D environment.As part of our team, you'll serve as a strategic technical advisor to the Advanced Research Projects Agency for Health (ARPA‐H), helping conceptualize, create, and execute advanced government-funded R&D programs to accelerate better health outcomes for everyone. You'll work with world-class scientists and engineers to support the development of high-impact solutions to society's most challenging health problems. You'll leverage technical expertise to provide strategic assessments of new technologies in support of senior ARPA‐H decision makers. You'll maintain responsibility for producing and presenting findings and recommendations to a team of colleagues and clients on the feasibility and potential impact of future research programs, assisting with the management of current programs, and facilitating commercialization of successfully developed technologies.You'll serve as an AI/ML Software Engineer advising program leadership and supporting software engineering to ensure that program-wide technical architecture and engineering meet rigorous AI development, evaluation, and long-term impact standards. Your attention to detail, flexibility, communication skills, understanding of the client's mission, and problem‐solving will enable the mission's success.What You'll Work OnDeploy AI models and services on cloud platforms such as AWS, Google Cloud, or Azure.Collaborate with cross-functional teams, including data scientists, researchers, and other engineers, to integrate LLMs and RAG into broader projects.Extend agentic capabilities, including agent orchestration and tool calling, using MCP and A2A.Communicate technical concepts and project progress to non-technical stakeholders and team members.Conduct experiments to fine-tune parameters and optimize model performance.Implement techniques to improve the efficiency and scalability of LLM and RAG systems.Stay up-to-date on the most current developments in the generative‐AI space.Monitor and maintain deployed models to ensure they perform reliably and meet performance benchmarks.Ensure code quality by following best practices in software development, including version control, testing, and continuous integration or continuous deployment (CI/CD).Document code and maintain comprehensive technical documentation to support team knowledge sharing and project handovers.You Have5+ years of experience delivering production-ready, industrial strength code, and implementing CI/CD pipelines.Experience in Python and with libraries such as TensorFlow and PyTorch.Experience with neural network architecture, including transformers, and NLP techniques.Experience with LLM tool or function calling and agentic orchestration.Experience handling and preprocessing large datasets, and with data pipelining and ETL processes.Experience with cloud platforms such as AWS, Google Cloud, or Azure for deploying ML models.Experience with software engineering fundamentals, including version control, testing, and CI/CD.Ability to design efficient algorithms for data retrieval and model training, and optimize models through hyperparameter tuning.Ability to travel up to 25% of the time.Master's degree.Nice If You HaveExperience producing and presenting high-quality technical documentation and presentations in a fast-paced, dynamic environment.Experience operating in an R&D environment.CompensationAt Booz Allen, we celebrate your contributions, provide you with opportunities and choices, and support your total well-being. Our offerings include health, life, disability, financial, and retirement benefits, as well as paid leave, professional development, tuition assistance, work‐life programs, and dependent care. Our recognition awards program acknowledges employees for exceptional performance and superior demonstration of our values. Full-time and part-time employees working at least 20 hours a week on a regular basis are eligible to participate in Booz Allen's benefit programs. Individuals that do not meet the threshold are only eligible for select offerings, not inclusive of health benefits. The projected compensation range for this position is $69,400.00 to $158,000.00 (annualized USD). This posting will close within 90 days from the Posting Date.Work ModelRemote: If this position is listed as remote, there may still be occasions when you are required to work in person at a Booz Allen or customer facility.Hybrid: If this position is listed as hybrid, you will be expected to work from a Booz Allen facility frequently, in alignment with leadership expectations and the needs of the role. You may also be required to work from or visit a customer facility.Onsite: If this position is listed as onsite, work will primarily be performed at a Booz Allen office or customer facility, where employees will collaborate directly with colleagues and customers as required by the role.Commitment to Non-DiscriminationAll qualified applicants will receive consideration for employment without regard to disability, status as a protected veteran or any other status protected by applicable federal, state, local, or international law.#J-18808-Ljbffr