Checkpoint working on mlp
Checkpoint
Gradients properly calculating for MLP (I Hope)
Checkpoint, broken
Added test_mlp