C++/Python AI Framework Engineer

Our client is a global technology leader whose chips run in data centres, laptops and edge devices around the world. Its engineering team in Gdańsk is making PyTorch run correctly and at full speed on the company's own data-centre AI accelerators. Much of the work goes upstream to open source, so your code reaches the global PyTorch community.

• Prior hands-on development experience with the Gaudi AI accelerator software stack (Gaudi SW stack / Gaudi PyTorch bridge)
• Strong C++ and Python skills
• Experience with Python testing using pytest and the PyTorch test framework (OpInfo, common_device_type, instantiate_device_type_tests, common_utils)
• Experience with PyTorch internals or another deep learning framework internals
• Knowledge of GPU/accelerator programming basics (SYCL, CUDA, or OpenCL)
• Debugging and profiling skills
• Experience with CI using GitHub Actions and Jenkins
• Proficiency with Linux, Docker, git, including git bisect
• Understanding of deep learning and PyTorch fundamentals
• Familiarity with dynamo benchmark suites

• Reproduce and root-cause framework bugs, then fix them in C++ or Python
• Fix and extend tests in the PyTorch test framework and maintain CI stability
• Implement operators, from simple to complex performance-critical paths
• Debug and profile on accelerator hardware to close performance and accuracy gaps
• Take tasks end to end, from initial reproduction to merged upstream pull request
• Track down regressions using git bisect and benchmark suites
• Junior engineers start with bug reproduction, test fixes, and simple operators under guidance
• More experienced engineers independently own larger features including upstream PRs

The controller of your personal data is ALOIS SOLUTIONS, Warsaw. Data is processed for recruitment under GDPR provisions. Providing mandatory data is required, additional data is voluntary. Data will be stored until recruitment completion and limitation of claims or up to 24 months with consent. Rights under GDPR apply including access, rectification, erasure, restriction, portability, objection, and complaint to the Personal Data Protection Office (UODO). Consent to data processing can be withdrawn anytime without affecting prior processing.