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.
The role
Extend and optimize the ML training framework by implementing new ops, layers, and training features with a focus on performance.
Debug and tune model performance on our custom silicon, working closely with hardware teams.
Collaborate with compiler and kernel teams to ensure models compile and run as expected.
Support integration of real-world models and help bring them into production.
Work on memory layout and tensor concepts to map frameworks efficiently to hardware for scalable training.
Learn how ML frameworks connect to systems, translating workloads into low-level operations optimized for fast iteration.
Debug and tune model performance on our custom silicon, working closely with hardware teams.
Collaborate with compiler and kernel teams to ensure models compile and run as expected.
Support integration of real-world models and help bring them into production.
Work on memory layout and tensor concepts to map frameworks efficiently to hardware for scalable training.
Learn how ML frameworks connect to systems, translating workloads into low-level operations optimized for fast iteration.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.