Software Engineer – PyTorch (Remote)

About the role

As a Software Engineer specializing in PyTorch, you will design and build deep‑learning models and robust training pipelines that power advanced AI solutions. You will implement high‑performance neural networks and optimise their execution for production‑grade environments.

Key responsibilities

• Design and implement deep‑learning architectures and training workflows using PyTorch.
• Optimise model performance and training efficiency with CUDA and GPU acceleration.
• Develop and maintain scalable data pipelines for model training, evaluation, and deployment.
• Troubleshoot and refine complex model behaviours to ensure accuracy and reliability.
• Stay current with the PyTorch ecosystem and contribute custom extensions when needed.
• Collaborate with data scientists to translate research into production‑ready software.

Required profile

• 4+ years of experience in software engineering, ML engineering, or data science.
• Proven expertise developing deep‑learning models and training pipelines with PyTorch.
• Strong proficiency in CUDA and GPU‑accelerated model optimisation.
• Hands‑on experience with core deep‑learning architectures and best‑practice training methods.
• Deep understanding of systems programming and PyTorch internals or custom extensions.
• Advanced proficiency in English.

Required skills

• Python
• PyTorch
• CUDA
• GPU acceleration
• Deep learning model development
• Data pipeline engineering

What we offer

• 100% remote work from anywhere.
• Competitive compensation in USD or local currency.
• Hardware and software setup for home office.
• Flexible hours and self‑managed schedule.
• Paid parental leave, vacations, and national holidays.
• Multicultural environment with mentorship, promotion, and skill‑development opportunities.