Software Engineer, ML Efficiency
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
Enable the team to develop large-scale end-to-end driving models with high hardware efficiency, reliable training and inference, and a great developer experience.
Participate in model design from a hardware-efficiency perspective to boost performance and scalability.
Collaborate with internal and partner teams to advance ML infrastructure and tooling for scalable workloads.
Drive system performance improvements through parallelism, buffering/prefetching, pipelining, and asynchronous execution.
Implement and optimize using Python and ML frameworks (PyTorch/JAX/TensorFlow) and accelerator programming (CUDA, Triton).
Candidates should have a BS in Computer Science or related field, strong software engineering skills, and experience with large-scale reliability and C++.
Participate in model design from a hardware-efficiency perspective to boost performance and scalability.
Collaborate with internal and partner teams to advance ML infrastructure and tooling for scalable workloads.
Drive system performance improvements through parallelism, buffering/prefetching, pipelining, and asynchronous execution.
Implement and optimize using Python and ML frameworks (PyTorch/JAX/TensorFlow) and accelerator programming (CUDA, Triton).
Candidates should have a BS in Computer Science or related field, strong software engineering skills, and experience with large-scale reliability and C++.
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.