Data Science Ai Engineer (Remote)

Take2 is seeking multiple experienced Data Science & AI Platform Engineers to build and maintain a secure, monitored, cloud-hosted production laboratory in support of our US Department of Treasury customer. The laboratory will be used by data science teams to create analytics for insights and AI/ML models for deployment in a high-stakes production environment.

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.

Requirements:

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

Desired:

• Previous experience working in a regulated environment, particularly within government agencies like the IRS.
• Certifications related to AWS or cloud architecture.
• Knowledge of Responsible AI practices and principles.