trntensor/mkdocs.yml at main · trnsci/trntensor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
site_name: trntensor
site_description: Tensor contractions for AWS Trainium via NKI
site_url: https://trnsci.github.io/trntensor/
repo_url: https://github.com/trnsci/trntensor
repo_name: trnsci/trntensor

theme:
  name: material
  font:
    text: Atkinson Hyperlegible
    code: Atkinson Hyperlegible Mono
  palette:
    scheme: default
    primary: indigo
  features:
    - content.code.copy
    - navigation.sections

nav:
  - Home: index.md
  - Installation: installation.md
  - Quickstart: quickstart.md
  - API Reference:
      - Overview: api/index.md
      - einsum: api/einsum.md
      - Planning: api/plan.md
      - Decompositions: api/decompose.md
      - Quantum primitives: api/quantum.md
      - NKI dispatch: api/nki.md
  - Architecture: architecture.md
  - Benchmarks: benchmarks.md
  - Tutorials:
      - DF-MP2 with trntensor: tutorial_df_mp2.md
      - Contraction patterns (cuTENSOR ↔ trntensor): contraction_patterns.md
  - Deployment:
      - AWS Setup for Neuron CI: aws_setup.md

markdown_extensions:
  - pymdownx.highlight
  - pymdownx.superfences
  - admonition
  - toc:
      permalink: true