mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-26 08:46:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
23,091 Commits 508 Branches 263 Tags
d4d1518c3d
Commit Graph

27 Commits

Author SHA1 Message Date
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
Jake VanderPlas
25da7add37 Add method argument for jnp.isin 2024-08-13 19:04:14 -07:00
Jake VanderPlas
52c269c8cd jnp.setxor1d: add support for static size argument 2024-08-13 05:24:59 -07:00
Jake VanderPlas
c2c116dc5c jnp.intersect1d: add support for static size argument. 2024-08-10 05:22:05 -07:00
Dan Foreman-Mackey
7857bd3319 Fix compatibility of jnp.unique with numpy nightly 
In https://github.com/numpy/numpy/pull/26914, the behavior of the
`return_inverse` argument to `np.unique` was partially reverted to the
pre-v2.0 behavior. The PR brings JAX's implementation compatible with
the `numpy>2.0.0` behavior.
 2024-07-15 11:41:00 -04:00
Jake VanderPlas
3cff806496 fix doc typo 2024-06-27 06:50:41 -07:00
rajasekharporeddy
b93da3873b Fix Typos 2024-06-17 13:55:46 +05:30
Ayaka
ef22b465f2 Fix typo 2024-06-04 17:39:13 +04:00
Jake VanderPlas
7ed7780b96 Improve docs for jax.numpy set-like operations 2024-05-13 15:19:14 -07:00
Jake VanderPlas
43a9faa06a Rename _wraps to implements 2024-01-24 14:14:19 -08:00
Jake VanderPlas
fa6d3f26ff jnp.unique: make return_inverse shape match NumPy 2.0 2024-01-16 11:47:45 -08:00
Jake VanderPlas
5e957c6063 array api: add unique_* interfaces 2023-12-21 15:49:54 -08:00
Jake VanderPlas
e3e26f2dde jnp.unique: add support for the equal_nan keyword 2023-12-21 12:37:09 -08:00
Jake VanderPlas
e356d76913 Remove a number of deprecated APIs 
All of these were deprecated prior to the JAX 0.4.16 release, on Sept 18 2023.
As of Monday Dec 18, we have met the 3 month deprecation period specified by the [API Compatiblity Policy](https://jax.readthedocs.io/en/latest/api_compatibility.html).

PiperOrigin-RevId: 591933493
 2023-12-18 10:08:47 -08:00
Sergei Lebedev
f936613b06 Upgrade remaining sources to Python 3.9 
This PR is a follow up to .

The changes were generated by adding

    from __future__ import annotations

to the files which did not already have them and running

    pyupgrade --py39-plus --keep-percent-format {jax,tests,jaxlib,examples,benchmarks}/**/*.py
 2023-12-13 10:29:45 +00:00
Jake VanderPlas
caf930e467 jnp.unique: small implementation cleanup 2023-11-09 14:24:43 -08:00
Peter Hawkins
816ba91263 Use lower-case PEP 585 names for types. 
Issue https://github.com/google/jax/issues/16537

PiperOrigin-RevId: 542969282
 2023-06-23 15:12:14 -07:00
Peter Hawkins
f48dbf039e Use cast instead of pytype suppression in setops.py. 
pytype cannot tell from the type signature that unique() returns an Array, not a tuple. Add a cast to help it along.

It's possible that a future use of @overload on the definition of jnp.unique might help.

PiperOrigin-RevId: 520389675
 2023-03-29 11:17:00 -07:00
Rebecca Chen
5ae2e79d43 Silence some pytype errors. 
PiperOrigin-RevId: 520150523
 2023-03-28 15:24:48 -07:00
Jake VanderPlas
760deb310e Remove leading underscores in jax._src.numpy.util 2023-03-13 12:18:36 -07:00
Jake VanderPlas
c8c269f5f5 internal: avoid unused imports in lax_numpy 2023-03-08 10:29:04 -08:00
Peter Hawkins
8fb1fd318d Replace jax._src.util.prod with math.prod. 
math.prod() was added in Python 3.8, so we can assume it is always present.

PiperOrigin-RevId: 513011144
 2023-02-28 12:41:00 -08:00
Roy Frostig
cb8dcce2fe migrate more internal dependencies from jax.core to jax._src.core 
PiperOrigin-RevId: 509736368
 2023-02-14 23:01:11 -08:00
Jake VanderPlas
fd45035b21 [typing] add full annotations for lax_numpy setops 2022-10-03 12:52:28 -07:00
Peter Hawkins
ba557d5e1b Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.". 
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.

PiperOrigin-RevId: 476167538
 2022-09-22 12:27:19 -07:00
Jake VanderPlas
06c3857321 jnp.unique: improve error when run under JIT 2022-06-08 15:57:41 -07:00
Jake VanderPlas
466bea1662 lax_numpy: refactor set operations into separate private submodule 2022-03-21 09:38:11 -07:00