Added OTFlow Layers
Checklist
- [ ] Appropriate tests were added
- [ ] Any code changes were done in a way that does not break public API
- [ ] All documentation related to code changes were updated
- [ ] The new code follows the contributor guidelines, in particular the SciML Style Guide and COLPRAC.
- [ ] Any new documentation only uses public API
Additional context
Add any other context about the problem here.
Addresses issue #350
This PR includes the implementation OTFlow: Fast and Accurate Continuous Normalizing Flows via Optimal Transport as mentioned in the paper https://arxiv.org/abs/2006.00104
There are some issues with code in GPUs.md in the examples directory. No changes were made to that file and this PR does not address that issue
Rebase
Needs tests
Needs tests
Sure. Will implement that
@avik-pal all the tests here have passed. Could you please approve the other workflows, to see if this is fine ?
@avik-pal Would you please review if there are any changes to make ??