- 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
- 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