Flax

https://github.com/google/flax

Flax is a neural network library and ecosystem for JAX, designed for flexibility and enabling users...

Tokens:179,364
Snippets:1,649
Trust Score:8.9
License:Apache-2.0
Update:2 months ago
Tokens:
Raw