Commit graph

8 commits

Author SHA1 Message Date
EiffL
6ca4c9191e Adding an example of jaxdecomp implementation 2022-11-26 17:27:14 +01:00
EiffL
72ae0fd88f fixed a whole lot of issues 2022-10-22 15:58:32 -04:00
EiffL
3c1abbafcd adds mpi tools 2022-10-22 08:30:43 -04:00
Francois Lanusse
4a584b1d9c
Update jaxpm/painting.py 2022-05-18 10:22:21 +02:00
EiffL
88cff99736 adds fix to make code jittablel 2022-05-17 23:37:55 +02:00
EiffL
e188d5efb6 adding function for doing 2d paintinng 2022-05-17 11:19:56 +02:00
EiffL
d8a5b702a9 adding cic compensation tools 2022-03-26 00:06:34 +01:00
EiffL
afdfcfa644 Adds a trivial jaxpm implementation 2022-02-13 21:36:03 +01:00