XLA Jobs
1 - 10 of 10
Search Results - XLA Jobs
TekSynap-Dahlgren (VA)
Responsibilities & Qualifications The TekSynap is seeking a Program Manager to l ead all aspects of contract execution, serving as the single accountable authority for performance, compliance, customer engagement, and delivery of IT services.
The PM...
Read more
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...
Read more
resume-library.com -
Prima Mente-San Francisco
training, optimizing and deploying large‑scale models (7B+ parameters).
• Low level algorithm optimisation
• Quantization (8bit or lower)
• JIT compilation
• XLA/Mosaic/Triton/CUDA
• Hardware optimisation (GPU/TPU/HPU)
• Finetuning Optimization (QLora...
Read more
appcast.io -
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...
Read more
amazon.jobs -
Toyota Research Institute-Los Altos
and development grounded in important fundamental problems.
• Continuous integration
Bonus Qualifications
• Familiarity with modern ML efficiency frameworks (e.g., FSDP, DeepSpeed, XLA, Ray, Hugging Face Accelerate).
• Experience with machine learning...
Read more
appcast.io -
Amazon-Cupertino
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 .
Strong software...
Read more
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...
Read more
workable.com -
Amazon-Seattle
frameworks, such as PyTorch, JAX, and XLA.
A successful candidate will have an established background in building AI/ML and performance analysis tools. Experience with ML-specific profiler tools (like PyTorch Profiler or TensorFlow Profiler) is highly...
Read more
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...
Read more
amazon.jobs -
Amazon-Seattle
hardware. This isn't just about just optimization—it's about revolutionizing how AI models run at scale.
Technical Impact You'll Drive:
• Spearhead distributed inference architecture for PyTorch and JAX using XLA
• Engineer breakthrough performance...
Read more
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