Ecosyste.ms: Packages
An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.
conda-forge.org : flax : 0.5.2
Flax is a neural network library for JAX that is designed for flexibility.
Registry -
Download -
JSON
purl: pkg:conda/[email protected]
Published:
Indexed:
Related tag:
v0.5.2
- compare
Loading...
Readme
Loading...
Dependencies
- jax >=0.3.2
- matplotlib-base
- msgpack-python
- numpy >=1.12
- optax
- python >=3.6
- pyyaml >=5.4.1
- rich ~=11.1.0
- typing_extensions >=4.1.1