Software Engineer (PyTorch) - Remote Work

BairesDev; Brasil; Remote; BRL 611,000 - 916,000; Full time

Benefits offered by this job

Remote work (100% from anywhere)

Salary in USD or local currency

Home setup provided

Flexible hours

Parental leave and vacations

Mentorship and growth opportunities

Job summary

BairesDev is offering a fully remote Software Engineer role focused on PyTorch to build and optimize production-grade DL solutions. You will design architectures, implement training workflows, and collaborate with data scientists to translate research into scalable software.

The role requires 4+ years in ML/DS, strong PyTorch experience, and proficiency with CUDA. Enjoy flexible, globally distributed teams and opportunities to work on cutting-edge AI projects.

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.
• Deep understanding of systems programming and PyTorch internals or custom extensions.
• Advanced proficiency in English.

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

GPU acceleration

Deep learning

Data pipelines

English proficiency

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.