0.0.10

fix warmup period of 0 for optimizer with warmup

0.0.9

decorator that automatically unwraps a model with accelerator instanc…

0.0.8

use optree

0.0.7

add another decorator that autocasts inputs into a method onto the de…

0.0.6

handy context manager for running a lot of wandb experiments with a t…

0.0.5

bundle warmup, scheduler, gradient clipping and optimizer into one class

0.0.4

simple fn for getting adam optimizer with maybe weight decay, and omi…

0.0.3

quick device access

0.0.2

quick device access

0.0.1

start drying some software engineering stuff outside of the actual re…