- NVIDIA (Santa Clara, CA)
- …across multi-GPU, multi-node, and multi-cloud environments. You'll collaborate across inference, compiler , scheduling, and performance teams to push the frontier of ... + Develop, optimize, and benchmark GPU kernels (hand-tuned and compiler -generated) using techniques such as fusion, autotuning, and memory/layout optimization;… more
- quadric.io, Inc (Burlingame, CA)
- …role involves managing the testing strategy and execution for the Chimera Graph compiler , LLVM backend, and various kernel operators. You will be responsible for ... strategy for the AI software stack, covering the entire end-to-end pipeline from compiler to hardware execution. + Establish and drive a robust testing framework… more
- Amazon (Cupertino, CA)
- …the Neuron Core Tools team * Work closely with the frameworks and compiler teams. * Collect requirements from various other teams including training, inference and ... runtime. * Collaborate with the compiler performance team to ensure timely release of the...AWS Neuron. Neuron is a Software that include ML compiler and native integration into popular ML frameworks. Our… more
- Amazon (Cupertino, CA)
- …The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training ... build distributed training support into PyTorch and JAX using XLA, the Neuron compiler , and runtime stacks. You will optimize models to achieve peak performance and… more
- IBM (Cambridge, MA)
- …capabilities to enable advantage-scale experiments * Developing and studying compiler optimizations for quantum circuits targeting fault tolerant architectures ... experience * Experience with quantum algorithm design and analysis, quantum compiler or computer architecture, experimental realization of quantum algorithms, or… more
- Google (Sunnyvale, CA)
- …algorithms in either an academic or industry setting. + Experience with compiler improvement, code generation and runtime systems for GPU architectures (OpenXLA, ... across the breadth and depth of Google's GPU software stack from ML compiler cost model design (OpenXLA, Triton, MLIR) to optimizing performance GPU kernels (Pallas… more
- NVIDIA (Santa Clara, CA)
- …systems for the CUDA ecosystem. We build innovative agentic runtimes and compiler -integrated orchestration that work together with NVIDIA's software stack to provide ... team, you will develop new agent abstractions, GPU-centric runtimes, and compiler - or runtime-driven system solutions to accelerate agent planning, tool-use, code… more
- Microsoft Corporation (Redmond, WA)
- …for latency, bandwidth, and throughput. We work hand-in-hand with hardware teams, compiler and model teams, and observability partners to ensure every byte moved ... primitives across devices and execution units + Collaborate with hardware, firmware, and compiler teams to define and refine the runtime contract between layers. +… more
- pony.ai (Fremont, CA)
- …and efficiency in autonomous driving. + Work across the entire AI framework/ compiler stack (eg Torch, CUDA and TensorRT), support model development and prototype ... frameworks and libraries. + Strong knowledge on system performance, GPU optimization or ML compiler . Note + This position is fully onsite in Fremont, at least 3… more
- Google (Sunnyvale, CA)
- …and software test engineering. **About the job** The XLA (Accelerated Linear Algebra) compiler is used across the Research to Production pipeline for both Training ... different hardware, frameworks and use cases by simplifying and standardizing compiler integration across different stacks.The AI and Infrastructure team is… more