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

BairesDev; Cajamarca; Remote; PEN 412,000 - 618,000; Full time

Benefits offered by this job

Remote work

USD or local currency compensation

Hardware provided for home setup

Flexible hours

Paid parental leaves & vacations

Mentorship and growth opportunities

Job summary

BairesDev is seeking a Software Engineer (PyTorch) to design and implement deep learning models and training pipelines for production-grade AI solutions. This fully remote role offers the chance to work with a global Top 1% team and impactful projects across diverse domains.

You will leverage PyTorch, CUDA, and GPU acceleration to optimize performance, collaborate with data scientists, and translate research into robust software that scales.

Qualifications

• 4+ years of experience in Software/ML engineering.
• Proven expertise with PyTorch in model development and training.
• Proficiency in CUDA and GPU acceleration.
• Strong English skills.

Responsibilities

• Design and implement deep learning architectures and training workflows with PyTorch.
• Optimize model performance and training efficiency using CUDA and GPUs.
• Build scalable data pipelines for training, evaluation, and deployment.
• Collaborate with data scientists to productionize research.

Skills

PyTorch

CUDA

Deep learning

GPU acceleration

Training pipelines

English

English proficiency

Tools

PyTorch internals

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!