Search results

GPT-OSS

Model request

Status: Open.

#143 In jax-ml/bonsai;

· Aatman09 opened on Jan 24, 2026

umT5 caching

enhancementNew feature or requestNew feature or request

Status: Open.

#137 In jax-ml/bonsai;

· chapman20j opened on Jan 21, 2026

Olmo3

Model request

Status: Open.

#126 In jax-ml/bonsai;

· chapman20j opened on Jan 16, 2026

Gemma3 Local Attention Cache

enhancementNew feature or requestNew feature or request

Status: Open.

#124 In jax-ml/bonsai;

· chapman20j opened on Jan 8, 2026

VGGT

Model request

Status: Open.

#116 In jax-ml/bonsai;

· imgengineer opened on Dec 31, 2025

Llama 3.2 (1B & 3B)

Model request

Status: Open.

#115 In jax-ml/bonsai;

· Moriyuki-S opened on Dec 30, 2025

Failed to save and load checkpoints in UNet segmentation example

Status: Open.

#113 In jax-ml/bonsai;

· xrhd opened on Dec 27, 2025

Graph Attention Network (GAT)

Model request

Status: Open.

#112 In jax-ml/bonsai;

· xrhd opened on Dec 27, 2025

Update gelu to include <code>approximate=False</code>

enhancementNew feature or requestNew feature or request

good first issueGood for newcomersGood for newcomers

Status: Open.

#111 In jax-ml/bonsai;

· chapman20j opened on Dec 26, 2025

Denoising Diffusion Probabilistic Models (DDPM)

Model request

Status: Open.

#109 In jax-ml/bonsai;

· yotamyaniv opened on Dec 23, 2025

T5Gemma and T5Gemma2

Model request

Status: Open.

#108 In jax-ml/bonsai;

· jenriver opened on Dec 22, 2025

Qwen2.5-VL

Model request

Status: Open.

#93 In jax-ml/bonsai;

· pathfinder-pf opened on Nov 25, 2025