Brent Yi

Actually, this is a little more complicated than I first imagined. Just pushed another test; this one fails: ```python from typing import Generic, TypeVar T = TypeVar("T") @dataclass class Z:...

@davidhalter thanks for the quick response! I spent a bit more time looking at this and pushed a slightly less specific fix, which just knows to iterate into `ValueWrapper` and...

@davidhalter any chance you could take a second look at this PR? It'd still be super helpful for myself + some collaborators. (though I totally understand if you'd prefer not...

Sounds sensible; appreciate the update!

Sorry for the delay in getting back here! My general experience is that `jaxfg` will usually be slower than optimizers written in C++, particularly if you factor in JIT compilation...

Hm, that does sound much slower than comparable experiments we've run. Would appreciate the opportunity to take a look if you're open to sharing your code.

Thanks @AvanDavad!! Will try to get this merged soon, just gotta go through the typing/mypy errors. (which I can do)

Hi! Yes, this is something I can look into. Do you have suggestions for a similar example using a different library/framework/language? The easiest thing would probably be to just reproduce...

Hey, sorry for missing this! Can you clarify (or link some references on) the expected behavior of the linear space operators that you mentioned? This isn't ringing any bells and...

So behavior like this? ```python from typing import TypeVar import jaxlie import jax T = TypeVar("T", bound=jaxlie.MatrixLieGroup) def add(x: T, y: T) -> T: return type(x)(x.parameters() + y.parameters()) # alternatively:...