guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-11 21:50:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

JAX version is 0.4.35 until Diffrax new release

Browse source
This commit is contained in:
Wassim Kabalan 2024-12-08 23:13:31 +01:00
parent d8c68ace7a
commit b264da5b17
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/tests.yml vendored
View file

@ -29,7 +29,7 @@ jobs:
run: | run: |
sudo apt-get install -y libopenmpi-dev sudo apt-get install -y libopenmpi-dev
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install jax=0.4.35 pip install jax==0.4.35
pip install .[test] pip install .[test]
- name: Run Single Device Tests - name: Run Single Device Tests