JAX AI Stack

https://github.com/jax-ml/jax-ai-stack

JAX AI Stack is a curated collection of open-source JAX-based libraries, including Flax, Optax, and...

Tokens:199,011
Snippets:1,517
Trust Score:8.4
License:Apache-2.0
Update:1 week ago
Tokens:
Raw