C++ Machine Learning Engineer, Models Training
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
Track your applications on mobile The free Whileresume app, on iPhone and Android.
The role
Extend and optimize the ML training framework with new ops, layers, and training features.
Write high-performance C++ code focused on tensors, memory layout, and compiler graphs.
Debug and tune model performance on the accelerator and collaborate with the compiler and kernel teams to ensure correct compilation and execution.
Support integration of real-world models and help bring them into production.
Leverage PyTorch knowledge to bridge training workloads to hardware and translate workloads into efficient low-level operations.
This role is hybrid and based in Poland (Warsaw or Gdansk) with partial remote options and case-by-case consideration for remote candidates.
Write high-performance C++ code focused on tensors, memory layout, and compiler graphs.
Debug and tune model performance on the accelerator and collaborate with the compiler and kernel teams to ensure correct compilation and execution.
Support integration of real-world models and help bring them into production.
Leverage PyTorch knowledge to bridge training workloads to hardware and translate workloads into efficient low-level operations.
This role is hybrid and based in Poland (Warsaw or Gdansk) with partial remote options and case-by-case consideration for remote candidates.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inYou might also like these jobs
No closely matching jobs yet — here are the most recent ones.