CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
USN.egg-info/PKG-INFO
USN.egg-info/SOURCES.txt
USN.egg-info/dependency_links.txt
USN.egg-info/entry_points.txt
USN.egg-info/requires.txt
USN.egg-info/top_level.txt
docs/DOCUMENTATION.md
examples/quick_start.py
tests/test_backends.py
tests/test_beam_search_repetition.py
tests/test_generator_streaming.py
tests/test_optim_factory.py
tests/test_trainer_eval_checkpoint.py
usn/__init__.py
usn/__init__.pyi
usn/exceptions.py
usn/py.typed
usn/backends/__init__.py
usn/backends/acceleration.py
usn/backends/detection.py
usn/backends/fallbacks.py
usn/backends/triton_kernels.py
usn/cli/__init__.py
usn/cli/main.py
usn/config/__init__.py
usn/config/generation_config.py
usn/config/model_config.py
usn/config/training_config.py
usn/core/__init__.py
usn/core/activations.py
usn/core/base.py
usn/core/interfaces.py
usn/core/types.py
usn/datasets/__init__.py
usn/datasets/collate.py
usn/datasets/math_dataset.py
usn/datasets/usn_dataset.py
usn/inference/__init__.py
usn/inference/generator.py
usn/layers/__init__.py
usn/layers/block.py
usn/layers/chunked_scan.py
usn/layers/norm.py
usn/layers/parallel_scan.py
usn/losses/__init__.py
usn/losses/cross_entropy.py
usn/models/__init__.py
usn/models/embedding.py
usn/models/usn_model.py
usn/modules/__init__.py
usn/modules/channel_mixing.py
usn/modules/exponential_gating.py
usn/modules/input_projection.py
usn/modules/selective_writing.py
usn/modules/state_readout.py
usn/modules/state_update.py
usn/modules/temporal_mixing.py
usn/optim/__init__.py
usn/optim/factory.py
usn/optim/schedulers.py
usn/serialization/__init__.py
usn/serialization/export.py
usn/serialization/format_spec.py
usn/serialization/migration.py
usn/serialization/reader.py
usn/serialization/validator.py
usn/serialization/writer.py
usn/tokenizers/__init__.py
usn/tokenizers/bpe_tokenizer.py
usn/tokenizers/char_tokenizer.py
usn/tokenizers/word_tokenizer.py
usn/training/__init__.py
usn/training/curriculum.py
usn/training/distributed.py
usn/training/stability.py
usn/training/trainer.py
usn/utils/__init__.py
usn/utils/counting.py
usn/utils/diagnostics.py
usn/utils/profiling.py
usn/utils/seed.py
usn/utils/timing.py
usn/utils/visualization.py