Peter Hawkins
e4f3f8f064
Use libtpu releases rather than libtpu-nightly for jax[tpu].
...
PiperOrigin-RevId: 688632409
2024-10-22 11:47:07 -07:00
jax authors
90cd8a79dc
Merge pull request #24290 from jax-ml:dependabot/github_actions/actions/upload-artifact-4.4.3
...
PiperOrigin-RevId: 685764189
2024-10-14 11:00:05 -07:00
dependabot[bot]
93adc0e931
Bump actions/upload-artifact from 4.4.1 to 4.4.3
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.4.1 to 4.4.3.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](604373da63...b4b15b8c7c
2024-10-14 17:42:00 +00:00
dependabot[bot]
0fdd653509
Bump actions/cache from 4.1.0 to 4.1.1
...
Bumps [actions/cache](https://github.com/actions/cache ) from 4.1.0 to 4.1.1.
- [Release notes](https://github.com/actions/cache/releases )
- [Changelog](https://github.com/actions/cache/blob/main/RELEASES.md )
- [Commits](2cdf405574...3624ceb22c
2024-10-14 17:41:55 +00:00
Michael Hudgins
2b4a3af0e5
Update CI-Build job to not reference core count as part of job name
...
PiperOrigin-RevId: 684430824
2024-10-10 07:07:08 -07:00
jax authors
b65be4e1ae
Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1.
...
Set max Python version to 3.13.0 in JAX Kokoro jobs.
PiperOrigin-RevId: 684216757
2024-10-09 16:48:58 -07:00
Michael Hudgins
a7b3a05f11
Update ci-build to use new runners
2024-10-09 18:50:03 +00:00
Peter Hawkins
667221ba34
Move ASAN build to Python 3.13.0.
2024-10-07 21:49:57 -04:00
jax authors
db47e18f39
Merge pull request #24169 from jax-ml:dependabot/github_actions/actions/checkout-4.2.1
...
PiperOrigin-RevId: 683264063
2024-10-07 11:40:19 -07:00
jax authors
6f53e851d4
Merge pull request #24171 from jax-ml:dependabot/github_actions/actions/cache-4.1.0
...
PiperOrigin-RevId: 683262182
2024-10-07 11:35:43 -07:00
dependabot[bot]
eaa099b9b7
Bump actions/cache from 4.0.2 to 4.1.0
...
Bumps [actions/cache](https://github.com/actions/cache ) from 4.0.2 to 4.1.0.
- [Release notes](https://github.com/actions/cache/releases )
- [Changelog](https://github.com/actions/cache/blob/main/RELEASES.md )
- [Commits](0c45773b62...2cdf405574
2024-10-07 17:58:32 +00:00
dependabot[bot]
ab41f4871b
Bump actions/upload-artifact from 4.4.0 to 4.4.1
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.4.0 to 4.4.1.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](50769540e7...604373da63
2024-10-07 17:58:27 +00:00
dependabot[bot]
d083c52ba6
Bump actions/checkout from 4.2.0 to 4.2.1
...
Bumps [actions/checkout](https://github.com/actions/checkout ) from 4.2.0 to 4.2.1.
- [Release notes](https://github.com/actions/checkout/releases )
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md )
- [Commits](d632683dd7...eef61447b9
2024-10-07 17:58:24 +00:00
Adam Paszke
6508b52dab
Bump "oldest supported" libtpu version to fall within the 12w window in CI
2024-10-07 12:27:55 +00:00
Peter Hawkins
44998609e4
Remove extraneous backslash in asan.yaml build.
2024-10-04 11:58:24 -04:00
Peter Hawkins
6e4f8bb838
Fix missing line continuation character in asan build.
2024-10-03 13:36:56 -04:00
Peter Hawkins
0f863b2092
Change asan build to use a self-hosted runner and Ubuntu 24.04.
2024-10-03 12:42:59 -04:00
Peter Hawkins
dea8069fad
Add a nightly ASAN build.
2024-10-03 09:48:33 -04:00
Peter Hawkins
9400ef57a1
Fix Windows CI to install the jaxlib wheel it builds.
...
Currently we install jax first and then try to install jaxlib. But pip won't overwrite the jaxlib installed as a jax dependency, so our CI builds were always using the released jaxlib.
2024-10-02 09:48:14 -04:00
dependabot[bot]
3196b9b153
Bump actions/checkout from 4.1.1 to 4.2.0
...
Bumps [actions/checkout](https://github.com/actions/checkout ) from 4.1.1 to 4.2.0.
- [Release notes](https://github.com/actions/checkout/releases )
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md )
- [Commits](b4ffde65f4...d632683dd7
2024-09-30 17:25:13 +00:00
Jake VanderPlas
f0e27fd73b
actions/checkout: pin specific hash
2024-09-27 10:51:12 -07:00
Jake VanderPlas
cb9f26c8fc
actions: remove use of ratchet tool
...
We no longer use ratchet manually, but rather rely on dependabot
to update these versions when necessary.
2024-09-27 09:22:47 -07:00
dependabot[bot]
d556b59255
Bump actions/setup-python from 5.1.1 to 5.2.0
...
Bumps [actions/setup-python](https://github.com/actions/setup-python ) from 5.1.1 to 5.2.0.
- [Release notes](https://github.com/actions/setup-python/releases )
- [Commits](39cd14951b...f677139bbe
2024-09-25 14:24:13 +00:00
Dan Foreman-Mackey
e1a68eee5e
Add FFI example project and test on CI.
...
This PR includes an end-to-end example project which demonstrates the
use of the FFI. This complements [the FFI
tutorial](https://jax.readthedocs.io/en/latest/ffi.html ) by putting all
of the code in one place, as well as demonstrating how FFI extensions
can be packaged. Alongside the example project, I have also added a new
GitHub Actions workflow to test the example as part of CI.
For now, the tests only run on CPU, but once we have GPU runners for
GitHub Actions (soon!), I plan on migrating the custom call examples
from `docs/gpu_ops` and `docs/cuda_custom_call` into this test case.
Similarly, I wanted to start small and this example project only
includes exactly the same functions as the tutorial for now, but I think
this could be a good place to showcase more advanced examples (including
custom calls with state).
2024-09-24 17:23:13 -04:00
Jake VanderPlas
45af8742ca
trigger array API tests for all PRs.
...
We should have done this when we deprecated jax.experimental.array_api
2024-09-23 18:18:15 -07:00
dependabot[bot]
c4c30e1cfd
Bump actions/upload-artifact from 4.3.6 to 4.4.0
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.3.6 to 4.4.0.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](834a144ee9...50769540e7
2024-09-20 14:53:30 +00: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
Michael Hudgins
7f3a90c63b
Change references in setup.py and utilities to reference the JAX repo move to the JAX-ML org
...
PiperOrigin-RevId: 676838502
2024-09-20 07:32:15 -07:00
jax authors
e27f1e9b3a
Change Python version 3.13.0rc2 to 3.13.0-rc.2.
...
The value is taken from [the versions manifest](https://raw.githubusercontent.com/actions/python-versions/main/versions-manifest.json ).
PiperOrigin-RevId: 676012255
2024-09-18 09:04:32 -07:00
jax authors
8bcdb12852
Add CI jobs for python 3.13.0rc2.
...
PiperOrigin-RevId: 675758096
2024-09-17 16:51:35 -07:00
Jake VanderPlas
cd08337717
CI: set concurrency for workflows
2024-09-16 14:55:44 -07:00
Peter Hawkins
1594d2f30f
Prepare for v0.4.32 release.
2024-09-11 14:29:55 -04:00
Jake VanderPlas
f86cc91cce
array API: use latest array-api-tests commit hash
2024-09-09 08:58:31 -07:00
dependabot[bot]
802abfef92
Bump actions/upload-artifact from 4.3.5 to 4.3.6
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.3.5 to 4.3.6.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](89ef406dd8...834a144ee9
2024-08-12 17:54:09 +00:00
Jake VanderPlas
d999208863
[array API] update test suite to most recent commit
2024-08-08 12:33:30 -07:00
dependabot[bot]
6d7cf3fcfd
Bump actions/upload-artifact from 4.3.3 to 4.3.5
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.3.3 to 4.3.5.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](65462800fd...89ef406dd8
2024-08-05 17:09:11 +00:00
Jake VanderPlas
521c94c6c6
Tighten the public API for jax.lib.xla_client & xla_extension
2024-08-03 05:26:22 -07: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
jax authors
83f0c979fa
Merge pull request #22456 from google:dependabot/github_actions/actions/setup-python-5.1.1
...
PiperOrigin-RevId: 654787031
2024-07-22 09:36:33 -07:00
Sergei Lebedev
7a62b8dd18
Re-enabled PallasCallPrintTest on Cloud TPUs
...
PiperOrigin-RevId: 652823653
2024-07-16 06:55:20 -07:00
dependabot[bot]
98f978f32b
Bump actions/setup-python from 5.1.0 to 5.1.1
...
Bumps [actions/setup-python](https://github.com/actions/setup-python ) from 5.1.0 to 5.1.1.
- [Release notes](https://github.com/actions/setup-python/releases )
- [Commits](82c7e631bb...39cd14951b
2024-07-15 18:01:34 +00:00
Peter Hawkins
3019467992
Update references to jaxlib_nightly_releases (which is a legacy index and no longer updated) to jax_nightly_releases.
...
Fixes CI jobs using old jaxlib nightly wheels.
2024-07-15 12:42:50 -04:00
dependabot[bot]
8bc4643627
Bump actions/download-artifact from 4.1.7 to 4.1.8
...
Bumps [actions/download-artifact](https://github.com/actions/download-artifact ) from 4.1.7 to 4.1.8.
- [Release notes](https://github.com/actions/download-artifact/releases )
- [Commits](65a9edc588...fa0a91b85d
2024-07-08 17:27:27 +00:00
Jake VanderPlas
d03572eac5
CI: add jax2tf_test action
2024-06-27 10:23:58 -07:00
Peter Hawkins
945fde41e4
Update minimum Python version to 3.10.
2024-06-26 13:47:14 -04:00
Peter Hawkins
f5f9da34e6
Update Github actions builds to use Python 3.10 instead of Python 3.9.
...
Change in preparation for dropping Python 3.9 support.
PiperOrigin-RevId: 646997432
2024-06-26 10:33:11 -07:00
Dan Foreman-Mackey
5b5214b507
Update Windows Actions workflow name
...
When skipped, the build matrix for the Windows workflow on GitHub
Actions doesn't get populated, so the run name ends up as
`${{ matrix.os }} CI build`. It doesn't seem necessary to include the
matrix value in the name since there is only one build anyways. This PR
replaces the name with a constant.
2024-06-18 12:54:03 -04:00
Peter Hawkins
b0b02682d5
Add --allow-downgrade to Windows wheel builds.
...
We want a specific LLVM version, downgrades are fine.
Also add --no-progress for a more compact log.
2024-06-18 08:15:26 -04:00
Peter Hawkins
160e09e235
Use NumPy 2.0.0 and SciPy 1.13.1 in builds.
...
Don't override the XLA repository in the nightly Windows CI builds,
which should be building JAX as it exists in the source repository.
2024-06-17 19:35:08 +00:00
