Jake VanderPlas
|
200f826398
|
[array api] return all devices in devices()
|
2025-03-31 08:50:39 -07:00 |
|
Jake VanderPlas
|
096810a721
|
[array API] make capabilities more accurate
|
2025-03-26 12:11:47 -07:00 |
|
Jake VanderPlas
|
c7b0d681bd
|
Remove deprecated jax.experimental.array_api
|
2025-01-06 15:19:02 -08:00 |
|
Jake VanderPlas
|
48c5fab023
|
[array api] fix deprecation to support old import pattern
|
2024-08-01 14:38:59 -07:00 |
|
Jake VanderPlas
|
14fa06298e
|
[array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api
|
2024-08-01 11:19:17 -07:00 |
|
Meekail Zain
|
b88e2e808b
|
Refactor array_api namespace, relying more directly on jax.numpy
|
2024-05-02 18:17:45 +00:00 |
|
Meekail Zain
|
28c76c0187
|
Expose tile function in array_api namespace
|
2024-04-26 17:34:58 +00:00 |
|
Meekail Zain
|
ceeb975735
|
Add new cumulative_sum function to numpy and array_api
|
2024-04-16 19:57:55 +00:00 |
|
Meekail Zain
|
6bdc83c680
|
Add new unstack function to numpy/array_api namespaces
|
2024-04-15 21:03:26 +00:00 |
|
jax authors
|
5f22b12576
|
Merge pull request #20754 from Micky774:array-api-hypot
PiperOrigin-RevId: 625035601
|
2024-04-15 11:56:53 -07:00 |
|
Meekail Zain
|
2899213efb
|
Fixed hypot bug on nan/inf pairings, began deprecation of non-real values
|
2024-04-15 17:56:16 +00:00 |
|
Meekail Zain
|
8b93da1830
|
Expose existing functions in array API namespace
|
2024-04-15 16:25:30 +00:00 |
|
jax authors
|
301c3518d8
|
Merge pull request #20294 from Micky774:array_namespace_info
PiperOrigin-RevId: 623877931
|
2024-04-11 11:09:37 -07:00 |
|
Meekail Zain
|
e6508a4f47
|
Add __array_namespace_info__ and corresponding utilities
|
2024-04-11 14:20:44 +00:00 |
|
Meekail Zain
|
8b7aae586b
|
Update jnp.clip to Array API 2023 standard
|
2024-04-04 22:55:10 +00:00 |
|
Jake VanderPlas
|
0c7b959dac
|
jnp.linalg: add matrix_norm, matrix_transpose, vector_norm, vector_transpose
These have been added upstream to numpy.linalg in NumPy 2.0, as part of the Array API standard.
|
2023-12-15 14:17:36 -08:00 |
|
Jake VanderPlas
|
0d073a4862
|
[array-api] add simple smoketest target for standard CI testing
|
2023-11-27 15:20:41 -08:00 |
|