AI Software Engineer – Machine Learning Platform

We’re looking for an AI Software Engineer to join our Machine Learning Platform team. You’ll design, build, and scale cloud infrastructure that powers ML models, pipelines, and data processing — ensuring reliability, performance, and ease of use for our ML and Data Science teams.

Key Responsibilities

• Architect & Develop: Build scalable ML infrastructure on Google Cloud (Vertex AI, BigTable, BigQuery, Cloud Composer).
• Automate ML Workflows: Streamline data ingestion, feature engineering, training, and deployment.
• Optimize Performance: Improve data pipelines for low latency and cost efficiency.
• Ensure Reliability: Implement monitoring, alerting, and failover for high availability.
• Drive Innovation: Stay current with best practices in cloud, ML ops, and data engineering.

What Sets You Apart

• Customer-First Mindset: You build seamless, self-service ML platforms for data scientists.
• Collaborative & Analytical: Skilled in diagnosing complex infrastructure and ML pipeline issues.
• Automation-Focused: You prioritize reliability through scripting and DevOps excellence.
• Adaptable & Ownership-Driven: Always learning, improving, and taking initiative.

Qualifications

• Education: Bachelor’s or Master’s in Computer Science, Engineering, or related field.
• Experience: 6+ years in software or ML infrastructure engineering.
• Technical Skills:
• Hands-on with GCP (Vertex AI, BigTable, BigQuery, Cloud Composer)
• Proficient in Python (required for test), Java, or SQL
• Experienced with Apache Airflow, CI/CD, Docker, Kubernetes