XLA Jobs
1 - 15 of 15
Search Results - XLA Jobs
Selby Jennings-New York
TensorFlow, etc.
• Thorough knowledge of computer architecture.
• Proficiency in programming with C++ and Python.
Preferred Qualifications:
• Experience with the JAX ecosystem, including XLA, Flax, etc.
• Proficiency in programming for GPUs or other...
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 -
BrainChip-Laguna Hills
Expertise: Strong background in compiler internals (e.g., MLIR, LLVM, XLA, or similar) and code optimization strategies. Familiarity with deep learning frameworks (e.g., PyTorch, TensorFlow) and related libraries. Understanding of quantization, pruning...
mindmatch.ai -
Tesla-Palo Alto (CA)
community to keep the stack compatible with latest developments Proficient C/C++ programming (C++14/17/20)Basic Python proficiencyPrior experience working with ML compilers (e.g. (e.g. MLIR, XLA, TensorRT)Experience with ML frameworks and ML architecture...
tesla.com -
Amazon-Cupertino
with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM
• Familiarity with CUDA programming for GPU acceleration
Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race...
amazon.jobs -
Amazon-Seattle
/TensorFlow), compilers (XLA, MLIR), or hardware accelerators
• Technical leadership through open-source contributions, research collaborations, or mentoring experience
Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity...
amazon.jobs -
Amazon-Cupertino
optimizing Tensorflow, PyTorch or JAX deep learning models
• Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM
• Knowledge of CUDA programming for GPU acceleration
Amazon is committed to a diverse and inclusive workplace. Amazon...
amazon.jobs -
Amazon-Seattle
the efforts building distributed inference 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 the efficiency of them running on the customer...
amazon.jobs -
Amazon-Cupertino
with toolchains (LLVM, GCC) and code generation techniques for new 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...
amazon.jobs -
Amazon-Seattle
distributed training support into Pytorch and 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...
amazon.jobs -
Amazon-Cupertino
community direction, a robust technical ability and a motivation to achieve results. Experience with technologies and tools such as XLA, vLLM or Hugging Face transformers is highly valued.
*Utility Computing (UC)*
AWS Utility Computing (UC) provides product...
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-Seattle
the efforts building distributed inference 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 the efficiency of them running on the customer...
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
Tensorflow 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 , Inf1 servers...
amazon.jobs -
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