LICENSE
README.md
pyproject.toml
levanter.egg-info/PKG-INFO
levanter.egg-info/SOURCES.txt
levanter.egg-info/dependency_links.txt
levanter.egg-info/requires.txt
levanter.egg-info/top_level.txt
src/levanter/__init__.py
src/levanter/callbacks.py
src/levanter/checkpoint.py
src/levanter/config.py
src/levanter/distributed.py
src/levanter/doremi.py
src/levanter/eval.py
src/levanter/grad_accum.py
src/levanter/logging.py
src/levanter/lora.py
src/levanter/mesh.py
src/levanter/shapes.py
src/levanter/tensorstore_serialization.py
src/levanter/trainer.py
src/levanter/trainer_state.py
src/levanter/types.py
src/levanter/visualization.py
tests/test_attention.py
tests/test_audio.py
tests/test_background_iterable.py
tests/test_backpack.py
tests/test_checkpoint.py
tests/test_config.py
tests/test_data_mixture.py
tests/test_datetime_utils.py
tests/test_distributed.py
tests/test_doremi.py
tests/test_eval_lm.py
tests/test_export_to_hf.py
tests/test_flash_attention.py
tests/test_gemma.py
tests/test_grad_accum.py
tests/test_hf_checkpoints.py
tests/test_hf_gpt2_serialize.py
tests/test_hf_utils.py
tests/test_in_progress_sequence.py
tests/test_jax_utils.py
tests/test_llama.py
tests/test_llama3.py
tests/test_logging.py
tests/test_longformer.py
tests/test_lora.py
tests/test_mistral.py
tests/test_mpt.py
tests/test_py_utils.py
tests/test_replicated_loader.py
tests/test_shard_cache.py
tests/test_sharded_dataset.py
tests/test_sharded_loader.py
tests/test_shuffle_dataset.py
tests/test_sophia.py
tests/test_tensorstore_serialization.py
tests/test_text.py
tests/test_tokenized_document_cache.py
tests/test_torch_serialization.py
tests/test_tracker.py
tests/test_train_asr.py
tests/test_train_lm.py
tests/test_utils.py
tests/test_viz_lm.py
tests/test_weight_decay_mask.py