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.
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.
Already have an account? Log in
Similar openings
Other roles that could suit you.