Remote PyTorch Software Engineer — Deep Learning & GPU Optimized

BairesDev; United States; Remote; MXN 1,868,000 - 3,226,000; Full time

Benefits offered by this job

100% remote work

USD or local currency compensation

Hardware and software setup for home

Flexible hours

Paid parental leaves and holidays

Global Top 1% talent community

Job summary

BairesDev is seeking a Software Engineer (PyTorch) to design and implement deep learning architectures and robust training pipelines powering state-of-the-art AI solutions.

You will optimize performance with CUDA, build scalable data pipelines, and collaborate with data scientists to translate research into production-grade software. This role emphasizes remote work and global impact.

Qualifications

• 4+ years in Software Engineering, ML Engineering, or Data Science.
• Proven experience building deep learning models and training pipelines with PyTorch.
• Proficiency in optimizing model performance using CUDA and GPU acceleration.
• Strong English communication and collaboration skills.

Responsibilities

• Design and implement deep learning architectures and training workflows using PyTorch.
• Optimize model performance and training efficiency via 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 expertise

CUDA

Deep learning

English

Tools

PyTorch

CUDA

Job description

BairesDev is seeking a Software Engineer (PyTorch) to design and implement deep learning architectures and robust training pipelines powering state-of-the-art AI solutions.

You will optimize performance with CUDA, build scalable data pipelines, and collaborate with data scientists to translate research into production-grade software. This role emphasizes remote work and global impact.