Home
Jobs
Avenue Code
AI/ML Infrastructure Engineer

AI/ML Infrastructure Engineer

Hybrid

Toronto, Canada

Mid level

Full Time

18-12-2025

Share this job:

Score my CV

Skills

Python Go Kubernetes Training Machine Learning PyTorch TensorFlow Deep Learning Agile

Job Specifications

About The Opportunity

About the job

The team is dedicated to developing a robust platform for training and serving machine learning models. This platform streamlines the productionization of AI and ML models by mitigating the incidental complexities involved in creating backend services for serving predictions and training models.

Responsibilities

Contribute to the client's ML Platform SDK and build tools for various ML operations.
Collaborate with Machine Learning Engineers (MLE), researchers, and various product teams to deliver scalable ML platform tooling solutions that meet the timelines and specifications of given requirements.
Work independently and collaboratively on squad projects that often requires learning and applying new technologies that may go beyond existing skillsets.
Manage and maintain large scale production Kubernetes clusters for ML workloads, including ML platform infrastructure and necessary dev ops.
Designs, documents and implements reliable, testable and maintainable solutions ML infrastructure capabilities.

Required Qualifications

You have 6+ years of hands-on experience implementing production ML infrastructure at scale in Python, Go or similar languages.
You have knowledge of deep learning fundamentals, algorithms, and open-source tools such as Huggingface, Ray, PyTorch or TensorFlow
You have an understanding of distributed training leveraging GPUs and Kubernetes
You have a general understanding of data processing for ML
You have experience with agile software processes and modular code design following industry standards

About the Company

What happens when you put together a group of talented engineers, decades of combined experience working in digital transformation, and the determination to deliver an extraordinary experience for clients, partners, and employees? You create the leading software consultancy focused on delivering end-to-end digital transformation solutions for enterprise organizations of all kinds. Avenue Code is more than just a software development company. When it comes to our practice areas, we are experts in web solutions, infrastructur... Know more