Jules
Source academy requests too much access into a student's Github account, in particular, **requesting read and write permissions to all repository data, including private repositories**. This is unreasonable as a...
This is a follow up of the previous discussions at #15128. I've changed the `mode` keyword (as in previous discussion) into a `largest` bool flag. This follows API such as...
### Description `partition` and `argpartition` return the wrong results for unsigned dtype arrays, and is probably caused by the following [line](https://github.com/google/jax/blob/main/jax/_src/numpy/lax_numpy.py#L6384): `bottom_ind = lax.top_k(-arr, kth + 1)[1]`. POC: ```py >>>...
The purpose of this PR is to continue several threads of discussion regarding `top_k`. This follows roughly the specifications of `top_k` in data-apis/array-api#722, with slight modifications to the API to...
This references the PR data-apis/array-api-tests#274 and implements the compatibility layer for `top_k`. ## Summary of Compatibility * `jax`: * JAX's `top_k` does not implement `axis` or `largest` arguments. While `axis`...