v0.0.2

nyemb

Reset config.yaml

v0.0.3

Implement train_lax with support for mini-batch training. Created an adjusted_ode_loss function that gives more consideration to time intervals far from the initial condition.