Description
Neural networks can be built in JAX with DeepMind Haiku’s module abstractions. This library is for machine-learning researchers and developers designing model architectures. Training can consume significant compute and datasets may contain sensitive records.