JAX backend for SGL
Python
354
640 commits
updated Sep 24, 2026
(top 30 of 58)
109 commits
50 commits
43 commits
ryanbarouki/JaxGPT
GPT-2 implemented in Flax/JAX. Pretrained weights can be loaded from huggingface.
0
n2cholas/awesome-jax
JAX - A curated list of resources https://github.com/google/jax
2,164
google/CommonLoopUtils
CLU lets you write beautiful training loops in JAX.
371
jlapeyre/Symata.jl
language for symbolic mathematics
175
fsprojects/fsharp-ai-tools
TensorFlow API for F# + F# for AI Models eDSL
211
dfdx/gendo
Generative models
2
SmilingWolf/JAX-CV
Repo for my JAX CV experiments. Mostly revolving around the Danbooru20xx dataset
49
yuta0306/JPTScore
GPTScore with Japanese LLMs
99.6%
