Orbax

https://orbax.readthedocs.io/en/latest/

Orbax is an umbrella namespace providing common training utilities for JAX users, including flexible...

Tokens:321,255
Snippets:2,520
Trust Score:8.3
Update:1 month ago
Tokens:
Raw