Senior ML Framework Developer (PyTorch)

Project Description:

Development of modern tools for working with AI / ML on the GPU. Implementation of algorithms for working with graphic processors. Working with GPU neural network calculations. Work with the AI/ML development team and the open-source community to analyze, develop, test, and deploy performance improvements for neural networks on GPUs with AMD ROCm, working on enabling ROCm HIPSOLVER/cuSOLVER in PyTorch

Responsibilities:

Develop, support and optimize PyTorch project for GPU. Communication with developers, customers, and project managers. Create, document, and execute unit/component/functional tests. This might involve one or more of the following:

● Modifying PyTorch UT code to adapt it for ROCm

● Filing JIRAs on ROCm components that have missing features needed for the UTs

Mandatory Skills Description:

Proficiency with PyTorch, C/C++ and Python.

• Knowledge of GPU computing (HIP or CUDA or OpenCL)

• Experience in CMake, make/ninja build system, Docker

• Software Version Control (Git)

• Knowledge in ML/NN

The successful candidate should possess the following traits:

● Proficiency in problem-solving and troubleshooting technical issues

● Time management, prioritization and multi-tasking skills

● Ability to handle multiple competing priorities in a fast-paced environment

● Personal initiative, commitment, perseverance and resilience

● Well-developed communication and teamwork skills

● Aspiration to engineering excellence

What do we offer our employees?

💰Copyrights which means higher net salary for Developers

🏠 Mode of work: remote

🏠 Fully remote recruitment process

👩‍⚕️ LuxMed health & dental care

🏦 Life insurance, travel insurance

🏋🏽‍♀️ MyBenefit program (sports card, well-being program etc.)

🖋 Stable employment based on an employment contract

🎓 LuxTalent platform (webinars, training, courses with certificates)

🌎 Internal Mobility program - possibility of rotation between projects, locations, accounts

...and even more!