-
I will admit i am not confident this is in fact a bug. Still it seems like odd behaviour and it took me a few hours to pin-point the NaN-source.
Minimal reproducing example:
```python
import jax…
-
## The problem:
Currently, when you compute the reverse mode derivative of some function of a `struct MyStruct` in enzyme, i.e. `autodiff(Reverse, f, Duplicated(A::MyStruct, dA::MyStruct))`, the o…
-
I figured out how to do a relatively fast (on the order of the inversion of the gradient finding step [the backpropagation or reverse mode autodiff or whatever] as opposed to the naive order of the gr…
-
Hello,
When solving the (trivial) SDE $d y_t = -y_t\ dt + 0.2\ dW_t$, the Diffrax Euler solver is ~200x slower than a naive for loop. Am I doing something wrong? The speed difference is consistent …
-
Right now Klara is not usable under 1.0.
One reason is the dependency on ReverseDiff, which is not yet ported as well :(
OT: Is there any gradient-based MCMC solution that I can use for now?
-
Currently, scikit-hep phasespace is used to produce phase space decays. It uses Tensroflow making HelixGen dependent on Tensorflow, too. Once phasespace generation is implemented with Jax, HelixGen wi…
-
I recently came across this paper from last years NeurIPS
https://ghliu.github.io/assets/pdf/neurips-snopt-slides.pdf
https://github.com/ghliu/snopt
I was wondering if there are any plans to …
-
Hi Enzyme Team,
I'm trying to compute a hessian (or a hessian-vector product) using Forward-over-Reverse, doing my best to follow what's in the documentation, but I'm crashing and can't tell if I'v…
-
I did fast look too the code. And my first idea to add allocator, but I'm not sure. @allanleal If you have any ideas, please tell =)
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
source
### TensorFlow version
2.14.0-dev20230630
### Custom code
Yes
### OS platform and distributi…