Owen L
Owen L
https://github.com/google/jax/issues/196
A few thoughts. TL;DR: Jax/XLA took care of it. First, although I'm no programming language design expert, I'm not sure if equinox objects actually fall under the standard OOP paradigm....
Sure you can construct the hessian from `vmap`, `vjp`, and `jvp` all of which are supported by equinox (https://docs.kidger.site/equinox/api/filtering/transformations/#equinox.filter_jvp).
Here is a simple example, probably not optimized for speed, but just a quick draft to help you get started in a direction. ```python import jax import equinox as eqx...
Indeed, that's all my example does, partitions then uses `jax.hessian` then combines (there is also a flattening operation just to show the matrix inversion).
I've only used numpyro a little, but as Patrick noted, equinox is pretty compatible with most jax libraries because it operates (or you can get it to operate) at a...
It's still in very preliminary stages, but something you might be interested in is distreqx: https://github.com/lockwo/distreqx. I can't get an exact benchmark since when I run the distrax code on...
> If at the output of a measurement I have a density matrix, what would it happen if I have set the circuit to just measure with 1 shot? The...
I can't check what works (the only mac I have access to has an i9), but if the problem for pip installing is TF == 2.4.1, could you work with...
Windows hasn't been supported for a while now. I think TFQ 0.3 was the last version to support it. You can always install that version from source.