OpenXLA

1.85K subscribers

140 videos

View on YouTube

Latest Videos

JAX-GCM A differentiable atmospheric model JAXOpenXLA DevLab Fall 2025

JAX-GCM A differentiable atmospheric model JAXOpenXLA DevLab Fall 2025

OpenXLA

196 views·4 months ago

SparseCore Offloading JAXOpenXLA DevLab Fall 2025

SparseCore Offloading JAXOpenXLA DevLab Fall 2025

OpenXLA

448 views·4 months ago

PyTorch on TPUs Native TPU acceleration for PyTorch JAXOpenXLA DevLab Fall 2025

PyTorch on TPUs Native TPU acceleration for PyTorch JAXOpenXLA DevLab Fall 2025

OpenXLA

2.4K views·4 months ago

Pallas TPU Compilation for TensorCore & SparseCore JAXOpenXLA DevLab Fall 2025

Pallas TPU Compilation for TensorCore & SparseCore JAXOpenXLA DevLab Fall 2025

OpenXLA

867 views·4 months ago

Model Explorer Visualization and Debugging for Large Model Graphs JAXOpenXLA DevLab Fall 2025

Model Explorer Visualization and Debugging for Large Model Graphs JAXOpenXLA DevLab Fall 2025

OpenXLA

273 views·4 months ago

HLO Sharding V3 Mesh & Axis based sharding representation JAXOpenXLA DevLab Fall 2025

HLO Sharding V3 Mesh & Axis based sharding representation JAXOpenXLA DevLab Fall 2025

OpenXLA

224 views·4 months ago

Asynchronous PjRT for JAX on GPU JAXOpenXLA DevLab Fall 2025

Asynchronous PjRT for JAX on GPU JAXOpenXLA DevLab Fall 2025

OpenXLA

232 views·4 months ago

JAX on GPUs JAXOpenXLA DevLab Fall 2025

JAX on GPUs JAXOpenXLA DevLab Fall 2025

OpenXLA

299 views·4 months ago

Keras Deep Learning Framework for JAX JAXOpenXLA DevLab Fall 2025

Keras Deep Learning Framework for JAX JAXOpenXLA DevLab Fall 2025

OpenXLA

704 views·4 months ago

AMD's JAX Ecosystem JAXOpenXLA DevLab Fall 2025

AMD's JAX Ecosystem JAXOpenXLA DevLab Fall 2025

OpenXLA

242 views·4 months ago

ML Performance Debugging with XProf JAXOpenXLA DevLab Fall 2025

ML Performance Debugging with XProf JAXOpenXLA DevLab Fall 2025

OpenXLA

300 views·4 months ago

Leveraging CuTeDSL in JAX JAXOpenXLA DevLab Fall 2025

Leveraging CuTeDSL in JAX JAXOpenXLA DevLab Fall 2025

OpenXLA

340 views·4 months ago

JaxPP A library for MPMD training in JAX JAXOpenXLA DevLab Fall 2025

JaxPP A library for MPMD training in JAX JAXOpenXLA DevLab Fall 2025

OpenXLA

197 views·4 months ago

Tokamax A library of high-performance custom kernels JAXOpenXLA DevLab Fall 2025

Tokamax A library of high-performance custom kernels JAXOpenXLA DevLab Fall 2025

OpenXLA

173 views·4 months ago

Pallas TPU New and Advanced Features for Kernels JAXOpenXLA DevLab Fall 2025

Pallas TPU New and Advanced Features for Kernels JAXOpenXLA DevLab Fall 2025

OpenXLA

1.2K views·4 months ago

Decoupling MaxText for Training on non-GCE Environments JAXOpenXLA DevLab Fall 2025

Decoupling MaxText for Training on non-GCE Environments JAXOpenXLA DevLab Fall 2025

OpenXLA

87 views·4 months ago

TorchAX at Lightricks — When PyTorch Meets JAX JAXOpenXLA DevLab Fall 2025

TorchAX at Lightricks — When PyTorch Meets JAX JAXOpenXLA DevLab Fall 2025

OpenXLA

161 views·4 months ago

JAX AI Stack updates Orbax, Chex, Optax, and more JAXOpenXLA DevLab Fall 2025

JAX AI Stack updates Orbax, Chex, Optax, and more JAXOpenXLA DevLab Fall 2025

OpenXLA

263 views·4 months ago

RL at Google Scale, with MaxText 2.0 JAXOpenXLA DevLab Fall 2025

RL at Google Scale, with MaxText 2.0 JAXOpenXLA DevLab Fall 2025

OpenXLA

398 views·4 months ago

Sharding the Sphere with jax.shard_map JAXOpenXLA DevLab Fall 2025

Sharding the Sphere with jax.shard_map JAXOpenXLA DevLab Fall 2025

OpenXLA

148 views·4 months ago

Contributing to the JAX Ecosystem JAXOpenXLA DevLab Fall 2025

Contributing to the JAX Ecosystem JAXOpenXLA DevLab Fall 2025

OpenXLA

175 views·5 months ago

Accelerating LLM models on Blackwell JAXOpenXLA DevLab Fall 2025

Accelerating LLM models on Blackwell JAXOpenXLA DevLab Fall 2025

OpenXLA

218 views·5 months ago

JAXformer Focusing on scaling MoE pretraining on TPUs JAXOpenXLA DevLab Fall 2025

JAXformer Focusing on scaling MoE pretraining on TPUs JAXOpenXLA DevLab Fall 2025

OpenXLA

828 views·5 months ago

Simplifying NNX New features and updates JAXOpenXLA DevLab Fall 2025

Simplifying NNX New features and updates JAXOpenXLA DevLab Fall 2025

OpenXLA

199 views·5 months ago

Models in JAX & NNX by Hand JAXOpenXLA DevLab Fall 2025

Models in JAX & NNX by Hand JAXOpenXLA DevLab Fall 2025

OpenXLA

322 views·5 months ago

TPU Kernels Ragged Paged Attention v3 and Fused MoE JAXOpenXLA DevLab Fall 2025

TPU Kernels Ragged Paged Attention v3 and Fused MoE JAXOpenXLA DevLab Fall 2025

OpenXLA

640 views·5 months ago

vLLM TPU A new unified backend for JAX and PyTorch inference on TPU JAXOpenXLA DevLab Fall 2025

vLLM TPU A new unified backend for JAX and PyTorch inference on TPU JAXOpenXLA DevLab Fall 2025

OpenXLA

537 views·5 months ago

Marin An Open Laboratory for Foundation Models in JAX JAXOpenXLA DevLab Fall 2025

Marin An Open Laboratory for Foundation Models in JAX JAXOpenXLA DevLab Fall 2025

OpenXLA

146 views·5 months ago

Getting Started with JIMM for scaling CLIPSigLIP training JAXOpenXLA DevLab Fall 2025

Getting Started with JIMM for scaling CLIPSigLIP training JAXOpenXLA DevLab Fall 2025

OpenXLA

645 views·5 months ago

Grain Training data processing, tuning, and performance JAXOpenXLA DevLab Fall 2025

Grain Training data processing, tuning, and performance JAXOpenXLA DevLab Fall 2025

OpenXLA

307 views·5 months ago

Load More Videos

OpenXLA - NatokHD | NatokHD