- Google (Sunnyvale, CA)
- Compiler Engineer , XLA _corporate_fare_ X _place_ Sunnyvale, CA, USA **Minimum qualifications:** + Bachelor's degree or equivalent practical experience. + 5 ... on and is growing every day. As a software engineer , you will work on a specific project critical...continue to push technology forward. Our team builds the XLA compiler which enables TPUs, Google's in-house… more
- Meta (Menlo Park, CA)
- …work is open source, cutting-edge, and industry leading. **Required Skills:** Software Engineer , Systems ML - PyTorch Compiler , PyTorch Framework, PyTorch ... direct engagements with OSS and industry users. The PyTorch Compiler team is dedicated to making PyTorch run faster...Qualifications: 9. Experience in developing PyTorch/PT2, Triton, MLIR, JAX, XLA , TVM is a huge plus 10. Knowledge in… more
- NVIDIA (Santa Clara, CA)
- …make a lasting impact on the world We are looking for a Deep Learning Compiler Engineer . NVIDIA is hiring software engineers for its Deep Learning Compiler ... spanning across data centers, personal devices, automotive, and robotics. The compiler must deliver leading inference performance, fast build time, reduced memory… more
- General Motors (Sunnyvale, CA)
- **Job Description** **Role:** **We are looking for a deep learning compiler engineer to build out our ML compiler for deploying machine learning models to a ... hardware accelerators. You will develop and enhance GM's internal ML compiler for high performance, usability, and retargetability by leveraging open-source… more
- NVIDIA (Santa Clara, CA)
- …and efficiency. You will work hands-on with emerging AI models, collaborating across compiler , AI inference, and model performance teams. The focus is on building ... the kernel level. + Develop, optimize, and evaluate both hand-tuned and compiler -generated kernels for inference workloads, balancing speed and flexibility. + Design… more
- NVIDIA (Santa Clara, CA)
- …+ Hands-on experience with deep learning frameworks like TensorRT-LLM, vLLM, SGLang, Jax/ XLA , or related compiler /runtime environments. + Proven ability to ... We're now seeking a Senior AI Software Engineer , in our LLM Inference Performance Analysis and...to optimize LLM inference performance. Our team collaborates with compiler , kernel, hardware, and framework teams to assess bottlenecks,… more
- Meta (Menlo Park, CA)
- …AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new ... high performance for production environments across specialized hardware architectures. The compiler stack, DL graph optimizations, and kernel authoring for specific… more
- Meta (Menlo Park, CA)
- …open source, cutting-edge, and industry leading. **Required Skills:** ML Framework Software Engineer (PhD) Responsibilities: 1. Develop the PT2 compiler (eg, ... through direct engagements with OSS and industry users.The PyTorch Compiler team is dedicated to making PyTorch run faster...Qualifications: 8. Experience in developing PyTorch/PT2, Triton, MLIR, JAX, XLA , TVM is a huge plus 9. Knowledge in… more
- Google (Mountain View, CA)
- …modern ML framework architectures (JAX, TensorFlow, PyTorch) with the knowledge of ML compiler toolchains (eg, XLA ). + Experience with performance I/O, data ... Senior Staff Software Engineer , Machine Learning Frameworks _corporate_fare_ Google _place_ Mountain...as a key technical leader interfacing with Google-wide stakeholders ( XLA /Compilers, Platform Teams, and major ML consumers) to drive… more
- 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
- Amazon (Cupertino, CA)
- …will help lead the efforts building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will ... Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products...Training team works side by side with chip architects, compiler engineers and runtime engineers to create , build… more
- Google (Mountain View, CA)
- Senior Software Engineer , TPU Performance _corporate_fare_ Google _place_ Mountain View, CA, USA **Advanced** Experience owning outcomes and decision making, solving ... and performance tuning techniques. + Experience with machine learning, compiler optimization, code generation, and runtime systems for GPU...on and is growing every day. As a software engineer , you will work on a specific project critical… more
- Amazon (Cupertino, CA)
- …will help lead the efforts building distributed training support into Pytorch and Jax using XLA and the Neuron compiler and runtime stacks. This role will help ... that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team...training team works side by side with chip architects, compiler engineers and runtime engineers to create , build… more
- Amazon (Cupertino, CA)
- …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 ... accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron,...Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and… more
- Amazon (Cupertino, CA)
- …accelerators. You will work closely with teams across AWS Neuron including compiler , training and inference optimization to optimize frameworks for AWS's accelerator ... a motivation to achieve results. Experience with technologies and tools such as XLA , vLLM or Hugging Face transformers is highly valued. *Utility Computing (UC)* AWS… more
- Meta (Menlo Park, CA)
- …stack with one of the following core focus areas: AI frameworks, compiler stack, high performance kernel development and acceleration onto next generation of ... performance 3. Analyze deep learning networks, develop & implement compiler optimization algorithms 4. Collaborating with AI research scientists...optimizations such as SIMD. Experience with MLIR, LLVM, IREE, XLA , TVM, Halide is a plus 11. OR AI… more