LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.cfg
jaxflow/__init__.py
jaxflow.egg-info/PKG-INFO
jaxflow.egg-info/SOURCES.txt
jaxflow.egg-info/dependency_links.txt
jaxflow.egg-info/requires.txt
jaxflow.egg-info/top_level.txt
jaxflow/activations/__init__.py
jaxflow/activations/activations.py
jaxflow/callbacks/__init__.py
jaxflow/core/__init__.py
jaxflow/core/auto_name.py
jaxflow/core/variable.py
jaxflow/gradient/__init__.py
jaxflow/gradient/gradient.py
jaxflow/initializers/__init__.py
jaxflow/initializers/initializers.py
jaxflow/layers/__init__.py
jaxflow/layers/layer.py
jaxflow/layers/convolutional/__init__.py
jaxflow/layers/convolutional/conv1d.py
jaxflow/layers/convolutional/conv2d.py
jaxflow/layers/convolutional/conv3d.py
jaxflow/layers/embedding/__init__.py
jaxflow/layers/embedding/embedding.py
jaxflow/layers/fully_connected/__init__.py
jaxflow/layers/fully_connected/dense.py
jaxflow/layers/normalization/__init__.py
jaxflow/layers/normalization/layer_normalization.py
jaxflow/layers/pooling/__init__.py
jaxflow/layers/pooling/global_average_pooling1d.py
jaxflow/layers/pooling/global_average_pooling2d.py
jaxflow/layers/pooling/global_average_pooling3d.py
jaxflow/layers/pooling/max_pooling1d.py
jaxflow/layers/pooling/max_pooling2d.py
jaxflow/layers/regularization/__init__.py
jaxflow/layers/regularization/dropout.py
jaxflow/layers/reshaping/__init__.py
jaxflow/layers/reshaping/flatten.py
jaxflow/layers/rnn/__init__.py
jaxflow/layers/rnn/gru.py
jaxflow/layers/rnn/lstm.py
jaxflow/layers/rnn/rnn.py
jaxflow/losses/__init__.py
jaxflow/losses/loss.py
jaxflow/losses/losses.py
jaxflow/math/__init__.py
jaxflow/metrics/__init__.py
jaxflow/metrics/accuracy.py
jaxflow/metrics/confusion.py
jaxflow/metrics/metric.py
jaxflow/metrics/regression.py
jaxflow/models/__init__.py
jaxflow/models/model.py
jaxflow/optimizers/__init__.py
jaxflow/optimizers/optimizer.py
jaxflow/optimizers/optimizers.py
jaxflow/random/__init__.py
jaxflow/regularizers/__init__.py