Skip to content

Add more PDF and video resources #272

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
melissawm wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Add more PDF and video resources #272

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
941f224
Add Megablocks to ToC
melissawm Nov 14, 2025
b765028
Add Megablocks slides and video
melissawm Nov 14, 2025
e6477b4
Move PDFs to _static
melissawm Nov 14, 2025
11405b4
Move learning resources
melissawm Dec 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file added docs/source/_static/resources/jax_and_tpus_casml.pdf
View file
Open in desktop
Binary file not shown.
Binary file added docs/source/_static/resources/megablocks_jax.pdf
View file
Open in desktop
Binary file not shown.
2 changes: 1 addition & 1 deletion docs/source/index.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ JAX AI Stack

install
getting_started
more_learning

.. toctree::
:hidden:
Expand Down Expand Up @@ -52,3 +51,4 @@ JAX AI Stack
news
events
videos
more_learning
14 changes: 13 additions & 1 deletion docs/source/more_learning.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,15 @@
# More Learning Resources
# Learning resources

## Intro to JAX & TPUs

Learn more about TPUs, how they work, and how to use JAX and Pallas to use TPUs effectively in this [Intro to JAX & TPUs slide deck](_static/resources/jax_and_tpus_casml.pdf) and [video](https://rsvp.withgoogle.com/events/intro-to-jax-and-google-tpu-for-casml-attendees/sessions/inro-to-jax-and-google-tpu).

## Megablocks

[Megablocks](https://arxiv.org/pdf/2211.15841) is a library for building large-scale machine learning models with JAX.

Check out [this presentation](_static/resources/megablocks_jax.pdf) and watch [this video](https://www.youtube.com/watch?v=tWkMj6lUp1c) for an overview of Megablocks and its capabilities, including dealing with sparsity and *ragged dot* operations.

## More learning resources

In https://goo.gle/learning-jax, you can find a curated list of learning resources about JAX and the JAX AI Stack, including slide decks, coding exercises and quick references.
2 changes: 2 additions & 0 deletions docs/source/videos.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,5 @@
A collection of videos related to the JAX AI Stack, including talks, tutorials, and presentations.

- [JAX DevLab 2025 Playlist](https://www.youtube.com/playlist?list=PLlFotmaRrOztzhilpCT4vBsa0R9hz5ADD)
- Learn more about TPUs, how they work, and how to use JAX and Pallas to use TPUs effectively in this [video](https://rsvp.withgoogle.com/events/intro-to-jax-and-google-tpu-for-casml-attendees/sessions/inro-to-jax-and-google-tpu).
- Watch [this video](https://www.youtube.com/watch?v=tWkMj6lUp1c) for an overview of [Megablocks](https://arxiv.org/pdf/2211.15841), a library for building large-scale machine learning models with JAX.