Skip to content

Trying jax for gradients#24

Open
jmlarson1 wants to merge 20 commits intomainfrom
feature/jax_for_gradients
Open

Trying jax for gradients#24
jmlarson1 wants to merge 20 commits intomainfrom
feature/jax_for_gradients

Conversation

@jmlarson1
Copy link
Copy Markdown
Collaborator

@jmlarson1 jmlarson1 commented Sep 30, 2025

Our python gradients/jacobians were computed in matlab by AdiMAT and then translated to python.

We can also compute them using Jax. This allows us to easily compute Hessians (if desired).

@jmlarson1 jmlarson1 marked this pull request as ready for review October 29, 2025 18:18
@jmlarson1 jmlarson1 requested a review from mmenickelly October 29, 2025 18:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant