Add aboucaud comments

DifferentiableUniverseInitiative · Dec 8, 2024 · d8c68ac · d8c68ac
1 parent adaf7d2
commit d8c68ac
Show file tree

Hide file tree

Showing 10 changed files with 26 additions and 777 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -29,7 +29,7 @@ jobs:
       run: |
         sudo apt-get install -y libopenmpi-dev
         python -m pip install --upgrade pip
-        pip install jax=0.4.35 
+        pip install jax=0.4.35
         pip install .[test]
 
     - name: Run Single Device Tests

diff --git a/README.md b/README.md
@@ -11,7 +11,7 @@ Provide a modern infrastructure to support differentiable PM N-body simulations
 - Any order forward and backward automatic differentiation
 - Support automated batching using `vmap`
 - Compatibility with external optimizer libraries like `optax`
-- Now fully distributable on **multi-GPU and multi-node** systems using [jaxDecomp](https://github.com/DifferentiableUniverseInitiative/jaxDecomp) working with the latex `JAX v0.4.35`
+- Now fully distributable on **multi-GPU and multi-node** systems using [jaxDecomp](https://github.com/DifferentiableUniverseInitiative/jaxDecomp) working with`JAX v0.4.35`
 
 
 ## Open development and use

diff --git a/benchmarks/bench_pm.py b/benchmarks/bench_pm.py