Software Engineer, ML Data Infrastructure
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
As a Software Engineer, ML Data Infrastructure, you will design and ship scalable data infrastructure powering petabyte-scale, multi-modal model training.
Tackle complex distributed systems challenges, optimizing data processing workflows for high throughput and reliability.
Collaborate with research scientists to translate data requirements into production-grade systems that accelerate model development cycles.
Own features end-to-end—from scoping and implementation to launch—balancing performance, reliability, and maintainability.
Contribute to a Python-based stack and related tools (Kubernetes, Docker, Terraform) while leveraging GCP services (Bigtable, BigQuery, Spanner, Pub/Sub) and TPU infrastructure.
Based in Toronto with on-site collaboration and ample autonomy to experiment and access compute resources as needed.
Tackle complex distributed systems challenges, optimizing data processing workflows for high throughput and reliability.
Collaborate with research scientists to translate data requirements into production-grade systems that accelerate model development cycles.
Own features end-to-end—from scoping and implementation to launch—balancing performance, reliability, and maintainability.
Contribute to a Python-based stack and related tools (Kubernetes, Docker, Terraform) while leveraging GCP services (Bigtable, BigQuery, Spanner, Pub/Sub) and TPU infrastructure.
Based in Toronto with on-site collaboration and ample autonomy to experiment and access compute resources as needed.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.
? Machine Learning Intern Fall 2026 (Toronto) ? Director, Data Science ? Research Intern – Explainable AI and Reporting Framework for Power Grid ML Applications ? Software Development Manager, Measurement, AdTech and Data Science, Ads Performance Measurement ? Software Engineer, Machine Learning Infrastructure ? Principal Software Engineer - AI Platform
Remote workno
CityToronto, Canada