Senior ML Framework Developer (PyTorch) Opportunity

Luxoft company

Subscribe to our Telegram & Twitter Channel

Senior ML Framework Developer (PyTorch) in POLAND

Visa sponsorship & Relocation 6 months ago

๐Ÿ”” 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,

Apply now

Subscribe our newsletter

New Things Will Always Update Regularly