Senior AI/ML Engineer

Summary

Join T-Rex Solutions as a Senior AI/ML Engineer to contribute to the US Department of the Treasury's T-Cloud enterprise cloud development. This mission-critical role involves working on the IRS's Analytics Application Platform (AAP), a Platform-as-a-Service (PaaS) enabling secure AI/ML workloads. You will build reusable patterns, enhance infrastructure, and ensure platform services meet IRS standards. Responsibilities include developing data pipelines, operationalizing MLFlow, integrating with CI/CD pipelines, managing AWS S3 feature stores, and supporting GenAI enablement. Collaboration with various teams and implementation of Responsible AI tooling are also key aspects of this position. The ideal candidate will possess strong AI/ML engineering experience and a deep understanding of secure, compliant ML infrastructure in a regulated environment.

Requirements

• Must be a US Citizen with the ability to obtain a Public Trust clearance
• Bachelors degree and seven (7) years or more of related experience or Masters and five (5) or more years of related experience
• Strong experience in AI/ML engineering, with significant hands-on experience using Databricks, JupyterHub, and AWS AI/ML services (e.g., SageMaker, Bedrock, Lambda, RDS, S3)
• Proficient in Python, SQL, Spark, and machine learning frameworks used in both model development and deployment
• Demonstrated experience with MLFlow, CI/CD pipelines, infrastructure-as-code, and Git-based version control
• Deep familiarity with building secure, compliant ML infrastructure in a regulated federal environment

Responsibilities

• Build and maintain robust pipelines for data ingestion, exploration, feature engineering, and model training across Databricks and AWS-native services
• Operationalize tools such as MLFlow for experiment tracking, model registry, versioning, and lifecycle governance
• Integrate with T-Cloud Bitbucket and CI/CD pipelines (e.g., Bamboo) to enable secure, traceable development workflows
• Configure and manage IRS-compliant AWS S3 feature stores, ensuring UNAX-compliant isolation and secure access per team
• Develop and orchestrate AWS services (Lambda, RDS, SNS, EventBridge) to support automated model promotion and metadata tracking
• Support GenAI enablement by engineering reusable, secure pathways for customers to leverage AWS Bedrock and SageMaker for LLM-based applications
• Collaborate across the Infrastructure, Customer Success (CSx), and ATO/OneSDLC teams to support onboarding and productionalization for IRS customer use cases
• Implement and maintain Responsible AI tooling, including the Responsible AI Toolbox, to support auditability and ethical deployment practices
• Produce high-quality documentation and reusable code artifacts that simplify customer experience and platform adoption
• Support platform-wide upgrades (e.g., Databricks E2), refactoring of inherited IaC, and improvements in CI/CD automation and monitoring

Preferred Qualifications

• Prior experience supporting AI/ML in regulated environments such as IRS, Treasury, or other federal agencies
• Familiarity with Immuta, Trustworthy AI controls, and model auditability frameworks
• AWS certification or similar credentials in cloud engineering or DevOps
• Exposure to customer onboarding, cross-team collaboration, and platform-as-a-service delivery models
• Security Clearance desired

Benefits

• PTO available to use immediately upon joining (prorated based on start date)
• Paid parental leave
• Individual and family health, vision, and dental benefits
• Annual budget for training, professional development and tuition reimbursement
• A 401(k) plan with company match fully vested after 60 days of employment
• Competitive salaries
• Performance bonuses
• Training and educational reimbursement
• Transamerica 401(k) and Cigna healthcare benefits