Jake VanderPlas
7975192f92
Expose jax.typing & update docs
2023-02-13 15:53:08 -08:00
Peter Hawkins
a28b01243b
Move contents of jax.monitoring to jax._src.monitoring.
...
PiperOrigin-RevId: 508191560
2023-02-08 15:03:22 -08:00
Peter Hawkins
cc8d7fae32
Move jax.interpreters.mlir to jax._src.interpreters.mlir.
...
Replace jax.interpreters.mlir with a shim that re-exports names that are likely to be used externally.
PiperOrigin-RevId: 508187063
2023-02-08 14:39:01 -08:00
Roy Frostig
55c2b6dad6
move jax.interpreters.batching to jax._src.interpreters.batching
...
Re-export roughly all of the same symbols via `jax.interpreters.batching` for now.
PiperOrigin-RevId: 508107044
2023-02-08 09:51:00 -08:00
Peter Hawkins
6860cb8d2a
Move jax.interpreters.xla to jax._src.interpreters.xla.
...
Replace jax.interpreters.xla with a shim that re-exports names that are likely to be used externally.
PiperOrigin-RevId: 507895040
2023-02-07 15:01:32 -08:00
Peter Hawkins
38a59a313b
Move jax.interpreters.pxla to jax._src.interpreters.pxla.
...
Make jax.interpreters.pxla a shim that at the moment re-exports everything in the implementation, with the goal of reducing it over time.
PiperOrigin-RevId: 507584264
2023-02-06 14:29:10 -08:00
Peter Hawkins
365262b77a
Reapply: move jax.interpreters.ad to jax._src.interpreters.ad
...
Re-export roughly all of the same symbols via `jax.interpreters.ad` for now.
This version of the PR includes the names jax.interpreters.ad.source_info_util and jax.interpreters.ad.config, which the neural tangents is using.
PiperOrigin-RevId: 506642132
2023-02-02 09:29:05 -08:00
Peter Hawkins
04525e896e
Revert: move jax.interpreters.ad to jax._src.interpreters.ad
...
Re-export roughly all of the same symbols via `jax.interpreters.ad` for now.
This change broke some tests.
PiperOrigin-RevId: 506606721
2023-02-02 06:52:47 -08:00
Roy Frostig
0e77af0a28
move jax.interpreters.ad to jax._src.interpreters.ad
...
Re-export roughly all of the same symbols via `jax.interpreters.ad` for now.
PiperOrigin-RevId: 506490796
2023-02-01 19:46:47 -08:00
Jake VanderPlas
4a6bbde409
Move jax.linear_util to jax._src.linear_util
2022-12-20 14:49:27 -08:00
Peter Hawkins
aacf44ae3a
flake8 now rejects inline comments.
...
See:
https://flake8.pycqa.org/en/latest/user/configuration.html (search for
"inline comments").
2022-12-08 16:22:28 +00:00
Jake VanderPlas
7f89fd40a2
Cleanup: remove unused imports in private modules
...
Also improve our flake8 filter rules to avoid ignoring these.
2022-10-20 14:37:21 -07:00
Jake VanderPlas
5782210174
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
Peter Hawkins
b232d09440
Enable flake8 checks for spaces around operators.
2021-07-30 08:45:38 -04:00
Matthew Johnson
3595af2ed2
add linearize, vjp, grad. fix bugs.
2021-03-26 21:27:37 -07:00
Jake Vanderplas
6aa8f2461c
Fix remaining flakes and use exclude within setup.cfg ( #3371 )
2020-06-08 22:58:03 -07:00
Jake Vanderplas
2a10dbbf37
deflake remainder of jax ( #3343 )
2020-06-06 10:51:34 -07:00
Jake Vanderplas
fb1717233a
Cleanup: deflake jax.experimental and jax.ops ( #3329 )
2020-06-05 19:00:04 -07:00
Jake Vanderplas
a63b9cc256
Cleanup: deflake interpreters, lib, nn, third_party, and tools ( #3327 )
2020-06-04 15:27:48 -07:00
Jake Vanderplas
bc51e9c7f6
deflake jax/scipy/* and add to setup.cfg ( #3316 )
2020-06-04 14:38:41 -07:00
Jake Vanderplas
b187663a87
deflake jax/lax & add to flake8 check ( #3310 )
2020-06-04 13:50:44 -07:00
Jake Vanderplas
c77c0838fe
deflake jax.numpy and add to flake8 check ( #3312 )
2020-06-03 14:18:48 -07:00
Jake Vanderplas
9ee4ef1107
Cleanup: de-lint tests directory & add flake8 to travis ( #3304 )
...
* Cleanup: fix lint errors in tests/*.py
* Add flake8 step to travis
* add setup.cfg
2020-06-02 19:25:47 -07:00
