Adds a trivial jaxpm implementation

This commit is contained in:
EiffL 2022-02-13 21:36:03 +01:00
parent d4d0a03c79
commit afdfcfa644
5 changed files with 714 additions and 0 deletions

567
dev/JaxPM_ODE.ipynb Normal file

File diff suppressed because one or more lines are too long