Merge pull request #240 from kazewong/kazewong-patch-1

Update README.md

Merge pull request #236 from kazewong/flowMatchingSMC

Merge pull request #235 from kazewong/add-whitening-to-flow-matching

Format test_flowmatching.py for improved readability

Update test_flowmatching.py

add tests

Merge pull request #233 from kazewong/add-whitening-to-flow-matching

Refactor flow matching code for readability and add diffrax to

Add whitening procedure to FlowMatchingModel

Update train_flow_match.ipynb

Update flowMC log_prob notebook

Add log_prob function

Update train_flow_match.ipynb

Add dt parameter to sample method in FlowMatchingModel