Skip to content

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

New pull request New
10 Open 168 Closed
10 Open 168 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Upgrade ViT image segmentation tutorial for JAX AI Stack
#194 opened Apr 14, 2025 by 8bitmp3 • Review required
CI: add testing on Python 3.13
#192 opened Apr 9, 2025 by jakevdp • Review required
@jakevdp
add example of using LoRA in JAX
#187 opened Mar 22, 2025 by nikolasavic3 • Review required
1
Upgrade Machine Translation JAX AI Stack doc
#185 opened Mar 17, 2025 by 8bitmp3 • Review required
3
Upgrade Transformer Classification doc for JAX AI Stack
#183 opened Mar 17, 2025 by 8bitmp3 • Review required
5
Add tutorial to port Llama 3.2 1B model to JAX
#167 opened Feb 12, 2025 by windmaple • Review required
3
Upgrade JAX AI Stack for miniGPT doc
#158 opened Jan 28, 2025 by 8bitmp3 • Approved
@jakevdp
23
Update tutorial: Visualize JAX model
#157 opened Jan 27, 2025 by pavithraes • Approved
3
Update JAX AI for PyTorch users tutorial
#135 opened Dec 20, 2024 by 8bitmp3 • Draft
1
Adding tutorial for data loaders in multi device setups
#110 opened Nov 21, 2024 by selamw1 • Review required
12
ProTip! Add no:assignee to see everything that’s not assigned.