LICENSE
MANIFEST.in
README.md
setup.py
docs/conf.py
examples/__init__.py
examples/approx_metrics/__init__.py
examples/approx_metrics/main.py
examples/approx_metrics/main_test.py
examples/flax_integration/__init__.py
examples/flax_integration/main.py
examples/flax_integration/main_test.py
examples/t5x/models.py
examples/t5x/models_test.py
examples/t5x/tasks.py
examples/t5x/tasks_test.py
rax/__init__.py
rax/types.py
rax/utils.py
rax.egg-info/PKG-INFO
rax.egg-info/SOURCES.txt
rax.egg-info/dependency_links.txt
rax.egg-info/requires.txt
rax.egg-info/top_level.txt
rax/_src/__init__.py
rax/_src/losses.py
rax/_src/losses_test.py
rax/_src/metrics.py
rax/_src/metrics_test.py
rax/_src/t12n.py
rax/_src/t12n_test.py
rax/_src/types.py
rax/_src/utils.py
rax/_src/utils_test.py
requirements/requirements-docs.txt
requirements/requirements-examples.txt
requirements/requirements-test.txt
requirements/requirements.txt