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
JL2718
c53ae2c477
automatic detection of wheel version ( #2373 )
2020-03-09 11:43:45 -07:00
Peter Hawkins
73443d28d8
Update release notes for jaxlib 0.1.40. ( #2359 )
2020-03-05 08:41:07 -05:00
George Necula
89514f9278
Moved CHANGELOG to docs ( #2252 )
...
* Moved CHANGELOG to docs
This puts the documentation also on RTD, with TOC.
Also changed its format to .rst, for consistency.
Added GitHub links to the change log.
* Actually add the CHANGELOG.rst
* Added reminder comments to the CHANGELOG.rst
2020-02-23 19:18:06 +01:00
brett koonce
8372a70079
tweak readme pmap imports ( #2276 )
2020-02-20 16:44:21 -08:00
Skye Wanderman-Milne
2a6e60fd50
Bump jaxlib version in README to 0.1.39
2020-02-14 10:06:27 -08:00
lanctot
051d7b8956
Fix broken link in README ( #2196 )
2020-02-08 13:11:42 -05:00
Skye Wanderman-Milne
633aa00392
Update README to drop CUDA 9.0 and add CUDA 10.2
2020-01-29 11:23:27 -08:00
Skye Wanderman-Milne
09d2421f28
Update jaxlib version in README to 0.1.38
2020-01-29 10:48:20 -08:00
Matthew Johnson
d46e82d0ab
tweak readme announcement text again
2020-01-28 18:16:04 -08:00
Matthew Johnson
71811be3b9
tweak top-line announcement text in readme
2020-01-28 18:15:16 -08:00
Matthew Johnson
1afcac70df
tweak readme not to have bad line wrap
2020-01-28 16:41:21 -08:00
Peter Hawkins
0fe601227f
Update README.md and CHANGELOG.md.
2020-01-28 10:00:32 -05:00
Matthew Johnson
a61bcff54d
update readme gotchas about pure functions
2020-01-23 10:25:49 -08:00
Mu Li
2c80cd3d88
Fix Sysml paper link in README ( #2036 )
...
The original ULR was broken as sysml updated their links.
2020-01-21 13:26:36 -08:00
Jamie Townsend
371001aad1
Fix README typo ( #2020 )
2020-01-17 09:48:56 -08:00
Jamie Townsend
3974df0aee
[docs] Pmap compiles functions with XLA ( #2021 )
2020-01-17 09:48:27 -08:00
Matthew Johnson
a7eb5897d3
add mini-libraries readme
2020-01-11 16:31:59 -08:00
Peter Hawkins
dcc882cf6b
Drop Python 2 support from JAX. ( #1962 )
...
Remove six dependency.
2020-01-08 13:17:55 -05:00
Matthew Johnson
3bf12f6ecd
update readme jaxlib version
2019-12-24 11:16:42 -08:00
Matthew Johnson
5c800367d1
mention cloud tpus in readme
2019-12-14 10:32:04 -08:00
Matthew Johnson
764f007f9a
point pmap links to cloud tpu colabs
2019-12-14 10:32:04 -08:00
Matthew Johnson
0ad8837f2f
tweak wording around installing jaxlib
2019-12-14 10:32:04 -08:00
Matthew Johnson
8dad859e04
streamline readme, add pmap
2019-12-14 10:32:04 -08:00
Matthew Johnson
8927200316
wip
2019-12-14 10:32:04 -08:00
Skye Wanderman-Milne
15d276ce0c
Bump README to jaxlib 0.1.36
2019-11-22 10:41:59 -08:00
Skye Wanderman-Milne
c77cdb5144
Bump README to jaxlib 0.1.35
2019-11-21 11:02:54 -08:00
Peter Hawkins
c60f3fd65d
Minor documentation fixes. ( #1734 )
2019-11-21 09:51:26 -05:00
Skye Wanderman-Milne
ad09822067
Bump README to jaxlib 0.1.34
2019-11-20 10:59:54 -08:00
