LICENSE
MANIFEST.in
README.md
pyproject.toml
docs/conf.py
examples/adding_problem.ipynb
examples/char_rnn.ipynb
examples/dynamical_system_forecasting.ipynb
examples/ benchmarks/benchmark_clockwork.ipynb
examples/ benchmarks/benchmark_lstm.ipynb
rnn_jax/__init__.py
rnn_jax.egg-info/PKG-INFO
rnn_jax.egg-info/SOURCES.txt
rnn_jax.egg-info/dependency_links.txt
rnn_jax.egg-info/requires.txt
rnn_jax.egg-info/top_level.txt
rnn_jax/cells/__init__.py
rnn_jax/cells/antisymmetric.py
rnn_jax/cells/base.py
rnn_jax/cells/clockwork.py
rnn_jax/cells/gated.py
rnn_jax/cells/lipschitz.py
rnn_jax/cells/obscillator.py
rnn_jax/cells/unitary.py
rnn_jax/cells/vanilla.py
rnn_jax/datasets/smnist.py
rnn_jax/datasets/_reservoirpy/__init__.py
rnn_jax/datasets/_reservoirpy/_chaos.py
rnn_jax/datasets/_reservoirpy/_japanese_vowels.py
rnn_jax/datasets/_reservoirpy/_santafe_laser.py
rnn_jax/datasets/_reservoirpy/_seed.py
rnn_jax/datasets/_reservoirpy/_utils.py
rnn_jax/datasets/_reservoirpy/tests/__init__.py
rnn_jax/datasets/_reservoirpy/tests/test_datasets.py
rnn_jax/layers/__init__.py
rnn_jax/layers/encoder.py
rnn_jax/layers/modular.py
rnn_jax/layers/reservoir.py
rnn_jax/layers/rnn.py
rnn_jax/ssm/__init__.py
rnn_jax/ssm/base.py
rnn_jax/ssm/lru.py
rnn_jax/ssm/mixers.py
rnn_jax/ssm/models.py
rnn_jax/ssm/s5.py
rnn_jax/tests/__init__.py
rnn_jax/tests/test_datasets.py
rnn_jax/tests/test_rnn.py
rnn_jax/tests/test_ssm.py
rnn_jax/utils/__init__.py
rnn_jax/utils/utils.py
tests/test_imports.py
tests/test_stack.py