README.md
setup.py
fastax/__init__.py
fastax/utils.py
fastax.egg-info/PKG-INFO
fastax.egg-info/SOURCES.txt
fastax.egg-info/dependency_links.txt
fastax.egg-info/requires.txt
fastax.egg-info/top_level.txt
fastax/activations/__init__.py
fastax/activations/activations.py
fastax/initializers/__init__.py
fastax/initializers/initializers.py
fastax/layers/__init__.py
fastax/layers/attention.py
fastax/layers/attention_test.py
fastax/layers/base.py
fastax/layers/base_test.py
fastax/layers/combinators.py
fastax/layers/combinators_test.py
fastax/layers/convolution.py
fastax/layers/convolution_test.py
fastax/layers/core.py
fastax/layers/core_test.py
fastax/layers/initializers.py
fastax/layers/initializers_test.py
fastax/layers/layers.py
fastax/layers/normalization.py
fastax/layers/normalization_test.py
fastax/layers/pooling.py
fastax/layers/pooling_test.py
fastax/layers/reversible.py
fastax/layers/reversible_test.py
fastax/layers/rnn.py
fastax/layers/rnn_test.py
fastax/losses/__init__.py
fastax/losses/losses.py
fastax/optimizers/__init__.py
fastax/optimizers/optimizers.py