google/flax

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

PythonJupyter NotebookOtherjax
This is stars and forks stats for /google/flax repository. As of 06 May, 2024 this repository has 4770 stars and 543 forks.

Flax: A neural network library and ecosystem for JAX designed for flexibility Overview | Quick install | What does Flax look like? | Documentation This README is a very short intro. To learn everything you need to know about Flax, refer to our full documentation. Flax was originally started by engineers and researchers within the Brain Team in Google Research (in close collaboration with the JAX team), and is now developed jointly with the open source community. Flax is being used by a growing community...
Read on GithubGithub Stats Page
repotechsstarsweeklyforksweekly
Crypto-Cat/CTFPythonCOther1.2k02540
Visualize-ML/Book4_Power-of-MatrixPython6.2k08530
USTC3DV/NDR-codePythonC++Other4940570
conan-io/conanPython7.2k09280
ApexAI/rules_rosStarlarkPythonShell30070
dwcoder/QuantitativePrimerTeXPythonR895+3125+2
Linen-dev/linen.devTypeScriptSCSSJavaScript2.6k01320
inpla/inplaYaccCStandard ML204080
redhat-cop/controller_configurationYAMLJinjaPython22301160
Network-Goods/arcological-websiteSCSSAstroCSS0010