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 a0be772f3c Add transpose for single GPU
2024-07-09 02:35:13 +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 Create operator infra 2024-07-08 00:23:12 +02:00