.gitignore Add .hypothesis/ directory to .gitignore Jan 17, 2024 .pre-commit-config.yaml [pre-commit] bump pre-commit version to v5.0.0 May 16, 2025 .readthedocs.yml Fix grep for label on Read the Docs. Apr 11, 2025 AUTHORS
https://gitlab.com/postmarketOS/pmaports/-/issues/1945 https://salsa.debian.org/Mobian-team/devices/kernels/sunxi64-linux/-/issues/92 This was referenced Jan 3, 2024 v2023-007 release #281 Closed draft: sunxi: pinephone: differentiate hw revisions 1.2 and 1.2b Tow-Boot/U-Boot#1 Clos...
git clone https://github.com/google/jax cd jax pip install scipy If you are building on a Mac, make sure XCode and the XCode command line tools are installed. To build XLA with CUDA support, you can run python build/build.py --enable_cuda pip install -e build # install jaxlib (incl...
To upgrade to the latest version from GitHub, just run git pull from the JAX repository root, and rebuild by running build.py if necessary. You shouldn't have to reinstall because pip install -e sets up symbolic links from site-packages into the repository. Running the tests To run all ...
@software{jax2018github, author = {James Bradbury and Roy Frostig and Peter Hawkins and Matthew James Johnson and Chris Leary and Dougal Maclaurin and Skye Wanderman-Milne}, title = {{JAX}: composable transformations of {P}ython+{N}um{P}y programs}, url = {http://github.com/google/jax...
.github/workflows CI: use internal scipy when testing internal numpy Feb 6, 2021 benchmarks Revert import path changes to examples/ and benchmarks/ Jan 21, 2021 build Update README.md for jaxlib 0.1.60. Feb 4, 2021 cloud_tpu_colabs Update Pmap_Cookbook.ipynb Feb 1, 2021 design_notes ...
yashk2810 and jax authors Make jaxpr_has_pmap work for other primitives too … 8f09606 on Jul 12 Git stats 12,222 commits FilesType Name Latest commit message Commit time .github Update ci-build.yaml 4 months ago benchmarks Avoid top-level aliases of jax.tree_util.* 4 months ago...
hawkinsp and jax authors [XLA] Fix incomplete gamma functions where x is infinity. … 4f9eb64 May 11, 2021 Git stats 7,915 commits Files Type Name Latest commit message Commit time .github CI: Install wheel pkg to improve pip-cache April 16, 2021 02:55 benchmarks [JAX] Prune ...
git clone https://github.com/google/jax cd jax To build XLA with CUDA support, you can run python build/build.py --enable_cuda pip install -e build # install jaxlib (includes XLA) pip install -e . # install jax (pure Python) See python build/build.py --help for configuration option...
git clone https://github.com/google/jax cd jax pip install scipy If you are building on a Mac, make sure XCode and the XCode command line tools are installed. To build XLA with CUDA support, you can run python build/build.py --enable_cuda pip install -e build # install jaxlib (incl...