All issues

Search results

Failed build: Nightly Tests

build failed

Status: Open.

#328 In jax-ml/jax-ai-stack;

· github-actions opened on May 27, 2026

Failed build: Nightly Tests

build failed

Status: Open.

#329 In jax-ml/jax-ai-stack;

· github-actions opened on May 27, 2026

Failed build: Nightly Tests

build failed

Status: Open.

#327 In jax-ml/jax-ai-stack;

· github-actions opened on May 27, 2026

Failed build: Nightly Tests

build failed

Status: Open.

#326 In jax-ml/jax-ai-stack;

· github-actions opened on May 27, 2026

[Docs] JAX_Vision_transformer: ViT Tutorial at https://docs.jaxstack.ai/en/latest/JAX_Vision_transformer.html Fails on Kaggle (TPU v5e-8) or T4 (Colab)

Status: Open.

#281 In jax-ml/jax-ai-stack;

· Prakharprasun opened on Dec 4, 2025

Running the JAX_for_LLM_pretraining.ipynb in colab fails

Status: Open.

#265 In jax-ml/jax-ai-stack;

· Davidnet opened on Nov 10, 2025

Segmentation fault when initializing UNet model - (from tutorial problem)

Status: Open.

#261 In jax-ml/jax-ai-stack;

· Tenchberry opened on Oct 31, 2025

in digits_vae.ipynb, error: no attribute ModelAndOptimizer

Status: Open.

#241 In jax-ml/jax-ai-stack;

· joshtburdick opened on Oct 6, 2025

Execution errors in "Porting a PyTorch model to JAX"

Status: Open.

#218 In jax-ml/jax-ai-stack;

· pavithraes opened on Jun 18, 2025

Image captioning tutorial - transformers library with JAX

Status: Open.

#214 In jax-ml/jax-ai-stack;

· 8bitmp3 opened on Jun 12, 2025

Add LoRA Training Example to JAX Examples

Status: Open.

#186 In jax-ml/jax-ai-stack;

· nikolasavic3 opened on Mar 21, 2025

Discussion : What is the functionality of pygrain.ShardOption, like pygrain.ShardByJaxProcess?

Status: Open.

#154 In jax-ml/jax-ai-stack;

· carlesoctav opened on Jan 22, 2025