Conversation
Switches out `numpy` for `jax.numpy`.
When using JAX don't import Numpy
This adds the MGE code needed for the Sersic profile.
Add small radial offset to these two profiles to remove the `nan` values at the center. For the power law profile use the expansion of the `hyp2f1` function given by Tessore and Metcalf 2015 (eqn 29).
Update doc (merge from jax_wrapper branch)
This change makes the MGE profile work better for near-circular mass profiles. This is done by using more terms of the continued fraction for large values of `|z|^2`. This makes the auto grad more accurate and not evaluate to `0` when it should not.
Feature/jax remove preload
Main to JAX
Needed to make the `autolens.Tracer` example work.
Feature/jax speed up general
…tion Feature/jax simplify visualization
…r_adaptive Feature/jax wrapper rectangular adaptive
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Its finally happened!
This PR merges ~ 2 years of JAX development into main, with unit tests passing and integration scripts running.
Full testing will now be done via the build server.