First pass at a quickcheck-style property-based tester to test invariants of jit/grad/vmap etc on random functions
First pass at a quickcheck-style property-based tester to test invariants of jit/grad/vmap etc on random functions

google/jax,google/jax,tensorflow/probability,tensorflow/probability,google/jax,google/jax