LICENSE
README.md
setup.py
benchmarks/__init__.py
benchmarks/benchmark_metrics_per_time_imdb.py
benchmarks/benchmark_metrics_per_time_mnist.py
benchmarks/sparse_softmax_density_epochs_imdb.py
benchmarks/sparse_softmax_mnist_density_epochs.py
psfit/__init__.py
psfit/autodiff.py
psfit/data.py
psfit/exceptions.py
psfit/model.py
psfit/module.py
psfit/ops.py
psfit.egg-info/PKG-INFO
psfit.egg-info/SOURCES.txt
psfit.egg-info/dependency_links.txt
psfit.egg-info/requires.txt
psfit.egg-info/top_level.txt
psfit/backend/__init__.py
psfit/backend/backend_selection.py
psfit/backend/cpu.py
psfit/backend/gpu.py
psfit/optim/__init__.py
psfit/optim/component.py
psfit/optim/distributed.py
psfit/optim/optimizer.py
tests/__init__.py
tests/test_autodiff_backward.py
tests/test_autodiff_forward.py
tests/test_module.py