Skip to main content

vidaxDocumentation Benchmark Blog Gallery

Recent posts

2026

Sharding, parallelism, and JIT compilation on TPUs, with JAX
Understanding video diffusion architectures: DiT, MoT, and conditioning
Weight offloading: trading bandwidth for memory

Tags

E

Engineering Notes3

I

Infrastructure2

M

Modeling1

Documentation

Getting Started
Model Family Guides
Sharding & Topology
API Reference

Community

GitHub Issues
Discussions
arXiv Paper

More

Benchmark Explorer
Video Gallery
Blog
GitHub Repository

Copyright © 2026 the vidax authors. Built with Docusaurus. Supported by the GoogleTPU Research Cloud (TRC) program.