.gitignore
.gitlint
.pre-commit-config.yaml
.python-version
LICENSE
README.md
pyproject.toml
uv.lock
.github/workflows/ci.yaml
examples/diffusiongemma/README.md
examples/diffusiongemma/main.py
src/d3pm/__init__.py
src/d3pm.egg-info/PKG-INFO
src/d3pm.egg-info/SOURCES.txt
src/d3pm.egg-info/dependency_links.txt
src/d3pm.egg-info/requires.txt
src/d3pm.egg-info/top_level.txt
src/d3pm/_src/__init__.py
src/d3pm/_src/accept.py
src/d3pm/_src/config.py
src/d3pm/_src/early_stop.py
src/d3pm/_src/loss.py
src/d3pm/_src/process.py
src/d3pm/_src/sampler.py
src/d3pm/_src/schedule.py
src/d3pm/_src/tokenizer.py
src/d3pm/_src/models/__init__.py
src/d3pm/_src/models/diffusion_gemma.py
src/d3pm/_src/nn/__init__.py
src/d3pm/_src/nn/attention.py
src/d3pm/_src/nn/feedforward.py
src/d3pm/_src/nn/masks.py
src/d3pm/_src/nn/moe.py
src/d3pm/_src/nn/norm.py
src/d3pm/nn/__init__.py