Search results

Mamba2: SSM State Caching for Efficient Incremental Generation

Status: Open.

#120 In jax-ml/bonsai;

· CosmoNaught opened on Jan 6, 2026

VJEPA2

Model request

Status: Open.

#117 In jax-ml/bonsai;

· coder0143 opened on Jan 1, 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

Port Encoder-Decoder Example from Jax AI Stack

Status: Open.

#107 In jax-ml/bonsai;

· chapman20j opened on Dec 19, 2025

Gemma3

Model request

Status: Open.

#100 In jax-ml/bonsai;

· chapman20j opened on Dec 9, 2025

umT5

Model request

Status: Open.

#96 In jax-ml/bonsai;

· Iamleos opened on Dec 2, 2025