jax-ml/jax-ai-stack
View on GitHub- Stars
- 312
- Forks
- 50
- Open beginner issues
- 0
- Indexed issues
- 13
- Dominant language
- Python
- License
- Apache-2.0
- Last GitHub push
- Aug 17, 2026
- Latest indexed
- Sep 18, 2026
- Contributing guide
- Contributing guide
- Code of conduct
- No code of conduct
- Beginner labels
- No beginner labels indexed
- PR merge metrics
- No merged PRs in 30d
-
jax-ml/jax-ai-stack#149 · 2 comments · 2 reactions · 0 assignees ·
-
Discussion : What is the functionality of `pygrain.ShardOption`, like `pygrain.ShardByJaxProcess`? Open
jax-ml/jax-ai-stack#154 · 0 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#186 · 0 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#214 · 0 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#218 · 2 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#241 · 0 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#261 · 1 comment · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#265 · 3 comments · 0 reactions · 0 assignees ·
-
jax-ml/jax-ai-stack#281 · 0 comments · 0 reactions · 0 assignees ·
-
build failed
jax-ml/jax-ai-stack#326 · 1 comment · 0 reactions · 0 assignees ·
-
build failed
jax-ml/jax-ai-stack#327 · 2 comments · 0 reactions · 0 assignees ·
-
build failed
jax-ml/jax-ai-stack#328 · 1 comment · 0 reactions · 0 assignees ·
-
build failed
jax-ml/jax-ai-stack#329 · 88 comments · 0 reactions · 0 assignees ·