mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 08:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
7,282 Commits 508 Branches 263 Tags
Commit Graph

190 Commits

Author SHA1 Message Date
Tom Hennigan
4f74b3391c
Update README.md 
Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
 2021-03-12 16:38:29 +00:00
Tom Hennigan
0fd83a30be Link to DeepMind JAX blog post. 2021-03-12 12:01:20 +00:00
Skye Wanderman-Milne
f06bb9a7f4 Update jaxlib version etc. 2021-03-09 17:55:40 -08:00
Jake VanderPlas
749ad95514 DOC: add transformations doc to HTML & reorganize contents 2021-03-08 16:25:04 -08:00
Jake VanderPlas
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -08:00
Skye Wanderman-Milne
7a67b974ac jaxlib version bump etc. 2021-02-12 09:42:04 -08:00
Peter Hawkins
13f3819054 Update README.md for jaxlib 0.1.60. 
Bump jaxlib version to 0.1.61 and update changelog.

Change jaxlib numpy version limit to >=1.16 for next release. Releases older than 1.16 are deprecated per NEP 00029. Reenable NumPy 1.20.

Bump minimum jaxlib version to 0.1.60.
 2021-02-03 20:44:01 -05:00
Peter Hawkins
4fdd029b3e Clarify the CuDNN versions expected by jaxlib wheels. 2021-01-20 10:00:19 -05:00
Skye Wanderman-Milne
7c2454e969 Update jaxlib version, minimum jaxlib version, readme, and changelog. 
Bumping the min jaxlib version to support https://github.com/google/jax/pull/5213.
 2021-01-15 12:56:08 -08:00
Peter Hawkins
5747c91564 Update CUDA installation instructions. 2021-01-13 10:48:43 -05:00
jax authors
783e1b7c89 Merge pull request #5236 from bileschi:patch-1 
PiperOrigin-RevId: 350975202
 2021-01-09 20:43:00 -08:00
Stanley Bileschi
c1904ea7a7 fixing. Renaming per reviewer feedback. 2021-01-06 11:26:46 -05:00
John Aslanides
da4e27315d Add PyPI version badge to README. 2020-12-30 22:48:24 +00:00
Stanley Bileschi
80173aad62
(docfix) Typo in README.md 
`predict` function should apply nonlinearity to output
 2020-12-21 00:35:22 -05:00
Peter Hawkins
c83343d80b Add a few notes about Windows builds and a link to the build instructions. 2020-11-30 20:13:01 -05:00
jax authors
6cdd956a74 Merge pull request #5042 from redwrasse:redwrasse-patch-1 
PiperOrigin-RevId: 344799393
 2020-11-30 06:17:58 -08:00
redwrasse
06957e8caa
fix readme 
fix readme
 2020-11-29 18:51:24 -07:00
Matthew Johnson
f180ea86d4
fix capitalization 2020-11-16 19:03:15 -08:00
Matthew Johnson
c2a98c1820
update jax repository citation 
Add all full-time JAX team members.
 2020-11-16 19:01:00 -08:00
Skye Wanderman-Milne
36799f5007 README etc. updates for new jaxlib release 2020-11-12 13:48:44 -08:00
mariosasko
68ef05be46 Fix broken link in readme 2020-11-09 00:33:04 +01:00
Matthew Johnson
a7d1963bc8 update neural network libraries text 
Co-authored-by: George Necula <necula@google.com>
 2020-10-30 08:42:04 -07:00
Matthew Johnson
a4ce2813c8 link to nn libraries in readme 
Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>
Co-authored-by: James Bradbury <jekbradbury@google.com>
 2020-10-30 08:39:59 -07:00
Skye Wanderman-Milne
a082169642 Update README and jaxlib.__version__ for new jaxlib release 2020-10-20 11:46:48 -07:00
Alexey Radul
537427aae2 Copybara import of the project: 
--
138105a9ea44e7a8c3ce575a4e51b7ed51518d41 by Skye Wanderman-Milne <skyewm@google.com>:

Update README, CHANGELOG, and jaxlib.__version__ for new jaxlib release

PiperOrigin-RevId: 338063494
 2020-10-20 08:25:23 -07:00
Skye Wanderman-Milne
138105a9ea Update README, CHANGELOG, and jaxlib.__version__ for new jaxlib release 2020-10-14 13:32:56 -07:00
Skye Wanderman-Milne
cacb01753a Use local version identifiers to distribute cuda jaxlib wheels. 
This change:

