6 commits

Author	SHA1	Message	Date
EiffL	61578217fe	Added example code to compute distributed fft	2022-10-17 22:07:12 -07:00
EiffL	1795319e4d	adding hamiltonian gnn demo	2022-04-28 00:20:44 +02:00
EiffL	dbfecd6c41	Adding demo CAMELS notebook	2022-03-21 09:39:02 +01:00
EiffL	25bde5581e	Adds demo and notebooks	2022-02-14 01:59:12 +01:00
EiffL	afdfcfa644	Adds a trivial jaxpm implementation	2022-02-13 21:36:03 +01:00
EiffL	d4d0a03c79	Adds test scrript	2021-08-27 00:52:33 +02:00