.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/__init__.py
graphene/callbacks/__init__.py
graphene/callbacks/callback.py
graphene/callbacks/model_summary.py
graphene/callbacks/progress_bar.py
graphene/callbacks/wandb.py
graphene/core/__init__.py
graphene/core/datamodule.py
graphene/core/trainer.py
graphene/core/trainmodule.py
graphene/loops/__init__.py
graphene/loops/loop.py
graphene/loops/train_loop.py
graphene/loops/validation_loop.py
graphene/metrics/__init__.py
graphene/metrics/accumulation.py
graphene/metrics/accuracy.py
graphene/metrics/metric.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