Major Upgrade after HeLiMOS (#38)

* First draft

* update readme

* update readme

* WIP: visualizer

* Add polyscope visualizer

* Propagate parameters

* Fix this

* Small change

* Limit mos range

* Update readme

* Update

* This fixes error

* Minor changes

* Format

* Renaming

* Fix

Update pytorch-lightning, pytorch, and docker compose (#34)

* Remove deprecated gpus argument

* Formatting

* Fix epoch collectors

* Fix device error

* Use latest pytorch and cuda

* Need to use docker-compose, see docker/compose#9681

* This works but it's not ideal

* Better, at least as long as buildkit does not allow access to the GPU
during build.

See docker/compose#9681

* Relax these