At BairesDev®, we've been leading the way in technology projects for over 15 years. We deliver cutting-edge solutions to giants like Google and the most innovative startups in Silicon Valley. Our diverse 4,000+ team, composed of the world's Top 1% of tech talent, works remotely on roles that drive significant impact worldwide.

When you apply for this position, you're taking the first step in a process that goes beyond the ordinary. We aim to align your passions and skills with our vacancies, setting you on a path to exceptional career development and success. As an ML Engineer (GPU), you will specialize in multi-GPU and multi-node training to scale complex models efficiently.

You will act as the link between machine learning research and high-performance hardware, ensuring that models are optimized for the Nvidia ecosystem and mixed-precision execution.

What You'll Do Design and implement distributed training strategies using PyTorch, DDP, and FSDP to scale model training across multiple nodes.

Optimize model performance and memory usage through mixed-precision training and DeepSpeed integration.

Profile and debug GPU kernels and workloads using Nvidia Nsight to identify and resolve performance bottlenecks.

Develop high-performance inference pipelines leveraging TensorRT or Triton Inference Server for production deployment.





Implement low-level optimizations and maintain familiarity with CUDA and GPU architectures to maximize hardware utilization.

What We Are Looking For 4+ years of experience in Machine Learning Engineering, Software Engineering, or Systems Programming with a focus on GPU acceleration.

Proven expertise in distributed training using PyTorch, specifically DDP, FSDP, and DeepSpeed.

Proficiency in Nvidia GPU profiling with Nsight and model optimization using Triton or TensorRT.

Deep understanding of multi-GPU and multi-node training strategies, including mixed-precision optimization.

Advanced proficiency in English. How we do make your work (and your life) easier: 100% remote work (from anywhere).

Excellent compensation in USD or your local currency if preferred

Hardware and software setup for you to work from home.

Flexible hours: create your own schedule.

Paid parental leaves, vacations, and national holidays.

Innovative and multicultural work environment: collaborate and learn from the global Top 1% of talent.

Supportive environment with mentorship, promotions, skill development, and diverse growth opportunities. Join a general team where your unique talents can truly thrive and make a significant impact! Apply now!

📌 ML Engineer (GPU) - Remote Work (México)
🏢 BairesDev
📍 México