XLA Jobs
1 - 12 of 12
Search Results - XLA Jobs
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 -
Leidos-Dunn Loring
Description
Leidos is seeking a Sr. End User Services Technical SME for our Enterprise Digital Experience (EDX) Practice Area. Preferred location is Reston, VA, but can work hybrid remote.
POSITION OVERVIEW: This role will serve as a technical...
11 similar jobs: Cabin John, West Mclean, Great Falls, Mclean, Vienna...
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
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
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
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-Seattle
mentorship and other career-advancing resources here to help you develop into a better-rounded professional.
Key job responsibilities
This role will help lead the efforts building distributed training and inference support into Pytorch, Tensorflow using XLA...
amazon.jobs -
Amazon-Seattle
into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure highest performance and maximize the efficiency of them running on the customer AWS Trainium and Inferentia silicon and the TRn1...
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 -
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-Seattle
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 -
View XLA jobs in:
Don’t miss out on new job openings!
Create a job alert for: XLA
It's free, and you can cancel email updates at any time