Cockroach Labs

AI/ML Engineer

Cockroach Labs

full-time

Posted on:

Location Type: Hybrid

Location: Bangalore • 🇮🇳 India

Visit company website
AI Apply
Apply

Job Level

Mid-LevelSenior

Tech Stack

PythonSQL

About the role

  • Scale AI Agents and own classical ML projects across the company, with special focus on Revenue
  • Ensure data platforms (Looker and Snowflake) are AI-ready and enable safe, reliable agent access to structured data
  • Lead Predictive Modeling and Machine Learning projects, particularly within the Revenue organization
  • Work with the Head of Data Science to design and evaluate a framework for incorporating Generative AI into the data platform
  • Design and implement reusable tools and frameworks for building and deploying agents
  • Identify and execute high-priority AI workflows and agents with internal partners
  • Build Deal Health models and automate revenue forecasting using techniques such as Logistic Regression, Random Forests, and XGBoost
  • Build pipelines to monitor model performance, tracking metrics like precision at K and F1 score
  • Onboard, build stakeholder relationships, and drive the ML/AI roadmap for Cockroach Labs' Revenue organization
  • Report to David Welgus, Senior Manager of Data Science

Requirements

  • 5+ years of data science, machine learning, and advanced analytics experience in high velocity, high-growth companies
  • Experience collaborating with and understanding the needs of stakeholders from a variety of business functions
  • Strong coding skills in at least one general-purpose language, preferably Python, and familiarity with software engineering principles around testing, code reviews, and deployment
  • Proficient in data analysis using tools like Python, R, and SQL
  • BS, MS or Ph.D. in quantitative fields (e.g., Statistics, Math, Computer Science, Physics, Economics, Operations Research)
  • Experience or familiarity ensuring data platforms (Looker and Snowflake) are AI-ready
Benefits
  • Medical Insurance
  • Flexible Time Off
  • Paid Holidays
  • Paid Parental Leave
  • Mental Wellbeing Benefits
  • And more!

ATS Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
machine learningpredictive modelingdata analysisLogistic RegressionRandom ForestsXGBoostPythonRSQLdata platforms
Soft skills
stakeholder relationshipscollaborationcommunicationleadershiporganizational skills
Certifications
BS in StatisticsMS in StatisticsPh.D. in StatisticsBS in MathMS in MathPh.D. in MathBS in Computer ScienceMS in Computer SciencePh.D. in Computer ScienceBS in Physics
Aven

Data Scientist

Aven
Mid · Seniorfull-timeCalifornia · 🇺🇸 United States
Posted: 24 days agoSource: jobs.ashbyhq.com
PythonSQL
Mozilla

Staff Revenue Data Scientist

Mozilla
Leadfull-time🇺🇸 United States
Posted: 4 days agoSource: boards.greenhouse.io
PythonSQL
Vizcom

Lead Product Manager, Core Product

Vizcom
Seniorfull-time$250k–$285k / yearCalifornia · 🇺🇸 United States
Posted: 32 days agoSource: jobs.ashbyhq.com
PythonSQL
Twin Health

Senior AI Engineer

Twin Health
Seniorfull-time$200k–$220k / year🇺🇸 United States
Posted: 6 days agoSource: boards.greenhouse.io
PythonSQL
TD

Data Scientist III, Financial Crimes Modeling

TD
Mid · Seniorfull-time$95k–$155k / year🇺🇸 United States
Posted: 25 days agoSource: td.wd3.myworkdayjobs.com
PythonSQL