.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
LICENSE
README.md
__init__.py
pyproject.toml
train.py
.github/workflows/package.yml
.github/workflows/python-precommit.yml
examples/__init__.py
examples/cifar_datamodule.py
examples/resnet.py
graphene_mlx.egg-info/PKG-INFO
graphene_mlx.egg-info/SOURCES.txt
graphene_mlx.egg-info/dependency_links.txt
graphene_mlx.egg-info/requires.txt
graphene_mlx.egg-info/top_level.txt
requirements/requirements.dev.in
requirements/requirements.in
scripts/install.sh
scripts/lib.sh
scripts/lock-requirements.sh
src/__init__.py
src/callbacks/callback.py
src/callbacks/model_summary.py
src/core/__init__.py
src/core/datamodule.py
src/core/trainer.py
src/core/trainmodule.py
src/loops/__init__.py
src/loops/loop.py
src/loops/train_loop.py
src/loops/validation_loop.py
src/metrics/__init__.py
src/metrics/accumulation.py
src/metrics/accuracy.py
src/metrics/metric.py