Ai/Ml Platform Engineer II

Summary

Join dbt Labs and build the AI/ML infrastructure for a new product focused on data warehouse cost optimization. This greenfield opportunity involves designing and building AI/ML infrastructure for cost anomaly detection and predictive analytics, laying the foundation for ML-driven cost optimizations within dbt, and working with technologies like Snowflake, Kafka, PyTorch, and TensorFlow. You will collaborate with cross-functional teams and provide technical leadership. The ideal candidate possesses extensive hands-on experience in ML infrastructure, expertise in anomaly detection and predictive analytics, strong Python skills, and experience with data warehouses and large-scale ML systems. dbt Labs offers competitive compensation, equity, and benefits including unlimited vacation, 401k, excellent healthcare, paid parental leave, and wellness stipends.

Requirements

• Extensive hands-on experience in ML infrastructure, including building pipelines and frameworks from scratch
• Expertise in anomaly detection, fraud detection, predictive analytics, and confidence scoring
• Strong Python skills, with proficiency in PyTorch, TensorFlow, and Spark (PySpark preferred, not Java-based Spark)
• Experience working with data warehouses (Snowflake), streaming systems (Kafka), and large-scale ML systems
• A track record of delivering production-ready ML systems —not just research or Jupyter notebooks
• Ability to thrive in a fast-paced, fully remote, distributed team environment
• 5+ years of experience in ML infrastructure, AI/ML engineering, or data engineering
• Bachelor’s degree in a related field, or equivalent professional experience. OR Completed enrollment in related bootcamp
• Experience with high-scale ML infrastructure in a fully remote, asynchronous environment

Responsibilities

• Design and build the AI/ML infrastructure to power cost anomaly detection and predictive analytics
• Lay the foundation for ML-driven cost optimizations —detecting issues and automating remediation within dbt
• Work hands-on with Snowflake, Kafka, PyTorch, TensorFlow, Ibis, and Pandas to process billions of data points
• Collaborate cross-functionally with Product, Engineering, and Design to integrate ML models into dbt workflows
• Provide technical leadership in defining the AI/ML strategy for cost optimization and efficiency
• Influence the growth of our AI/ML engineering team as we scale

Preferred Qualifications

• Experience in ML model evaluation, benchmarking, and automated remediation systems
• A background in building scalable ML platforms that process billions of data points
• Familiarity with cost optimization techniques (CostOps/FinOps) within data platforms and cloud environments
• Prior experience in ML infrastructure engineering at a high-growth startup or large-scale platform

Benefits

• Salary: We offer competitive compensation packages commensurate with experience, including salary, equity, and where applicable, performance-based pay
• The typical starting salary range for this role is: $172,000 - $207,900 USD
• The typical starting salary range for this role in the select locations listed is: $191,000 - $231,000 USD
• Equity Stake
• Unlimited vacation (and yes we use it!)
• 401k w/3% guaranteed contribution
• Excellent healthcare
• Paid Parental Leave
• Wellness stipend
• Home office stipend, and more!