Correct a typo and some wording in doc/JaxOps.rst

doc/JaxOps.rst

@@ -58,8 +58,9 @@ Here’s an example for the Eye Op.
 .. code:: python
 
    from theano.tensor.basic import Eye
-   @jax_funcify.register(Eye) # The decorater 
-   def jax_funcify_Eye(op): # The function that takes an individual Theano-PyMC Op and returned the JAXified op
+
+   @jax_funcify.register(Eye) # The decorator
+   def jax_funcify_Eye(op): # The function that takes an Op and returns its JAX equivalent
        dtype = op.dtype
 
        def eye(N, M, k):