Senior ML Framework Developer (PyTorch)

🔔 Are you already in Poland? If YES this is the project for YOU!

Our benefits:

👩‍⚕️ Private Medical Care in Luxmed and Life Insurance

🏋️‍♀️ Multisport Card

👨‍👧‍👦 Paid referrals

📚 Self-learning libraries

🛫 Relocation package for seniors and assistance during all process...and MORE!

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

Nice-to-Have Skills Description:

Knowledge of ROCm, HIP, Jenkins, PyTorch Internals,