jaxmpi:master commitshttps://code.itp.ac.cn/codes/jaxmpi/-/commits/master2022-06-16T11:15:27+08:00https://code.itp.ac.cn/codes/jaxmpi/-/commit/872ba13074c8258b0b6e0919d84e8d70d3213af2& and wait2022-06-16T11:15:27+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/782f9665d5c62c919a8190ad2f5856ca9cba01d8is it working?2022-06-16T08:27:20+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/eca2ddbc3c5547d880cb31fc000992f420e03055multihost_pmap2022-06-16T01:19:34+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/9361c6645fa2b5d5ec26e35d1b7553cce4c717c4mnist using multi-host pmap2021-11-03T23:40:30+08:00Hao Xiexiehao18@mails.ucas.edu.cnhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/a0b5fa092606d0f838a56c4ba882acb7f36f5b26cuda-aware mpi on delta1032021-10-27T21:27:02+08:00Hao Xiexiehao18@mails.ucas.edu.cnhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/cf6272392d5bbcf925100e3d657c67328d1f45bbclean up2021-10-27T11:31:33+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/08225f3a705e5443cc5681139bd1e5feafaa16dfuse mpi4jax2021-10-27T10:56:36+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/3612290f34d1927cc326afbe4f584b1d5c810b79unravel and then transfer via numpy on cpu2021-10-26T16:38:50+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/4e5a2cf0e8b7c71ecd753ea12d7208e52156dc9edivide data2021-10-26T16:29:24+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/000b96fec8d0c232348a22e8d75c01ae69cf6d63mpi_mnist2021-10-26T16:15:22+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/2c2ebd90afae36dfa14ccff5bf9bd4a35cd1a112add pmap_mnist example2021-10-25T15:05:27+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/c239da2bdc7cc706e4b12890e782bf47336dbf02refs2021-10-25T14:44:17+08:00Lei Wangwangleiphy@gmail.comhttps://code.itp.ac.cn/codes/jaxmpi/-/commit/a0d00eb5ed154fa53a855f5dd949fa59cd4aa2feinitial tests2021-10-25T14:24:25+08:00Lei Wangwangleiphy@gmail.com