Issues

• Current Jax and Flax versions are incompatible

  jax-ml/bonsai#174 · Filter by author darsnack opened on Apr 27, 2026
  1

• Qwen3.5

  Model request
  jax-ml/bonsai#173 · Filter by author coder0143 opened on Mar 3, 2026

• SigLIP 2

  Model request
  jax-ml/bonsai#171 · Filter by author Moriyuki-S opened on Feb 23, 2026
  3

• Add in-repo inference notebook for DINOv3

  jax-ml/bonsai#168 · Filter by author Moriyuki-S opened on Feb 15, 2026

• Alphagenome

  Model request
  jax-ml/bonsai#156 · Filter by author coder0143 opened on Feb 6, 2026
  1

  

• BART

  Model request
  jax-ml/bonsai#151 · Filter by author Awesome075 opened on Feb 3, 2026
  3

  

• MobileNet

  Model request
  jax-ml/bonsai#150 · Filter by author Himanshu-639 opened on Feb 2, 2026
  2

  

• GPT-OSS

  Model request
  jax-ml/bonsai#143 · Filter by author Aatman09 opened on Jan 24, 2026
  

• umT5 caching

  enhancement
  jax-ml/bonsai#137 · Filter by author chapman20j opened on Jan 21, 2026
  1

• Olmo3

  Model request
  jax-ml/bonsai#126 · Filter by author chapman20j opened on Jan 16, 2026
  

• Gemma3 Local Attention Cache

  enhancement
  jax-ml/bonsai#124 · Filter by author chapman20j opened on Jan 8, 2026
  7

  

• VGGT

  Model request
  jax-ml/bonsai#116 · Filter by author imgengineer opened on Dec 31, 2025
  3

• Llama 3.2 (1B & 3B)

  Model request
  jax-ml/bonsai#115 · Filter by author Moriyuki-S opened on Dec 30, 2025
  3

  

• Failed to save and load checkpoints in UNet segmentation example

  jax-ml/bonsai#113 · Filter by author xrhd opened on Dec 27, 2025
  1

• Graph Attention Network (GAT)

  Model request
  jax-ml/bonsai#112 · Filter by author xrhd opened on Dec 27, 2025
  2

  

• Update gelu to include approximate=False

  enhancementgood first issue
  jax-ml/bonsai#111 · Filter by author chapman20j opened on Dec 26, 2025
  5

  

• Denoising Diffusion Probabilistic Models (DDPM)

  Model request
  jax-ml/bonsai#109 · Filter by author yotamyaniv opened on Dec 23, 2025
  1

  

• T5Gemma and T5Gemma2

  Model request
  jax-ml/bonsai#108 · Filter by author jenriver opened on Dec 22, 2025
  1

  

• Model Support : MiMo-Audio

  Model request
  jax-ml/bonsai#91 · Filter by author pengchengneo opened on Nov 23, 2025
  7

  

• Model Request: Wan2.1

  Model request
  jax-ml/bonsai#90 · Filter by author labyrinth-ssr opened on Nov 21, 2025
  1

  

• Add a pretrained checkpoint and tests for U-Net.

  good first issuehelp wanted
  jax-ml/bonsai#73 · Filter by author jenriver opened on Nov 5, 2025
  4

  

• CLIP (originally ITACLIP)

  Model request
  jax-ml/bonsai#72 · Filter by author Thakor-Yashpal opened on Nov 5, 2025
  4

  

• Whisper In-progress

  good first issuehelp wanted
  jax-ml/bonsai#71 · Filter by author chapman20j opened on Nov 4, 2025
  4

  

• Model Request: COSMOS-Transfer1

  Model request
  jax-ml/bonsai#8 · Filter by author jenriver opened on Jul 25, 2025