mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 20:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
3,487 Commits 508 Branches 263 Tags
Commit Graph

125 Commits

Author SHA1 Message Date
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
Matthew Johnson
5edda9e66c
remove very old "what we're working on" 2019-11-19 14:20:49 -08:00
James Bradbury
5fa68774ea
Bump jaxlib version in README 2019-11-16 17:51:16 -08:00
Peter Hawkins
3d4bb775a5
Update jaxlib version in README.md. (#1669) 2019-11-11 20:11:28 -05:00
George Necula
8880e262b0
Use redthedocs links for Colabs (#1572) 
Steer the documentation readers to readthedocs.
Also, minor fixes to the wording of How_jax_primitives_work, suggested by Dougal
 2019-10-29 08:53:35 +01:00
Skye Wanderman-Milne
d822d6bc45 Update README to point to jaxlib-0.1.31 2019-10-21 15:45:02 -07:00
Peter Hawkins
ae26cfb0e8 Remove stale "running the tests" link. 2019-10-18 15:36:53 -04:00
George Necula
eae59d0b2c
Moved all notebooks to docs/notebooks. (#1493) 
* Moved all notebooks to docs/notebooks.

Now all notebooks are in the same place, thus all are subject
to auto-doc generation at readthedocs.io and to automated testing
with travis.

Some notebooks are too slow, exclude them at docs/conf.py:exclude_patterns.

Cleanup a bit the section headings in notebooks so that they show
up well in readtehdocs.io.

* Increase the cell timeout for executing notebooks
* Exclude also the neural network notebook from auto-generation (timing out)
* Disable the score_matching notebook from auto-doc (travis does not have sklearn)
 2019-10-17 08:58:25 +02:00
Peter Hawkins
9ff9e9e34a
Describe minimum OS version requirements in README.md. (#1508) 2019-10-15 10:25:55 -04:00
Peter Hawkins
d161e1badb
Add text to documentation and to jaxlib version check suggesting a pip update. (#1506) 
* Add instructions to update `pip` to README.md.

* Add text that suggests upgrading pip if we see jaxlib 0.1.23.

* Remove extra text from README.md.
 2019-10-15 08:52:21 -04:00
joaogui1
d6bc62bd6b Fixes Readme links to notebooks 2019-10-13 20:30:32 -03:00
George Necula
a9d9504348 Fixes to the documentation 
* Included "Building from source" in README.md
* Added references from docs/README.md to docs/developer.rst
 2019-10-09 17:45:09 +02:00
George Necula
e42c010605 Create developer documentation. 
* Moved out of README.md some developer-only stuff to docs/developer.rst.
    * Added documentation about building the documentation
 2019-10-09 17:24:01 +02:00
George Necula
6c6e90d019
Merge pull request #1455 from gnecula/colabs 
Increase readthedocs/nbsphinx timeout
 2019-10-08 17:41:36 +02:00
Roy Frostig
cd29780fe6 minor README formatting and grammar adjustment 2019-10-07 16:55:17 -07:00
George Necula
41457633cc Increase readthedocs/nbsphinx timeout 
It seems tht RTD is timing out when compiling the How_JAX_primitives_work noteboook.
 2019-10-07 18:07:55 +02:00
Fabian Pedregosa
5201c36348 Typo: Gradeints -> Gradients 2019-10-03 17:48:33 -07:00
Skye Wanderman-Milne
7f98682f24 Update README to point to jaxlib-0.1.30 2019-10-03 14:21:44 -07:00
Stephan Hoyer
d13599e89c
Merge pull request #1425 from gnecula/colab_testing 
Added support for testing Colab notebooks.
 2019-10-03 12:23:12 -07:00
George Necula
0fdf377343 Switched testing to using sphinx (part of the documentation generation) 2019-10-03 20:54:28 +02:00
George Necula
57e39ce938 Added support for testing Colab notebooks. 
Testing is done by running "jupyter nbconvert --to notebook" and
then parsing the resulting notebook to look for errors.

One can declare expected errors, and the test will fail if those
are missing.

In the process of doig this, found and fixed a bug in the autodiff_cookbook
notebook.
 2019-10-03 20:07:49 +02:00
George Necula
6744389d90 Fixed links to Colabs in docs/notebooks 
This is related to recent change https://github.com/google/jax/pull/1121.
 2019-10-02 15:48:33 +02:00
Peter Hawkins
7a96e50277 Update README to point to jaxlib 0.1.29. 2019-09-28 18:42:52 -04:00
Roy Frostig
16484ccbed add a "Citing JAX" section to the README. fixes #1359 2019-09-19 15:02:03 -07:00
Skye Wanderman-Milne
2b075f9371 Update README to use jaxlib-0.1.28 2019-09-05 14:48:02 -07:00
Skye Wanderman-Milne
2c6f74dfbc Add CUDA 10.1 wheels. 2019-09-05 14:45:32 -07:00
Matthew Johnson
c760b05f9b update jaxlib version in readme 
fixes #1297
will update notebooks in #1260
 2019-09-02 07:25:06 -07:00
Matthew Johnson
94b1dda42f fix notebook link (fixes #1252) 2019-08-26 12:35:39 -07:00
Matthew Johnson
524d4254f8 update xla link, fixes #1248 2019-08-25 20:12:32 -07:00
Huang, Guangtai
09a2d06839
fix typo 2019-08-22 13:13:52 +08:00
Peter Hawkins
39b01564d4
Merge pull request #1167 from hawkinsp/master 
Update installation instructions.
 2019-08-12 14:56:54 -04:00
Roy Frostig
a657922309
Merge pull request #1132 from necula01/update_doc 
Update documentation for jax.vmap.
 2019-08-12 07:53:33 -07:00
Peter Hawkins
b6f0ec7a22 Update installation instructions. 
Prefer installing via a wheel, because it's simpler to explain.
Describe exactly which packages are needed to build on Ubuntu 18.04.
 2019-08-12 10:40:28 -04:00
Peter Hawkins
1b672dc28d Update jaxlib release to 0.1.23. 2019-08-08 22:09:30 -04:00
Denis Mazur
52e7c23e44 add Generative Modelling by Estimating Gradients of Data Distribution notebook tutorial 2019-08-07 22:02:02 +03:00
George Necula
9007d67109 Update documentation for jax.vmap. 2019-08-07 09:26:46 +02:00
Aditya Vaidya
35a430e1c8 Fix link in README 2019-07-17 03:18:28 -05:00
Peter Hawkins
6c3cf13f65 Remove outdated reference to incorrect NaN semantics. 2019-06-24 21:17:32 -04:00