JAX/OpenXLA DevLab 2025 - PyTorch/XLA Presentation

Click a thumbnail to watch in a lightweight modal. (No downloads — view only.)

• ▶
  JAX/OpenXLA DevLab 2025 - PyTorch/XLA Presentation

  JAX/OpenXLA DevLab 2025 PyTorch/XLA Presentation
• ▶
  PyTorch on TPUs: Native TPU acceleration for PyTorch | JAX/OpenXLA DevLab Fall 2025

  PyTorch on TPUs: Native TPU acceleration for PyTorch JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX/OpenXLA DevLab 2025 - XLA Architecture

  JAX/OpenXLA DevLab 2025 XLA Architecture
• ▶
  JAX DevLab 2025 - Serving JAX Models

  JAX DevLab 2025 Serving JAX Models
• ▶
  TorchAX at Lightricks — When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025

  TorchAX at Lightricks — When PyTorch Meets JAX JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX/OpenXLA DevLab 2025 - Performance Debugging Deep Dive with XProf

  JAX/OpenXLA DevLab 2025 Performance Debugging Deep Dive with XProf
• ▶
  JAX on GPUs | JAX/OpenXLA DevLab Fall 2025

  JAX on GPUs JAX/OpenXLA DevLab Fall 2025
• ▶
  Pallas TPU: New and Advanced Features for Kernels | JAX/OpenXLA DevLab Fall 2025

  Pallas TPU: New and Advanced Features for Kernels JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX/OpenXLA DevLab 2025 - Operator Fusion

  JAX/OpenXLA DevLab 2025 Operator Fusion
• ▶
  Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025

  Models in JAX NNX by Hand JAX/OpenXLA DevLab Fall 2025
• ▶
  Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025

  Contributing to the JAX Ecosystem JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX/OpenXLA DevLab 2025 - Large scale training techniques and best practices

  JAX/OpenXLA DevLab 2025 Large scale training techniques and best practices
• ▶
  Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025

  Keras: Deep Learning Framework for JAX JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX DevLab 2025 - Debugging JAX models

  JAX DevLab 2025 Debugging JAX models
• ▶
  JAX/OpenXLA DevLab 2025 Keynote Speech with Robert Hundt

  JAX/OpenXLA DevLab 2025 - PyTorch/XLA Presentation
• ▶
  Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025

  Asynchronous PjRT for JAX on GPU JAX/OpenXLA DevLab Fall 2025
• ▶
  vLLM TPU: A new unified backend for JAX and PyTorch inference on TPU | JAX/OpenXLA DevLab Fall 2025

  vLLM TPU: A new unified backend for JAX and PyTorch inference on TPU JAX/OpenXLA DevLab Fall 2025
• ▶
  Marin: An Open Laboratory for Foundation Models in JAX | JAX/OpenXLA DevLab Fall 2025

  Marin: An Open Laboratory for Foundation Models in JAX JAX/OpenXLA DevLab Fall 2025
• ▶
  JAX/OpenXLA DevLab - PJRT: '25 Update and overview of Extensions

  JAX/OpenXLA DevLab PJRT: '25 Update and overview of Extensions
• ▶
  Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025

  Leveraging CuTeDSL in JAX JAX/OpenXLA DevLab Fall 2025