Farama-Foundation / Jumpy

On-the-fly conversions between Jax and NumPy tensors
Apache License 2.0
46 stars 9 forks source link

Change jumpy to use same import statements as jax and numpy #26

Closed pseudo-rnd-thoughts closed 1 year ago

pseudo-rnd-thoughts commented 1 year ago

Jumpy simplifies the linalg and random functions to not be identical to numpy or jax For example, np.linalg.norm is jp.linalg_norm in jumpy. This makes it more complex users to change over to jumpy from jax or numpy.

Therefore, I propose adding a linalg and random file for these functions.

Additionally, if we are splitting jumpy functions into multiple functions, we can look to divide up the jumpy.py to several subfiles for different types of functions.

pseudo-rnd-thoughts commented 1 year ago

Done in #30