Slightly cleaner implementation of jvp for det
Slightly cleaner implementation of jvp for det

Replaced np.dot/np.linalg.inv with a single np.linalg.solve
google/jax,google/jax,tensorflow/probability,tensorflow/probability,google/jax,google/jax