Skip to content

Pull requests: jax-ml/jax-ai-stack

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Bump orbax-checkpoint from 0.11.34 to 0.11.36 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#319 opened Apr 20, 2026 by dependabot Bot Loading…
Add the jax Stack TR
#266 opened Nov 11, 2025 by ev-br Loading…
[ci] add python 3.14 coverage
#250 opened Oct 22, 2025 by jakevdp Collaborator Loading…
Adding Learning JAX
#233 opened Sep 15, 2025 by rcrowe-google Contributor Loading…
Update tutorial: Porting PyTorch model to JAX
#220 opened Jun 25, 2025 by pavithraes Contributor Loading…
[WIP] Add "The stack" section to left nav
#212 opened May 27, 2025 by skye Contributor Draft
Upgrade JAX AI Stack Data Loaders on CPU doc
#205 opened May 8, 2025 by 8bitmp3 Contributor Draft
Upgrade ViT image segmentation tutorial for JAX AI Stack
#194 opened Apr 14, 2025 by 8bitmp3 Contributor Loading…
CI: add testing on Python 3.13
#192 opened Apr 9, 2025 by jakevdp Collaborator Loading…
Upgrade Machine Translation JAX AI Stack doc
#185 opened Mar 17, 2025 by 8bitmp3 Contributor Loading…
Add tutorial to port Llama 3.2 1B model to JAX
#167 opened Feb 12, 2025 by windmaple Collaborator Loading…
Update tutorial: Visualize JAX model
#157 opened Jan 27, 2025 by pavithraes Contributor Loading…
Update JAX AI for PyTorch users tutorial
#135 opened Dec 20, 2024 by 8bitmp3 Contributor Draft
Adding tutorial for data loaders in multi device setups
#110 opened Nov 21, 2024 by selamw1 Contributor Loading…
ProTip! Add no:assignee to see everything that’s not assigned.