LICENSE.txt
README.md
setup.py
flax_gated_linear_rnn/__init__.py
flax_gated_linear_rnn/attention.py
flax_gated_linear_rnn/gated_linear_rnn.py
flax_gated_linear_rnn/text_2_speech_models.py
flax_gated_linear_rnn.egg-info/PKG-INFO
flax_gated_linear_rnn.egg-info/SOURCES.txt
flax_gated_linear_rnn.egg-info/dependency_links.txt
flax_gated_linear_rnn.egg-info/requires.txt
flax_gated_linear_rnn.egg-info/top_level.txt
flax_gated_linear_rnn/base_models/__init__.py
flax_gated_linear_rnn/base_models/channel_mixing.py
flax_gated_linear_rnn/base_models/sequence_model.py
flax_gated_linear_rnn/base_models/time_mixing.py
flax_gated_linear_rnn/language_models/__init__.py
flax_gated_linear_rnn/language_models/gated_linear_rnn_lm.py
flax_gated_linear_rnn/language_models/transformer_lm.py