Rare find

jaxtransformer. Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...

github.com/kvfrans/jaxtransformer
by @kvfrans

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.