Update README.md

Author: heptaflar

README.md

@@ -141,14 +141,14 @@ these indices. The code below provides an implementation.
     :returns: The law of propagation of uncertainty.
     """
 
+    @jax.jit
     def lpu(g: Array, u: Array) -> Array:
         """The law of propagation of uncertainty."""
         dims = tuple(range(-d, 0))
         return jnp.tensordot(jnp.tensordot(g, u, (dims, dims)), g, (dims, dims))
 
     return lpu
 
-
 [![CodeQL Advanced](https://github.com/bcdev/uncertaintyx/actions/workflows/codeql.yml/badge.svg)](https://github.com/bcdev/uncertaintyx/actions/workflows/codeql.yml)
 [![Python package](https://github.com/bcdev/uncertaintyx/actions/workflows/python-package.yml/badge.svg)](https://github.com/bcdev/uncertaintyx/actions/workflows/python-package.yml)
 [![codecov](https://codecov.io/gh/bcdev/uncertaintyx/graph/badge.svg?token=742AWtYDCD)](https://codecov.io/gh/bcdev/uncertaintyx)