Sharique Shamim
Sharique Shamim
Hi @keawchot, the arguments in the frontend should be similar to https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.gather.html and you should accordingly implement the function compositionally.
@JonathanYang-yzh is working on this.
Hi @karalleyna, a lot of changes from other commits are reflected here which are not required for `qr`. I would suggest you to create a new PR for your function...
Hi @hanafganesa, you need to write tests for your function as well. You shouldn't be committing changes for `.idea` files. Also, please resolve the lint errors.