_            _    _        _         _
      /\ \         /\ \ /\ \     /\_\      / /\
      \_\ \       /  \ \\ \ \   / / /     / /  \
      /\__ \     / /\ \ \\ \ \_/ / /     / / /\ \__
     / /_ \ \   / / /\ \ \\ \___/ /     / / /\ \___\
    / / /\ \ \ / / /  \ \_\\ \ \_/      \ \ \ \/___/
   / / /  \/_// / /   / / / \ \ \        \ \ \
  / / /      / / /   / / /   \ \ \   _    \ \ \
 / / /      / / /___/ / /     \ \ \ /_/\__/ / /
/_/ /      / / /____\/ /       \ \_\\ \/___/ /
\_\/       \/_________/         \/_/ \_____\/
python-flax 0.8.0
Propagated dependencies: python-einops@0.6.1 python-jax@0.4.20 python-optax@0.1.5 python-orbax-checkpoint@0.4.5 python-msgpack@1.0.4 python-numpy@1.24.4 python-pyyaml@6.0.1 python-rich@13.7.1 python-tensorstore@0.1.52 python-typing-extensions@4.10.0
Channel: guix-science
Location: guix-science/packages/machine-learning.scm (guix-science packages machine-learning)
Home page: https://github.com/google/flax
Licenses: ASL 2.0
Synopsis: Neural network library for JAX designed for flexibility
Description:

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

Total results: 1