Luxoft company
๐ 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,