Flax NNX

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

Flax NNX is a simplified API for Flax, a flexible end-to-end neural network library for JAX,...

Tokens:800,818
Snippets:5,404
Trust Score:8.3
Update:2 weeks ago
Tokens:
Raw