LICENSE
README.md
pyproject.toml
src/juml/__init__.py
src/juml/framework.py
src/juml/util.py
src/juml/_old/__init__.py
src/juml/_old/base.py
src/juml/_old/device.py
src/juml/_old/framework.py
src/juml/_old/test_utils.py
src/juml/_old/commands/__init__.py
src/juml/_old/commands/base.py
src/juml/_old/commands/compare_sweeps.py
src/juml/_old/commands/plot_1d_regression.py
src/juml/_old/commands/plot_confusion_matrix.py
src/juml/_old/commands/plot_sequential.py
src/juml/_old/commands/profile.py
src/juml/_old/commands/sweep.py
src/juml/_old/commands/sweep2d.py
src/juml/_old/commands/train.py
src/juml/_old/datasets/__init__.py
src/juml/_old/datasets/base.py
src/juml/_old/datasets/cifar10.py
src/juml/_old/datasets/fromdict.py
src/juml/_old/datasets/lineardataset.py
src/juml/_old/datasets/mnist.py
src/juml/_old/datasets/randimg.py
src/juml/_old/datasets/sinmix.py
src/juml/_old/datasets/split.py
src/juml/_old/datasets/synthetic.py
src/juml/_old/loss/__init__.py
src/juml/_old/loss/asmse.py
src/juml/_old/loss/base.py
src/juml/_old/loss/chamfer.py
src/juml/_old/loss/crossentropy.py
src/juml/_old/loss/mse.py
src/juml/_old/models/__init__.py
src/juml/_old/models/base.py
src/juml/_old/models/cnn.py
src/juml/_old/models/embed.py
src/juml/_old/models/linear.py
src/juml/_old/models/linearmodel.py
src/juml/_old/models/mlp.py
src/juml/_old/models/pool.py
src/juml/_old/models/rzcnn.py
src/juml/_old/models/rzmlp.py
src/juml/_old/models/sequential.py
src/juml/_old/tools/__init__.py
src/juml/_old/tools/display.py
src/juml/_old/tools/experiment.py
src/juml/_old/tools/plot_type.py
src/juml/_old/tools/profiler.py
src/juml/_old/tools/sweeper.py
src/juml/_old/train/__init__.py
src/juml/_old/train/base.py
src/juml/_old/train/bpsp.py
src/juml/_old/train/bpspde.py
src/juml/commands/__init__.py
src/juml/commands/command.py
src/juml/commands/sweep.py
src/juml/commands/train_classification.py
src/juml/data/__init__.py
src/juml/data/dataset.py
src/juml/data/mnist.py
src/juml/experiments/__init__.py
src/juml/experiments/experiment.py
src/juml/experiments/group.py
src/juml/experiments/plot_config.py
src/juml/models/__init__.py
src/juml/models/model.py
src/juml/models/relu_mlp.py
src/juml/models/layers/__init__.py
src/juml/models/layers/linear.py
src/juml_framework.egg-info/PKG-INFO
src/juml_framework.egg-info/SOURCES.txt
src/juml_framework.egg-info/dependency_links.txt
src/juml_framework.egg-info/entry_points.txt
src/juml_framework.egg-info/requires.txt
src/juml_framework.egg-info/top_level.txt
tests/test_cli_args.py
tests/test_test_utils.py