Jake VanderPlas
|
f652b6ad6a
|
Set __module__ attribute for objects in jax.numpy
|
2024-11-15 06:03:54 -08:00 |
|
Jake VanderPlas
|
aa551e66c5
|
Test that jax.numpy docstrings include examples
|
2024-09-21 07:39:17 -07:00 |
|
Jake VanderPlas
|
3833c46d10
|
jnp.vectorize: respect numpy_rank_promotion config
|
2024-07-08 09:03:03 -07:00 |
|
Peter Hawkins
|
7f4ef63cd8
|
Run pyupgrade --py310-plus.
Also apply manual fixes to import sorting and unused imports.
|
2024-06-26 16:10:18 -04:00 |
|
Sergei Lebedev
|
f5617d7323
|
Removed noop # type: ignore comments
mypy should now flag these by default.
|
2024-05-19 21:01:29 +01:00 |
|
Jake VanderPlas
|
ad3d743ca9
|
jnp.vectorize: support excluding arguments by keyword
|
2023-12-20 15:38:19 -08:00 |
|
Jake VanderPlas
|
a9452b98a3
|
jnp.vectorize: support None arguments
|
2023-11-10 14:25:42 -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 |
|
Giacomo Petrillo
|
95a5b4e48a
|
avoid unnecessary broadcasting in jax.numpy.vectorize
|
2023-03-07 11:56:34 +01: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 |
|
Jeppe Klitgaard
|
17de89b16a
|
feat: refactor code using pyupgrade
This PR upgrades legacy Python code to 3.7+ code using pyupgrade:
```sh
pyupgrade --py37-plus --keep-runtime-typing **.py
```
a
|
2022-05-17 22:14:05 +01:00 |
|
Matthew Johnson
|
d9dcd1394a
|
djax: let make_jaxpr build dyn shape jaxprs
|
2022-02-01 00:10:21 -08:00 |
|
Peter Hawkins
|
4e21922055
|
Use imports relative to the jax package consistently, rather than .-relative imports.
This is more consistent, since currently we use a mix of both styles. It may also help pytype yield more accurate types.
PiperOrigin-RevId: 412057514
|
2021-11-24 07:48:29 -08:00 |
|
elliotwaite
|
7392a57b75
|
DOC: many small fixes
|
2021-08-04 16:55:13 -07:00 |
|
Peter Hawkins
|
26e9ebcdae
|
Move jax.api to jax._src.api.
PiperOrigin-RevId: 368233837
|
2021-04-13 09:43:24 -07:00 |
|
Jake VanderPlas
|
8e789c7380
|
Run doctest on all source files except jax2tf
|
2021-04-05 10:39:59 -07:00 |
|
Peter Hawkins
|
3ac809ede3
|
[JAX] Move jax.util to jax._src_util.
PiperOrigin-RevId: 351234602
|
2021-01-11 14:21:07 -08:00 |
|
Peter Hawkins
|
aa107cf1f4
|
Move jax.numpy internals into jax._src.numpy.
|
2020-10-16 20:35:19 -04:00 |
|