Software Engineer (PyTorch) - Remote Work | REF#303309

BairesDev; United States; Remote; MXN 1,528,000 - 2,207,000; Full time

Benefits offered by this job

Remote work (100% from anywhere)

Excellent compensation in USD

Hardware and software setup for home

Flexible hours

Paid parental leaves, vacations, and H

Job summary

BairesDev is expanding a global, 100% remote software engineering team. We seek a PyTorch-focused Software Engineer to build and optimize deep learning models, develop robust training pipelines, and deploy production-grade AI solutions at scale.

Join a world-class, remote-first company that collaborates with top tech giants and startups, offering flexible hours, strong compensation in USD, and a supportive environment for career growth across borders.

Qualifications

• 4+ years of experience in Software Engineering, ML Engineering, or Data Science.
• Proven expertise in developing deep learning models and training pipelines using PyTorch.
• Proficiency in optimizing model performance using CUDA and GPU acceleration.
• Hands-on experience with core deep learning architectures and model training best practices.

Responsibilities

• Design and implement deep learning architectures and training workflows using PyTorch.
• Optimize model performance and training efficiency by leveraging CUDA and GPU acceleration.
• Develop and maintain scalable data pipelines for model training, evaluation, and deployment.
• Troubleshoot and refine complex model behaviors to ensure accuracy and reliability.
• Stay current with the PyTorch ecosystem to implement custom extensions or contribute to framework internals.
• Collaborate with data scientists to translate theoretical research into production-ready software.

Skills

PyTorch

CUDA

English proficiency

Tools

GPU acceleration

Model deployment

Job description

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.

Software Engineer (PyTorch) at BairesDev

As a Software Engineer specializing in PyTorch, you will develop deep learning models and robust training pipelines to power advanced AI solutions. You will be responsible for implementing high-performance neural networks and optimizing their execution for production-grade environments.

What You'll Do

• Design and implement deep learning architectures and training workflows using PyTorch.
• Optimize model performance and training efficiency by leveraging CUDA and GPU acceleration.
• Develop and maintain scalable data pipelines for model training, evaluation, and deployment.
• Troubleshoot and refine complex model behaviors to ensure accuracy and reliability.
• Stay current with the PyTorch ecosystem to implement custom extensions or contribute to framework internals.
• Collaborate with data scientists to translate theoretical research into production-ready software.

What We Are Looking For

• 4+ years of experience in Software Engineering, ML Engineering, or Data Science.
• Proven expertise in developing deep learning models and training pipelines using PyTorch.
• Proficiency in optimizing model performance using CUDA and GPU acceleration.
• Hands-on experience with core deep learning architectures and model training best practices.
• Deep understanding of systems programming and PyTorch internals or custom extensions.
• 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 global team where your unique talents can truly thrive and make a significant impact!