13 Commits

Author	SHA1	Message	Date
Shuhan Ding	1aa32f51ee	adopt jax.extend.backend	2024-10-07 12:27:35 -07:00
Peter Hawkins	fc4f554e09	Delete jax.lib.xla_client.execute_with_python_values. ... Nothing under jax.lib.xla_client is public, so there's no deprecation period required. PiperOrigin-RevId: 681166972	2024-10-01 14:32:22 -07:00
Michael Hudgins	d4d1518c3d	Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax ... PiperOrigin-RevId: 676843138	2024-09-20 07:52:33 -07:00
Shuhan Ding	21951db84a	adopt numpy rank promotion config in metal test	2024-07-18 10:57:43 -07:00
Peter Hawkins	7f24837eef	Update minimum NumPy version to v1.24.	2024-06-21 15:17:17 -07:00
Shuhan Ding	28fe45d872	metal_plugin ci with jaxlib nightly	2024-05-02 18:40:01 -07:00
Shuhan Ding	99e5b8e999	enable conv1d test	2024-05-02 18:20:39 -07:00
Shuhan Ding	0c517f2d83	update along with lax_numpy_test	2024-05-02 17:18:55 -07:00
Jake VanderPlas	3f6c4fc60d	CI: run the metal CI nightly & on PRs touching the config file	2024-04-11 11:18:27 -07:00
Meekail Zain	8b7aae586b	Update jnp.clip to Array API 2023 standard	2024-04-04 22:55:10 +00:00
Peter Hawkins	cf856ad4a9	Reverts 8e2a8b7b95e838947dcf581d146909d5c4128742 ... PiperOrigin-RevId: 615401711	2024-03-13 07:01:49 -07:00
Shuhan Ding	4c4bcde723	pass set of lax_numpy_test	2024-03-11 11:18:58 -07:00
Shuhan Ding	9bdad4debe	setup ci for jax-metal plugin	2024-03-06 10:43:13 -08:00