Skip to content

Pull requests: jax-ml/bonsai

New pull request New
7 Open 73 Closed
7 Open 73 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

vae implementation
#122 opened Jan 7, 2026 by eari100 Loading…
4 tasks done
1
8
VJEPA2 model, classifier works well! need some help with fm though
#121 opened Jan 6, 2026 by coder0143 Loading…
4 tasks done
1
13
Refactor tutorial to use dataclass for configuration
#119 opened Jan 4, 2026 by Aatman09 Loading…
4 tasks done
1
3
Fix UNet segmentation example
#114 opened Dec 27, 2025 by xrhd Loading…
2 tasks done
1
1
JAX AI Stack: Port UNETR model
#106 opened Dec 17, 2025 by jenriver Loading…
3 tasks done
5
add sglang-jax refer to README.md
#88 opened Nov 19, 2025 by Iamleos Loading…
4 tasks done
4
Add CLIP-JAX ADE20K implementation, training, and tests
#83 opened Nov 12, 2025 by Thakor-Yashpal Loading…
13
ProTip! Type g i on any issue or pull request to go back to the issue listing page.