.gitattributes
.gitignore
.gitlint
.pre-commit-config.yaml
.python-version
LICENSE
Makefile
README.md
pyproject.toml
uv.lock
.github/workflows/ci.yaml
.github/workflows/release.yaml
blaxbird/__init__.py
blaxbird/experimental.py
blaxbird.egg-info/PKG-INFO
blaxbird.egg-info/SOURCES.txt
blaxbird.egg-info/dependency_links.txt
blaxbird.egg-info/requires.txt
blaxbird.egg-info/top_level.txt
blaxbird/_src/__init__.py
blaxbird/_src/checkpointer.py
blaxbird/_src/test_blaxbird.py
blaxbird/_src/trainer.py
blaxbird/_src/experimental/__init__.py
blaxbird/_src/experimental/edm.py
blaxbird/_src/experimental/rfm.py
blaxbird/_src/experimental/samplers.py
blaxbird/_src/experimental/nn/__init__.py
blaxbird/_src/experimental/nn/dit.py
blaxbird/_src/experimental/nn/embedding.py
blaxbird/_src/experimental/nn/mlp.py
examples/.gitignore
examples/cifar10_flow_matching/dataloader.py
examples/cifar10_flow_matching/main.py
examples/mnist_classification/dataloader.py
examples/mnist_classification/main.py
examples/mnist_classification/model.py