Patrick Kidger
Patrick Kidger
Awesome :) Yup send a PR! The ecosystem lists appear in `README.md` and in `docs/index.md`.
Ah, good catch! Maybe we should just remove the `has_annotated_args or has_annotated_return` call? I don't think it's important, it was just a minor efficiency thing. I'd be happy to take...
Closing as accomplished in #205, which corresponds to jaxtyping version 0.2.29.
Haha, I love the pun in the name! On balance I'm inclined to leave these in diffraxtra. I think utilities like these are a great example of how people can...
I think this is a beartype thing: it only checks one element of a list. CC @leycec
How weird! One could potentially use py-spy to attach to the process and observe the stack trace of the stuck program.
Very nice! :D Thankyou for writing this down – this will be an excellent example to share for solving with mass matrices. ^^
This is disappointingly expected, I'm afraid. JAX has made a number of performance pessimisations over the past year or so. I think @lockwo had collected some of these examples /...
Looks like a minor issue in JAX; isolated and opened the GitHub issue above. Not the end of the world for us if it's not fixed - we just stop...
Here's a MWE without Lineax, using just JAX: ```python import jax import jax.numpy as jnp def test_func(x): return jnp.abs(x)**2 x = jnp.ones((3, 3)) jac_func = jax.jacrev(test_func) jax.linear_transpose(jac_func, jax.ShapeDtypeStruct((3,), jnp.float32))(x) ```...