Skip to content

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

New pull request New
21 Open 294 Closed
21 Open 294 Closed
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
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Bump orbax-checkpoint from 0.12.0 to 0.12.4 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#347 opened Aug 17, 2026 by dependabot Bot Loading…
Bump actions/checkout from 6.0.3 to 7.0.1 dependencies Pull requests that update a dependency file github_actions Pull requests that update GitHub Actions code
#342 opened Jul 27, 2026 by dependabot Bot Loading…
Bump actions/setup-python from 6.2.0 to 7.0.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update GitHub Actions code
#341 opened Jul 20, 2026 by dependabot Bot Loading…
Bump grain from 0.2.16 to 0.2.18 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#337 opened Jun 22, 2026 by dependabot Bot Loading…
Bump jax from 0.10.1 to 0.10.2 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#336 opened Jun 22, 2026 by dependabot Bot Loading…
Bump chex from 0.1.91 to 0.1.92 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#333 opened Jun 15, 2026 by dependabot Bot Loading…
Update JAX tutorial to reflect changes in device mesh creation and sharding strategies.
#288 opened Dec 22, 2025 by Davidnet Loading…
Add the jax Stack TR
#266 opened Nov 11, 2025 by ev-br Loading…
3
[ci] add python 3.14 coverage
#250 opened Oct 22, 2025 by jakevdp Collaborator Loading…
@jakevdp
1
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…
1
@jakevdp
3
[WIP] Add "The stack" section to left nav
#212 opened May 27, 2025 by skye Contributor • Draft
2
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…
10
CI: add testing on Python 3.13
#192 opened Apr 9, 2025 by jakevdp Collaborator Loading…
@jakevdp
add example of using LoRA in JAX
#187 opened Mar 22, 2025 by nikolasavic3 • Draft
1
Upgrade Machine Translation JAX AI Stack doc
#185 opened Mar 17, 2025 by 8bitmp3 Contributor Loading…
7
Add tutorial to port Llama 3.2 1B model to JAX
#167 opened Feb 12, 2025 by windmaple Collaborator Loading…
3
Update tutorial: Visualize JAX model
#157 opened Jan 27, 2025 by pavithraes Contributor Loading…
3
Update JAX AI for PyTorch users tutorial
#135 opened Dec 20, 2024 by 8bitmp3 Contributor • Draft
1
Adding tutorial for data loaders in multi device setups
#110 opened Nov 21, 2024 by selamw1 Contributor Loading…
13
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.