XLA Jobs
1 - 7 of 7
Search Results - XLA Jobs
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-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...
thecareerwallet.us -
Advanced Micro Devices-San Jose
programming background, C/C++ and Python
• ML Compiler and Runtime technologies: OneDNN, MLIR, XLA, OpenXLA, IREE, OpenAI Triton compiler
• Compiler building skills
• Code generation for a ML hardware accelerator
• GPU code generation
• Machine Learning...
appcast.io -
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 -
Amazon-Seattle
Candidates with strong interests and academic qualifications/research focus in: Artificial Intelligence, machine learning, and/or Generative AI, such as computer vision, deep learning models, XLA, TVM, MLIR, and LLVM.
• Experience in optimization mathematics...
amazon.jobs -
Amazon-New York
Generative Artificial Intelligence (GenAI) research experience with LLMs and multimodalities
• For system researchers, familiarity with deep learning compilers, auto-parallelization, and XLA/MLIR ecosystems
Amazon is committed to a diverse and inclusive...
amazon.jobs -
1 similar job: Sunnyvale
Amazon-Seattle
key information so customers can understand and optimize the performance of our custom hardware. Additionally, you will drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, TensorFlow, and XLA.
A successful...
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