Have the package at a more stable state with the different components set up nicely. More specifically:

• Have a nice setup for the different diffusions (euler solvers, splitting schemes)
• Clear documentation for how to use the package, how the samplers work, and how to add a new sampler
• learning rate schedules
• some basic tests
• An example for how the library can work easily with a framework like Flax