Updated Trainer and Documentation

The new trainer now supports numerous runs per checkpoint as well as notebooks for documentation.

Moving to PyTorch Infrastructure

The goal of this change took a lot of manpower and hours. However, the objective panned out and is now entirely built on PyTorch's infrastructure. This includes a PyTorch Dataset and a Trainer class aimed to facilitate research as quickly and easily as possible for computational physicists in the space.

Main Package

Welcome to vit4elm! This is the first commit of the installable repository from PyPi.

You can run the following to install:

pip install vit4elm