XLA Jobs in Cupertino
1 - 7 of 7
Search Results - XLA Jobs in Cupertino
Amazon-Cupertino
responsibilities
This role will help lead the efforts building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure highest performance and maximize...
amazon.jobs -
Acceler8 Talent-San Francisco, 38 mi from Cupertino
in working under-the-hood with kernel languages like OAI Triton, Pallas, and compilers like XLA.
• Proven hands-on experience in tuning LLM workloads. Familiarity with MLPerf or production workloads is a plus.
If youre passionate about driving AI innovation...
mindmatch.ai -
Amazon-Cupertino
to this and extending all of this for the Neuron based system is key.
Key job responsibilities
You will lead efforts to build distributed training support into PyTorch and JAX using XLA, the Neuron compiler, and runtime stacks. You will optimize models to achieve peak...
amazon.jobs -
Amazon-Cupertino
Tensorflow using XLA and the Neuron compiler and runtime stacks. You will help tune these models to ensure highest performance and maximize the efficiency of them running on the custom AWS Trainium and Inferentia silicon and the Trn1, Inf1/2 servers.
Strong...
amazon.jobs -
Amazon-Cupertino
hardware
• Experience with XLA, TVM, MLIR, LLVM, deep learning models and algorithms, and deep learning framework design.
• Interactions with open-source communities, in either a leadership or code contributor role
Amazon is committed to a diverse...
amazon.jobs -
Amazon-Cupertino
and training support in Pytorch, XLA, JAX as well as distributed training libraries like FSDP, DDP and others. Includes enabling models using MoE architectures and future newer architectures.
Lead the way to ensure support for key ML functionality...
amazon.jobs -
Amazon-Sunnyvale, 5 mi from Cupertino
frameworks such as TVM, PyTorch, XLA, MLIR
Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation...
amazon.jobs -
XLA jobs – More cities:
Don’t miss out on new job openings!
Create a job alert for: XLA, Cupertino
It's free, and you can cancel email updates at any time