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...

WIP: top_k draft implementation

2

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 >>>...