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: link directly to installation on the main page

Browse Source
This commit is contained in:
Jake VanderPlas 2024-11-13 12:44:05 -08:00
parent 558ebb9fb1
commit 72a4692b94
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/_static/style.css vendored
View File

@ -8,15 +8,15 @@
background-color: #fff;
}
.getting-started {
.installation {
background-color: rgba(78, 150, 253, var(--block-bg-opacity));
}
.user-guides {
.getting-started {
background-color: rgba(0, 169, 154, var(--block-bg-opacity));
}
.developer-docs {
.user-guides {
background-color: rgba(171, 0, 182, var(--block-bg-opacity));
}

12
docs/index.rst
View File

@ -32,6 +32,12 @@ designed for high-performance numerical computing and large-scale machine learni
.. grid:: 3
.. grid-item-card:: :material-regular:`laptop_chromebook;2em` Installation
:columns: 12 6 6 4
:link: installation
:link-type: ref
:class-card: installation
.. grid-item-card:: :material-regular:`rocket_launch;2em` Getting started
:columns: 12 6 6 4
:link: beginner-guide
@ -44,12 +50,6 @@ designed for high-performance numerical computing and large-scale machine learni
:link-type: ref
:class-card: user-guides
.. grid-item-card:: :material-regular:`laptop_chromebook;2em` Developer notes
:columns: 12 6 6 4
:link: contributor-guide
:link-type: ref
:class-card: developer-docs
If you're looking to train neural networks, use Flax_ and start with its tutorials.
For an end-to-end transformer library built on JAX, see MaxText_.