Closed sunnyayyl closed 3 months ago
See #1136, this makes jax array treated and handled the same way numpy ndarray get treated
(Apologies in advance for any rookie mistake I make, first time making a pull request and first time contributing to anything in general)
As a comparison
Cool, thanks @sunnyayyl !
See #1136, this makes jax array treated and handled the same way numpy ndarray get treated
(Apologies in advance for any rookie mistake I make, first time making a pull request and first time contributing to anything in general)