* Updates our jaxlib build scripts to add `+cudaXXX` to the wheel
  version, where XXX is the CUDA version number (e.g. `110`). nocuda
  builds remain unchanged and do not have this extra identifier.

* Adds `generate_release_index.py`, which writes an html page that pip
  can use to find the cuda wheels. (I based this format off of
  wheel PyTorch's index).

* Updates the README to use the new local version identifier + wheel
  index.

The end result is that the command to install cuda wheels is now much
simpler.

I manually made copies of the latest jaxlib 0.1.55 wheels that have
the local version identifiers, so the new installation commands
already work (as well as the old ones, until the next jaxlib release
using the new tooling).

Fow now, I put the html index to the GCP bucket with the wheels. We
can move it to a prettier URL if/when we have one.
 2020-10-09 13:47:54 -07:00
Qiao Zhang
7e694bd3b6
Update README to point to jaxlib-0.1.55. (#4256) 2020-09-10 18:28:44 -07:00
Skye Wanderman-Milne
428217b15c
Update auto jaxlib install to use nvcc CUDA version instead of nvidia-smi. (#4050) 
The CUDA version reported by `nvidia-smi` is based on the driver
version, not the runtime/toolkit version, and may be higher than the
toolkit version since the driver supports older runtimes. `nvcc`,
which is also installed as part of the CUDA toolkit, will correctly
report the right CUDA version to target.

Fixes #3984.
 2020-08-13 12:10:47 -07:00
Peter Hawkins
659dd39d74
Add MLPerf results link. (#3896) 2020-07-29 16:22:12 -04:00
Jake Vanderplas
1bc5896ee4
update README for new jaxlib version (#3825) 2020-07-22 19:39:51 -07:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks (#3760) 2020-07-15 13:17:38 -07:00
George Necula
166e795d63
Updated minimum jaxlib to 0.1.51 (#3642) 2020-07-02 18:53:58 +03:00
Peter Hawkins
b576417e5b
Update README for jaxlib 0.1.50 release. (#3574) 2020-06-26 13:20:59 -04:00
Roy Frostig
df3b507c43 update jaxlib version in readme 2020-06-16 14:19:03 -07:00
Peter Hawkins
ae60654dd4
Change build badge from Travis to Github actions. (#3414) 2020-06-11 19:12:29 -04:00
Peter Hawkins
701bc76166
Update README to recommend Windows Subsystem for Linux (WSL) to Windows users. (#3356) 2020-06-08 09:03:02 -04:00
Skye Wanderman-Milne
6e50aa9b80
Update installation directions in README to mention expected CUDA location. (#3190) 
See https://github.com/google/jax/issues/989
 2020-05-27 17:05:36 -07:00
Skye Wanderman-Milne
16cf845148 Update jaxlib version to 0.1.47 in README 2020-05-13 17:47:37 -07:00
Peter Hawkins
61a34f5903
Update README for jaxlib 0.1.46 release. (#2968) 2020-05-05 12:18:55 -04:00
Yufeng
aee18c8860
small typo fix (#2799) 2020-04-22 14:27:33 -07:00
Skye Wanderman-Milne
6db1f0c0d0 Bump jaxlib version in README to 0.1.45 2020-04-21 15:10:43 -07:00
Skye Wanderman-Milne
a1b4fe47bd
Update automatic jaxlib install command to use nvidia-smi instead of nvcc. (#2758) 
This is just to get the CUDA version number, and nvidia-smi is more
commonly available.
 2020-04-17 16:17:56 -07:00
Peter Hawkins
d356c41f77
Release jaxlib 0.1.44. (#2740) 2020-04-16 11:31:05 -04:00
Peter Hawkins
fcefc7e8f1
Add a link to Google's codesearch for JAX. (#2646) 2020-04-08 13:02:58 -04:00
Peter Hawkins
bd1708c707
Update changelog and README for jaxlib 0.1.43. (#2556) 2020-03-31 10:02:38 -04:00
Matthew Johnson
7e480fa923 add custom_jvp / vjp, delete custom_transforms 2020-03-21 22:08:03 -07:00
Peter Hawkins
c3f8909042
Update jaxlib version in README.md. (#2461) 2020-03-19 12:41:19 -04:00
Peter Hawkins
cbdf9a5a43
Drop support for Python 3.5. (#2445) 2020-03-18 10:54:28 -04:00
Peter Hawkins
ec9513fa29
Advertise jaxlib 0.1.41. (#2432) 
Bump minimum jaxlib version to 0.1.41.
 2020-03-16 16:10:26 -04:00