Senior Python Engineer

Role Description

Our team builds the developer tooling, platforms, systems and experiences that power Cloud AI Platform. In this role you will partner directly with internal customers to understand their use cases, evaluate technical requirements, and build AI-driven systems and solutions that leverage Cloud AI Platform capabilities. You will prototype quickly, harden solutions for production, build services and feed insights back to platform teams to influence roadmap and improve the developer experience.

You will also act as a bridge between product management, partner platform, and customer teams by helping define best practices, documenting patterns, and working closely with platform engineering groups to drive alignment and deliver systems. Success in this role requires a combination of strong engineering fundamentals, applied ML awareness, platform thinking, customer empathy, and the ability to deliver in fast-evolving environments.

• Partner directly with internal product teams to understand AI/ML use cases and translate requirements into technical solutions.
• Build production-ready services, integrations, workflows, and developer tooling on top of Cloud AI Platform.
• Prototype solutions rapidly, validate approaches with customers, and harden successful prototypes for production.
• Identify recurring customer needs and translate them into reusable platform capabilities and tooling.
• Collaborate with platform teams to improve APIs, SDKs, workflows, documentation, and developer experience.

Qualifications

• Experience designing, building, and maintaining ML infrastructure and deployment pipelines using containerization technologies (Docker, Kubernetes preferred) and cloud platforms (AWS, Azure, or GCP).
• Proficient coding skills in Python, Go, or Scala.
• Excellent grasp of software engineering fundamentals and DevOps practices.
• Strong experience with Infrastructure as Code (Terraform, CloudFormation) and CI/CD tools (Jenkins, GitLab CI, GitHub Actions).
• Experience with data pipeline orchestration tools (Airflow, Prefect, Dagster) and streaming platforms (Kafka, Kinesis).
• Proficient knowledge of Git and collaborative development workflows.
• Proficiency in monitoring and observability tools (Prometheus, Grafana, ELK stack) for ML model performance and system health.
• BS, MS in Computer Science, Software Engineering, Machine Learning, or equivalent degree with applicable experience.
• 3+ years of experience in MLOps, DevOps, or related infrastructure roles.
• Experience working in cross-functional teams and communicating technical concepts to diverse audiences.

Requirements

• Experience in ML frameworks (TensorFlow, PyTorch, MLflow, Kubeflow).
• Understanding of security best practices for ML systems and data governance.
• Knowledge of ML model versioning, experiment tracking, and feature stores (MLflow, Weights & Biases, Feast).
• Experience with automated testing frameworks for ML systems, including data validation and model testing.

Benefits

• Opportunity to work on cutting-edge projects.
• Work with a highly motivated and dedicated team.
• Competitive salary.
• Flexible schedule.
• Benefits package - medical insurance, vision, dental, etc.
• Corporate social events.
• Professional development opportunities.
• Well-equipped office.