Data Science And Ai Platform Engineer

Summary

Join T-Rex Solutions as a Data Science & AI Platform Engineer to build and maintain a secure, cloud-hosted production laboratory for the US Department of Treasury. You will develop and maintain data pipelines, implement and manage various data science tools (Databricks, JupyterHub, MLflow), integrate with AWS services (S3, Lambda, RDS, SNS), and ensure compliance with IRS standards. Responsibilities include establishing model development pipelines, managing data security, and supporting data science teams in adhering to Responsible AI principles. A Bachelor's or Master's degree in a related field and 7 years of experience are required, along with expertise in AWS and data science tools. US citizenship and the ability to obtain a Public Trust clearance are mandatory.

Requirements

• Be a US Citizen with the ability to obtain a Public Trust clearance
• Have a Bachelor’s or Master’s degree in Computer Science, Data Science, Engineering, or a related field and 7 years experience
• Have proven experience with cloud services, particularly AWS (S3, Lambda, RDS, SNS, etc.)
• Have expertise in setting up and managing data science environments with tools such as Databricks, JupyterHub, MLflow, and Bitbucket
• Have strong proficiency in programming languages and tools such as SQL, Python, R, and Spark
• Have familiarity with machine learning frameworks and libraries
• Have a deep understanding of data security and privacy, especially in compliance with IRS standards
• Have experience with CI/CD pipelines and infrastructure as code
• Have excellent communication skills and the ability to collaborate with cross-functional teams

Responsibilities

• Develop and maintain a data exploration and feature engineering pipeline to enable data readiness for modeling within a secure and compliant infrastructure
• Implement and manage Databricks, JupyterHub, and other modern tools for exploratory data analysis, AI design, model development, and training
• Integrate tools with TCloud Bitbucket for version control and collaboration
• Configure AWS S3 buckets for Feature Store, ensuring compliance with IRS UNAX rules and separate security contexts for different teams
• Establish and maintain a model development pipeline utilizing MLflow for experiment tracking and model registry
• Create AWS Lambda functions and implement AWS Flow for efficient model artifact storage and metadata management
• Integrate with AWS RDS for robust data management and retrieval
• Configure SNS notifications for model promotion workflows
• Support data science teams in adhering to Responsible AI Principles via the implementation of the Responsible AI Toolbox

Preferred Qualifications

• Have an Active Treasury clearance
• Have previous experience working in a regulated environment, particularly within government agencies like the IRS
• Have certifications related to AWS or cloud architecture
• Have knowledge of Responsible AI practices and principles

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
• Transamerica 401(k)
• Cigna healthcare benefits