LICENSE
README.md
pyproject.toml
src/juml/__init__.py
src/juml/base.py
src/juml/device.py
src/juml/framework.py
src/juml/test_utils.py
src/juml/commands/__init__.py
src/juml/commands/base.py
src/juml/commands/plot_1d_regression.py
src/juml/commands/plot_confusion_matrix.py
src/juml/commands/profile.py
src/juml/commands/sweep.py
src/juml/commands/train.py
src/juml/datasets/__init__.py
src/juml/datasets/base.py
src/juml/datasets/cifar10.py
src/juml/datasets/fromdict.py
src/juml/datasets/linear.py
src/juml/datasets/mnist.py
src/juml/datasets/randimg.py
src/juml/datasets/sinmix.py
src/juml/datasets/split.py
src/juml/datasets/synthetic.py
src/juml/loss/__init__.py
src/juml/loss/asmse.py
src/juml/loss/base.py
src/juml/loss/chamfer.py
src/juml/loss/crossentropy.py
src/juml/loss/mse.py
src/juml/models/__init__.py
src/juml/models/base.py
src/juml/models/cnn.py
src/juml/models/embed.py
src/juml/models/linear.py
src/juml/models/linearmodel.py
src/juml/models/mlp.py
src/juml/models/pool.py
src/juml/models/rzcnn.py
src/juml/models/rzmlp.py
src/juml/models/sequential.py
src/juml/train/__init__.py
src/juml/train/base.py
src/juml/train/bpsp.py
src/juml/train/profiler.py
src/juml/train/sweeper.py
src/juml_toolkit.egg-info/PKG-INFO
src/juml_toolkit.egg-info/SOURCES.txt
src/juml_toolkit.egg-info/dependency_links.txt
src/juml_toolkit.egg-info/entry_points.txt
src/juml_toolkit.egg-info/requires.txt
src/juml_toolkit.egg-info/top_level.txt
tests/test_cli_args.py