Filters
Create alert
Sort by
  • Relevance
  • Date
Company
  • Amazon
  • AMD
Work hours
  • Full-time

XLA Jobs

1 - 7 of 7
1 - 7 of 7
Search Results - XLA Jobs
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 -
Hugging Face-United States
preferably PyTorch, optionally XLA understanding  •  Strong knowledge of cloud platforms like AWS and services like Amazon SageMaker, EC2, S3, CloudWatch and/or Azure and GCP equivalents.  •  Experience in building MLOps pipelines for containerizing models...
workable.com -
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 -
Amazon-Cupertino
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 -
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
Get new jobs by email!
Get email updates for the latest XLA jobs
It's free, and you can cancel email updates at any time