XLA Jobs in California
1 - 8 of 8
Search Results - XLA Jobs in California
Acceler8 Talent-San Francisco
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 -
Two95 International Inc.-Los Angeles
on hardware is going to be key.
Experience in GPU programming models, such as OpenCL or CUDA.
Experience with cross platform model compilers like ngraph, XLA, TensorRT, etc.
Systems-level experience with Linux Operating System.
Previous experience leading...
resume-library.com -
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 -
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
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:
Broaden your job search:
Don’t miss out on new job openings!
Create a job alert for: XLA, California
It's free, and you can cancel email updates at any time