Fix var names and docs to mention numpy instead of jax in vector NumpyToTorch

Seems that NumpyToTorch was made from JaxToTorch, so some artifacts of referring jax instead of numpy in docs and variable names exist.

Author: pkuderov

gymnasium/wrappers/vector/numpy_to_torch.py

@@ -42,7 +42,7 @@ def __init__(self, env: VectorEnv, device: Device | None = None):
         """Wrapper class to change inputs and outputs of environment to PyTorch tensors.
 
         Args:
-            env: The Jax-based vector environment to wrap
+            env: The NumPy-based vector environment to wrap
             device: The device the torch Tensors should be moved to
         """
         super().__init__(env)
@@ -60,8 +60,8 @@ def step(
         Returns:
             The PyTorch-based Tensor next observation, reward, termination, truncation, and extra info
         """
-        jax_action = torch_to_numpy(actions)
-        obs, reward, terminated, truncated, info = self.env.step(jax_action)
+        numpy_action = torch_to_numpy(actions)
+        obs, reward, terminated, truncated, info = self.env.step(numpy_action)
 
         return (
             numpy_to_torch(obs, self.device),
@@ -81,7 +81,7 @@ def reset(
 
         Args:
             seed: The seed for resetting the environment
-            options: The options for resetting the environment, these are converted to jax arrays.
+            options: The options for resetting the environment, these are converted to NumPy arrays.
 
         Returns:
             PyTorch-based observations and info