Senior AI/ML Engineer

Job description

T-Rex Solutions is looking to select a Senior AI/ML Engineer to support the US Department of the Treasury TCloud enterprise cloud development effort. We are seeking a seasoned AI/ML Engineer to join the IRS’s flagship Analytics Application Platform (AAP) — a mission-critical Platform-as-a-Service (PaaS) that enables secure, compliant, and scalable AI/ML workloads across the agency. AAP empowers mission teams to develop and operationalize both traditional and GenAI models through a unified environment that integrates Databricks, JupyterHub, AWS SageMaker, Bedrock, and other core services. As an AI/ML Engineer, you will play a central role in building reusable patterns, advancing infrastructure readiness, and ensuring all platform services meet the technical, compliance, and performance standards required for IRS-wide production use.

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.

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.

Desired:

• 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

T-Rex Overview

Established in 1999, T-Rex Solutions, LLC is a proven mid-tier business providing data-centric mission services to the Federal government as it increasingly tries to secure and leverage the power of data. We design, integrate, secure, and deploy advanced technical solutions for our customers so they can efficiently fulfill their critical objectives. T-Rex offers both IT and professional services to numerous Federal agencies and is a leader in providing high quality and innovative solutions in the areas of Cloud and Infrastructure Services, Cyber Security, and Big Data Engineering.

T-Rex is constantly seeking qualified people to join our growing team. We have built a broad client base through our devotion to delivering quality products and customer service, and to do that we need quality individuals. But more than that, we at T-Rex are committed to creating a culture that supports the development of every employee’s personal and professional lives. T-Rex has made a commitment to maintain the status of an industry leader in compensation packages and benefits which includes competitive salaries, performance bonuses, training and educational reimbursement, Transamerica 401(k) and Cigna healthcare benefits.

T-Rex is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, religion, color, sex (including pregnancy and sexual orientation), parental status, national origin, age, disability, family medical history or genetic information, political affiliation, military service, or other non-merit based factors.

In compliance with pay transparency guidelines, the annual base salary range for this position is $145,000 - $175,000. Please note that the salary information is a general guideline only. T-Rex considers factors such as (but not limited to) scope and responsibilities of the position, candidate’s work experience, education/training, key skills, internal peer equity, as well as market and business considerations when extending an offer.

T-Rex offers a diverse and collaborative work environment, exciting opportunities for professional growth, and generous benefits, including: 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, and a 401(k) plan with company match fully vested after 60 days of employment among other benefits.