LICENSE
README.md
pyproject.toml
src/flatbuild/__init__.py
src/flatbuild/__main__.py
src/flatbuild/config.py
src/flatbuild/generate_data.py
src/flatbuild/py.typed
src/flatbuild.egg-info/PKG-INFO
src/flatbuild.egg-info/SOURCES.txt
src/flatbuild.egg-info/dependency_links.txt
src/flatbuild.egg-info/entry_points.txt
src/flatbuild.egg-info/requires.txt
src/flatbuild.egg-info/top_level.txt
src/flatbuild/callbacks/__init__.py
src/flatbuild/callbacks/base.py
src/flatbuild/callbacks/early_stopping.py
src/flatbuild/callbacks/grad_clip.py
src/flatbuild/callbacks/logger.py
src/flatbuild/checkpoint/__init__.py
src/flatbuild/checkpoint/manager.py
src/flatbuild/cli/__init__.py
src/flatbuild/cli/main.py
src/flatbuild/datasets/__init__.py
src/flatbuild/datasets/base.py
src/flatbuild/datasets/loader.py
src/flatbuild/datasets/registry.py
src/flatbuild/exporters/__init__.py
src/flatbuild/exporters/_fwg_reader.py
src/flatbuild/exporters/_tokenizer_copy.py
src/flatbuild/exporters/base.py
src/flatbuild/exporters/fwg.py
src/flatbuild/exporters/gguf.py
src/flatbuild/exporters/huggingface.py
src/flatbuild/exporters/safetensors.py
src/flatbuild/losses/__init__.py
src/flatbuild/losses/cross_entropy.py
src/flatbuild/metrics/__init__.py
src/flatbuild/metrics/language_modeling.py
src/flatbuild/models/__init__.py
src/flatbuild/models/transformer.py
src/flatbuild/models/layers/__init__.py
src/flatbuild/models/layers/attention.py
src/flatbuild/models/layers/block.py
src/flatbuild/models/layers/rmsnorm.py
src/flatbuild/models/layers/rope.py
src/flatbuild/models/layers/swiglu.py
src/flatbuild/optimizers/__init__.py
src/flatbuild/optimizers/factory.py
src/flatbuild/schedulers/__init__.py
src/flatbuild/schedulers/factory.py
src/flatbuild/tokenizers/__init__.py
src/flatbuild/tokenizers/bpe.py
src/flatbuild/tokenizers/template.py
src/flatbuild/trainer/__init__.py
src/flatbuild/trainer/datamodule.py
src/flatbuild/trainer/profiler.py
src/flatbuild/trainer/progress.py
src/flatbuild/trainer/tokenize.py
src/flatbuild/trainer/trainer.py
src/flatbuild/trainer/validation.py
src/flatbuild/utils/__init__.py
src/flatbuild/utils/fs.py
src/flatbuild/utils/logging.py
src/flatbuild/utils/reproducibility.py
tests/test_datasets.py
tests/test_exporters.py
tests/test_exporters_gguf_fwg.py
tests/test_model.py
tests/test_overfit.py
tests/test_preprocessing.py
tests/test_tokenizer.py
tests/test_trainer.py
tests/test_trainer_modules.py