C++/Python AI Framework Engineer

C++ / Python Software Engineer — AI Frameworks (PyTorch)Contract: B2B, 12 months, with extension possible · Work mode: remote, with occasional on-site visits in Gdańsk · About the projectOur 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. What you will work onReproducing and root-causing framework bugs, then fixing them in C++ or PythonFixing and extending tests in the PyTorch test framework and keeping CI greenImplementing operators, from simple ones up to complex, performance-critical pathsDebugging and profiling on accelerator hardware to close performance and accuracy gapsTaking tasks end to end, from first reproduction to a merged upstream pull requestTracking down regressions with git bisect and benchmark suitesTask size grows with experience: junior engineers start with bug reproduction, test fixes and simple operators with guidance. More experienced engineers own larger features independently, including the upstream PR. Must-have skillsPrior hands-on development experience with the Gaudi AI accelerator software stack (Gaudi SW stack / Gaudi PyTorch bridge). Applications without Gaudi development experience cannot be considered. Strong C++ and PythonPython testing with pytest and the PyTorch test framework (OpInfo, common_device_type / instantiate_device_type_tests, common_utils)Experience with PyTorch internals or the internals of another deep learning frameworkGPU/accelerator programming basics (SYCL, CUDA or OpenCL)Debugging and profiling skillsCI with GitHub Actions and JenkinsLinux, Docker, git, including git bisectAn understanding of deep learning and PyTorch fundamentalsFamiliarity with dynamo benchmark suitesWhy joinDeep, hands-on framework work on next-generation AI hardware, not glue codeUpstream open-source contributions you can show in your portfolioA remote-first setup, with on-site visits only when a task needs themA stable 12-month B2B engagementA hiring process designed to move quickly, with a clear answer at each stageHow to applySend your CV and, if you have one, a link to your GitHub or upstream contributions. Please say briefly where and how you have worked with the Gaudi software stack. You will be engaged through ALOIS Solutions, a global IT staffing partner operating in 21+ countries. We handle the contract, invoicing and onboarding, so you can focus on the engineering.