google/flax

Flax is a neural network library for JAX that is designed for flexibility.

语言：Jupyter Notebook · GitHub 星标：7,363 · Fork：847

话题：jax

在 GitHub 上查看原仓库