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

BairesDev

📍 puebla de zaragoza, puebla, Mexico

Full-time Desarrollo de software

Job Description

At BairesDev® we have over 15 years of experience building cutting‑edge technology solutions for companies worldwide.

Software Engineer (PyTorch)

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...
Apply for this Position