- 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 (Nashville, TN)
- …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: 10. Experience in developing PyTorch/PT2, Triton, MLIR, JAX, XLA , TVM is a huge plus 11. Knowledge in… more
- Google (Sunnyvale, CA)
- Staff Software Engineer , Compilers, Hardware Diversification _corporate_fare_ Google _place_ Sunnyvale, CA, USA **Advanced** Experience owning outcomes and decision ... C, C++, Java, or Python. + Experience with ML compilers (including MLIR, XLA ) or toolchains **Preferred qualifications:** + Master's degree or PhD in Engineering,… more
- Amazon (Seattle, WA)
- …You'll Drive: * Spearhead distributed inference architecture for PyTorch and JAX using XLA * Engineer breakthrough performance optimizations for AWS Trainium and ... AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be...AI models * Collaborate directly with silicon architects and compiler teams to push the boundaries of AI acceleration… 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
- Amazon (Seattle, WA)
- …efforts building distributed training and inference support into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will ... that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team...Apps 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 (Seattle, WA)
- …will help lead the efforts building distributed inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will ... that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team...The ML Apps team works side by side with compiler engineers and runtime engineers to create, build and… more
- Amazon (Seattle, WA)
- …will help lead efforts in building distributed inference support into PyTorch using XLA and the Neuron compiler and runtime stacks. Strong software development ... learning accelerators. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team...Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build… more
- Amazon (Seattle, WA)
- …will help lead the efforts building distributed inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will ... use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team...Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build… more
- Amazon (Seattle, WA)
- …efforts building distributed training and inference support into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will ... use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team...Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build… more
- Amazon (Seattle, WA)
- …and the Trn1 and Inf1 servers that use them. As the Software Development Engineer for the Neuron Foundation Tools Team, you will be responsible for working alongside ... with cross-functional teams to ensure that the our C++ compiler and runtime generates key information so customers can...to support multiple frameworks, such as PyTorch, JAX, and XLA . A successful candidate will have an established background… more