ivy-llc / ivy

Convert Machine Learning Code Between Frameworks
https://ivy.dev
Other
14.01k stars 5.76k forks source link

Add manipulation functions to Jax frontend #8422

Open fnhirwa opened 1 year ago

fnhirwa commented 1 year ago

These functions can be found in jax.numpy namespace

_

Please keep in mind that the proper way to link an issue to this list is to comment "- [ ] #issue_number" while the issue's title only includes the name of the function you've chosen. _

The main file paths where these functions are likely to be added are:

ivy/functional/frontends/jax/numpy/manipulations.py ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_jax\_numpy\_manipulation.py ivy\_tests/test\_ivy/test\_functional/test\_experimental/test\_core/test\_manipulation.py ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_numpy/test\_manipulations.py ivy/functional/frontends/jax/numpy/name\_space\_functions.py ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_jax\_numpy\_namespace\_functions.py

x0axz commented 1 year ago

@hirwa-nshuti jax.numpy.expand_dims is already implemented. https://github.com/unifyai/ivy/blob/master/ivy/functional/frontends/jax/numpy/manipulations.py#L76