guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-24 19:50:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
JaxPM/jaxpm/_src
History
Wassim KABALAN 1cade569af fixes for SPMD config
2024-07-09 02:35:37 +02:00
..
base_ops.py Add transpose for single GPU 2024-07-09 02:35:13 +02:00
painting_ops.py Adjust painting operators 2024-07-09 02:34:49 +02:00
spmd_config.py fixes for SPMD config 2024-07-09 02:35:37 +02:00