Job Summary
We are looking for a Senior C++ / PyTorch AI Framework Engineer to develop and validate software for PyTorch and AI workloads running on hardware accelerators.
The role requires strong systems/software engineering capability, ownership of complex implementation and debugging activities, and contribution to design and technical decisions.
Key Responsibilities
-
Design and implement features for PyTorch and AI framework software.
-
Develop software enabling deep-learning workloads on hardware accelerators.
-
Analyze complex C++ and Python code paths.
-
Perform root-cause analysis of framework and product defects.
-
Implement/debug framework functionality and deep-learning operators.
-
Ensure mathematical correctness and numerical stability.
-
Develop and maintain unit/integration tests.
-
Investigate software and performance regressions.
-
Work with dedicated hardware/test platforms.
-
Identify cross-component dependencies.
-
Contribute to software design and implementation roadmaps.
-
Perform/review code changes.
-
Provide technical guidance to other engineers.
-
Collaborate with globally distributed engineering teams.
Skill Requirements
-
Strong professional C++ software-development experience.
-
Good/strong Python programming.
-
Strong Linux development and debugging.
-
Good understanding of deep-learning frameworks.
-
Excellent software debugging and root-cause analysis.
-
Unit/integration-testing experience.
-
Git/source-control experience.
-
Ability to independently own complex engineering problems.
-
Good written and spoken English.
-
Ability to work onsite in Gdańsk when hardware/laboratory access is required.
Other Requirements
-
Strong hands-on PyTorch/framework internals experience.
-
ATen / tensor operations / custom operator development.
-
GPU or hardware-accelerator software development.
-
CUDA, SYCL, oneAPI, ROCm or OpenCL.
-
Intel GPU / Gaudi / Habana experience.
-
Kernel/operator development.
-
Numerical computing and floating-point precision/stability.
-
Performance analysis/profiling.
-
CMake / complex C++ build systems.
-
Open-source or upstream framework contributions.