PyTorch simulations are SLOW, trying Jax

norabelrose · Aug 28, 2021 · b493665 · b493665
1 parent cb6e804
commit b493665
Show file tree

Hide file tree

Showing 11 changed files with 834,776 additions and 99,439 deletions.
diff --git a/Playground.ipynb b/Playground.ipynb
diff --git a/pathfinder/__init__.py b/pathfinder/__init__.py
diff --git a/pathfinder/constants.py b/pathfinder/constants.py
diff --git a/pathfinder/jax/__init__.py b/pathfinder/jax/__init__.py
diff --git a/pathfinder/jax/control.py b/pathfinder/jax/control.py
diff --git a/pathfinder/kepler.py b/pathfinder/kepler.py
diff --git a/pathfinder/problems/low_thrust_transfer.py b/pathfinder/problems/low_thrust_transfer.py
diff --git a/pathfinder/pytorch/__init__.py b/pathfinder/pytorch/__init__.py
diff --git a/pathfinder/pytorch/control.py b/pathfinder/pytorch/control.py
diff --git a/pathfinder/pytorch/ephemeris.py b/pathfinder/pytorch/ephemeris.py
diff --git a/pathfinder/utils.py b/pathfinder/utils.py