Job Description
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!