array-api
array-api copied to clipboard
data-apis
RFC document, tooling and other content related to the array API standard
This PR updates to include dlpack C exchange api for fast exchange at c-extension level without going through python.
This is a cross ref RFC on DLPack based exchange. As of now, DLPack exchange relies on python functions such as `tensor.__dlpack__()`. While they works well for common cases, the...
This PR - resolves https://github.com/data-apis/array-api/issues/893 by adding support for `broadcast_shapes` to the specification. - follows NumPy et al in supporting an arbitrary number of input shapes to be broadcasted. -...
Following up on an RFC at https://github.com/data-apis/array-api/issues/935, add non-symmetric eigenvalue solvers, `eig` and `eigvals`, to the `linalg` extension. The routines are available in `numpy`, `torch`, `jax.numpy` and `cupy` (as of...
This PR - resolves https://github.com/data-apis/array-api/issues/938 - updates `meshgrid` to return a `tuple` rather than a list. This follows NumPy and PyTorch. - updates `broadcast_arrays` to return a `tuple` rather than...
*Copied and adapted from @kgryte's proposal at gh-177 per https://github.com/data-apis/array-api/issues/177#issuecomment-2883011323* #### Proposal Add APIs for setting elements via an array of indices. #### Motivation Currently, the array API specification does...
In `xp.sort` and `xp.argsort`, the sort is stable by default. I understand very well the interest for argsort: I've seen a lot of bugs because people expected `np.argsort` to be...
Previously we landed stream=None mapping to legacy default stream (a safer case). As DLPack get popularized, one most canonical use-case is to exchange between library and pytorch. As most libraries...
It would be useful for users to have a list of implementations of the standard, with a brief description of the implementation and version of the array API that is...
I always wonder how to design a vectorizable function when the sizes of the input and output arrays are different. In other words, I am always unsure whether to shift...
