Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,10 @@

Long-time users may remember that **PyAutoLens** previously used `numba`, which recompiles individual Python
functions into fast machine code. JAX supersedes it: rather than accelerating functions one at a time, it compiles
and optimizes the likelihood function end-to-end, and adds GPU support and batching on top.
and optimizes the likelihood function end-to-end, and adds GPU support and batching on top. The one exception is
pixelized-source modeling on a CPU, where the `numba` sparse-operator path (see
`autolens_workspace/scripts/imaging/features/pixelization/cpu_fast_modeling.py`, and chapter 3) remains the faster
route on many-core machines.

Therefore, **PyAutoLens** is pretty well optimized and there are no 'low hanging fruit' speed ups available by
writing the code in a different language.
Expand Down
1 change: 1 addition & 0 deletions workspace_index.json
Original file line number Diff line number Diff line change
Expand Up @@ -373,6 +373,7 @@
"Wrap Up"
],
"cross_refs": [
"autolens_workspace/scripts/imaging/features/pixelization/cpu_fast_modeling.py",
"chapter_optional/tutorial_searches.py"
],
"notebook": "notebooks/chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb",
Expand Down
Loading