LICENSE
README.md
pyproject.toml
src/cupydiff/__init__.py
src/cupydiff/backend.py
src/cupydiff/checkpoint.py
src/cupydiff/cli.py
src/cupydiff/config.py
src/cupydiff/data.py
src/cupydiff/diffusion.py
src/cupydiff/distributed.py
src/cupydiff/layers.py
src/cupydiff/loss.py
src/cupydiff/model.py
src/cupydiff/optim.py
src/cupydiff/telemetry.py
src/cupydiff/train.py
src/cupydiff.egg-info/PKG-INFO
src/cupydiff.egg-info/SOURCES.txt
src/cupydiff.egg-info/dependency_links.txt
src/cupydiff.egg-info/entry_points.txt
src/cupydiff.egg-info/requires.txt
src/cupydiff.egg-info/top_level.txt
src/cupydiff/ops/__init__.py
src/cupydiff/ops/adaln.py
src/cupydiff/ops/attention.py
src/cupydiff/ops/linear.py
src/cupydiff/ops/matmul.py
src/cupydiff/ops/rmsnorm.py
src/cupydiff/ops/rope.py
src/cupydiff/ops/swiglu.py
src/cupydiff/tests/__init__.py
src/cupydiff/tests/run_all.py
src/cupydiff/tests/test_config.py
src/cupydiff/tests/test_data.py
src/cupydiff/tests/test_diffusion.py
src/cupydiff/tests/test_gpu.py
src/cupydiff/tests/test_loss.py
src/cupydiff/tests/test_model.py
src/cupydiff/tests/test_ops.py
src/cupydiff/tests/test_optim.py
src/cupydiff/tests/test_train_step.py
tests/test_config.py
tests/test_data.py
tests/test_diffusion.py
tests/test_gpu.py
tests/test_loss.py
tests/test_model.py
tests/test_ops.py
tests/test_optim.py
tests/test_train_step.py