a333b9bd1b
* Adds unit normalization and tests * Adds layer normalization and initial tests * Fixes formatting in docstrings * Fix type issues for JAX * Fix nits |
||
---|---|---|
.. | ||
activations | ||
attention | ||
convolutional | ||
core | ||
merging | ||
normalization | ||
pooling | ||
preprocessing | ||
regularization | ||
reshaping | ||
__init__.py | ||
input_spec.py | ||
layer_test.py | ||
layer.py | ||
layers_utils.py |