mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

DOC: update sphinx & sphinx-autodoc-typehints

Browse Source
This commit is contained in:
Jake VanderPlas 2023-03-01 11:02:56 -08:00
parent b348fce4dd
commit 55d9c06267
3 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docs/_static/style.css vendored
View File

@ -19,8 +19,3 @@
.developer-docs {
background-color: rgba(171, 0, 182, var(--block-bg-opacity));
}
.key-ideas
{
border: 0px
}

6
docs/index.rst
View File

@ -10,21 +10,21 @@ JAX is Autograd_ and XLA_, brought together for high-performance numerical compu
.. grid-item-card:: Familiar API
:columns: 12 6 6 4
:class-card: key-ideas
:class-card: sd-border-0
:shadow: None
JAX provides a familiar NumPy-style API for ease of adoption by researchers and engineers.
.. grid-item-card:: Transformations
:columns: 12 6 6 4
:class-card: key-ideas
:class-card: sd-border-0
:shadow: None
JAX includes composable function transformations for compilation, batching, automatic differentiation, and parallelization.
.. grid-item-card:: Run Anywhere
:columns: 12 6 6 4
:class-card: key-ideas
:class-card: sd-border-0
:shadow: None
The same code executes on multiple backends, including CPU, GPU, & TPU

9
docs/requirements.txt
View File

@ -1,10 +1,9 @@
absl-py
ipython>=8.8.0 # 8.7.0 has ipython3 lexer error
sphinx>=4
# TODO(jakevdp) unpin sphinx-autodoc-typehints when sphinx-book-theme
# 0.4.0 is released and no longer requires sphinx<5
sphinx-autodoc-typehints~=1.18.0
sphinx-book-theme>=0.3.3
# TODO(jakevdp) Update to sphinx>=6 when myst-nb supports it.
sphinx>=5
sphinx-autodoc-typehints
sphinx-book-theme>=1.0.0
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
jupyter-sphinx>=0.3.2