XLA Jobs in Santa Clara
Search jobs in
1 - 9 of 9
Search Results - XLA Jobs in Santa Clara
NVIDIA CorporationSanta Claraappcast.io
compiler frameworks (TVM, MLIR, XLA, Triton) or contributing to inference runtime development.
• Production deployment experience with autonomous vehicle perception or planning stacks, understanding the full pipeline from sensor input through trajectory...
NVIDIA CorporationSanta Claraappcast.io
Gemma and Phi and an understanding of optimization techniques for accelerating training and inference workloads.
• Frameworks ecosystem knowledge - Experience with compilers such as OAI Triton, XLA, MLIR, and frameworks like PyTorch, JAX, vLLM, sglang...
NVIDIA CorporationSanta Claraappcast.io
Framework Development: Hands-on experience developing within Megatron-Core, DeepSpeed, or JAX/XLA, with an understanding of how these frameworks interact with low-level communication runtimes is a plus.
Significant upstream contributions to major open...
NVIDIA CorporationSanta Claraappcast.io
arithmetic (e.g., NVFP4, MXFP4, FP8, INT8) and their impact on deep learning accuracy and performance.
• Background in deep learning compilers and ML systems, including graph-level and codegen tools (e.g., Triton, XLA, torch.compile) and highly parallel/RL...
NVIDIA CorporationSanta Claraappcast.io
clearly through documents, presentations, and in external customer-facing environments.
Ways to stand out from the crowd:
• Hands‑on experience with Deep Learning frameworks (PyTorch, JAX, etc.), compilers (Triton, XLA, etc.), and NVIDIA libraries...
NVIDIA CorporationSanta Claraappcast.io
or Tier-1 engagement.
• Background in training pipeline optimization, curriculum design, or end-to-end autonomous driving architectures.
• Experience with ML compiler frameworks (TVM, MLIR, XLA, Triton) or inference runtime development.
• Familiarity...
Google LLCMountain View (CA), 8 mi from Santa Claraappcast.io
level compiler intermediate representations (e.g., MLIR dialects, XLA, HLO), ML execution frameworks (JAX, PyTorch, PyTorch/XLA), or accelerator kernel programming (Pallas, Triton, CUDA).
• Demonstrated track record of architecting, validating...
Xoriant CorporationSan Jose, 4 mi from Santa Clara
PrivateUse1 backend development, OpenReg-style reference implementations, or out-of-tree backend maintenance.
• Experience with torch.compile (TorchDynamo, TorchInductor), MLIR/LLVM, Triton, or XLA, or optimizing large-scale LLM training (e.g. VeRL...
Advanced Micro Devices, Inc.San Jose, 4 mi from Santa Claraappcast.io
computing environments.
• Familiarity with technologies such as ROCm, HIP, CUDA, Triton, XLA, MLIR, NCCL, or similar performance-oriented tools and runtimes.
• Experience with distributed training, model serving, quantization, compilation, kernel...
XLA jobs – More cities:
Broaden your job search:
Don’t miss out on new job openings!
Create a job alert for: XLA, Santa Clara
It's free, and you can cancel email updates at any time