Issues

• Failed build: Nightly Tests

  build failed
  jax-ml/jax-ai-stack#328 · Filter by author github-actions opened on May 27, 2026
  1

• Failed build: Nightly Tests

  build failed
  jax-ml/jax-ai-stack#329 · Filter by author github-actions opened on May 27, 2026
  149

• Failed build: Nightly Tests

  build failed
  jax-ml/jax-ai-stack#327 · Filter by author github-actions opened on May 27, 2026
  2

• Failed build: Nightly Tests

  build failed
  jax-ml/jax-ai-stack#326 · Filter by author github-actions opened on May 27, 2026
  1

• [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)

  jax-ml/jax-ai-stack#281 · Filter by author Prakharprasun opened on Dec 4, 2025

• Running the JAX_for_LLM_pretraining.ipynb in colab fails

  jax-ml/jax-ai-stack#265 · Filter by author Davidnet opened on Nov 10, 2025
  3

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

  jax-ml/jax-ai-stack#261 · Filter by author Tenchberry opened on Oct 31, 2025
  1

• in digits_vae.ipynb, error: no attribute ModelAndOptimizer

  jax-ml/jax-ai-stack#241 · Filter by author joshtburdick opened on Oct 6, 2025

• Execution errors in "Porting a PyTorch model to JAX"

  jax-ml/jax-ai-stack#218 · Filter by author pavithraes opened on Jun 18, 2025
  2

• Image captioning tutorial - transformers library with JAX

  jax-ml/jax-ai-stack#214 · Filter by author 8bitmp3 opened on Jun 12, 2025

• Add LoRA Training Example to JAX Examples

  jax-ml/jax-ai-stack#186 · Filter by author nikolasavic3 opened on Mar 21, 2025

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

  jax-ml/jax-ai-stack#154 · Filter by author carlesoctav opened on Jan 22, 2025

• Colab XLA Error in JAX_for_LLM_pretraining.ipynb

  jax-ml/jax-ai-stack#149 · Filter by author Matt-Hurd opened on Jan 13, 2025
  2