Python check_jvp Examples

Programming Language: Python

Namespace/Package Name: jax._src.test_util

Method/Function: check_jvp

Examples at hotexamples.com: 2

Python check_jvp - 2 examples found. These are the top rated real world Python examples of jax._src.test_util.check_jvp extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

 def test_jvp_linearized(self, f, args):
   jtu.check_jvp(f, partial(jvp_unlinearized, f), args,
                 rtol={np.float32: 3e-2})

Example #2

Show file

 def test_jvp(self, f, args):
   jtu.check_jvp(f, partial(jvp, f), args, rtol={np.float32: 3e-2})