GV

GeezerVison

The World Wide Web, restored to 800 pixels of glory.

Home|About|Gaming cabinet|Archives|How it works|Starter sites

On this site

  • Sign in
  • ActionsCI/CD and automation
  • CodespacesCloud development environments
  • All Features
  • Documentation
  • Training
  • Certifications
  • Build with Copilot
  • Customize Copilot
  • Integrations
  • Trending repos
  • GitHub Sponsors
  • Developer articles
  • Enterprises

This view

  • Original page
  • New address

GeezerVison is showing github.com — jax/docs at amd-main · ROCm/jax

Expand file tree

/

docs

/
Copy path

Directory actions

More options

More options

Directory actions

More options

More options

Latest commit

 

History

History
History
 
 
/

docs

/
Copy path

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
101
101
 
 
201
201
 
 
301
301
 
 
401
401
 
 
501
501
 
 
601
601
 
 
_static
_static
 
 
_templates
_templates
 
 
_tutorials
_tutorials
 
 
debugging
debugging
 
 
export
export
 
 
ffi
ffi
 
 
internals
internals
 
 
jep
jep
 
 
notebooks
notebooks
 
 
pallas
pallas
 
 
sphinxext
sphinxext
 
 
README.md
README.md
 
 
__init__.py
__init__.py
 
 
about.md
about.md
 
 
accuracy.md
accuracy.md
 
 
advanced_autodiff.md
advanced_autodiff.md
 
 
advanced_guides.rst
advanced_guides.rst
 
 
aot.md
aot.md
 
 
api_compatibility.md
api_compatibility.md
 
 
array_refs.ipynb
array_refs.ipynb
 
 
array_refs.md
array_refs.md
 
 
array_refs.py
array_refs.py
 
 
async_dispatch.rst
async_dispatch.rst
 
 
autodidax.ipynb
autodidax.ipynb
 
 
autodidax.md
autodidax.md
 
 
autodidax.py
autodidax.py
 
 
autodidax2_part1.ipynb
autodidax2_part1.ipynb
 
 
autodidax2_part1.md
autodidax2_part1.md
 
 
autodidax2_part1.py
autodidax2_part1.py
 
 
automatic-differentiation.md
automatic-differentiation.md
 
 
automatic-vectorization.md
automatic-vectorization.md
 
 
beginner_guide.rst
beginner_guide.rst
 
 
benchmarking.md
benchmarking.md
 
 
buffer_donation.md
buffer_donation.md
 
 
build_custom_gpu.sh
build_custom_gpu.sh
 
 
building_on_jax.md
building_on_jax.md
 
 
changelog.md
changelog.md
 
 
complex-differentiation.md
complex-differentiation.md
 
 
concurrency.rst
concurrency.rst
 
 
conf.py
conf.py
 
 
config_options.rst
config_options.rst
 
 
contributing.md
contributing.md
 
 
contributor_guide.rst
contributor_guide.rst
 
 
control-flow.md
control-flow.md
 
 
debugging.md
debugging.md
 
 
deprecation.md
deprecation.md
 
 
developer.md
developer.md
 
 
device_memory_profiling.md
device_memory_profiling.md
 
 
direct_linearize_migration.md
direct_linearize_migration.md
 
 
distributed_data_loading.md
distributed_data_loading.md
 
 
errors.rst
errors.rst
 
 
extensions.rst
extensions.rst
 
 
external-callbacks.md
external-callbacks.md
 
 
faq.rst
faq.rst
 
 
fault_tolerance.rst
fault_tolerance.rst
 
 
ffi.ipynb
ffi.ipynb
 
 
ffi.md
ffi.md
 
 
ffi.py
ffi.py
 
 
glossary.rst
glossary.rst
 
 
gpu_memory_allocation.rst
gpu_memory_allocation.rst
 
 
gpu_performance_tips.md
gpu_performance_tips.md
 
 
gradient-checkpointing.md
gradient-checkpointing.md
 
 
hero.html
hero.html
 
 
higher-order.md
higher-order.md
 
 
hijax_custom_derivatives.ipynb
hijax_custom_derivatives.ipynb
 
 
hijax_custom_derivatives.md
hijax_custom_derivatives.md
 
 
hijax_custom_derivatives.py
hijax_custom_derivatives.py
 
 
hijax_types.ipynb
hijax_types.ipynb
 
 
hijax_types.md
hijax_types.md
 
 
hijax_types.py
hijax_types.py
 
 
index.rst
index.rst
 
 
installation.md
installation.md
 
 
investigating_a_regression.md
investigating_a_regression.md
 
 
jacobian-vector-products.md
jacobian-vector-products.md
 
 
jax-101.rst
jax-101.rst
 
 
jax.ad_checkpoint.rst
jax.ad_checkpoint.rst
 
 
jax.debug.rst
jax.debug.rst
 
 
jax.distributed.rst
jax.distributed.rst
 
 
jax.dlpack.rst
jax.dlpack.rst
 
 
jax.dtypes.rst
jax.dtypes.rst
 
 
jax.example_libraries.optimizers.rst
jax.example_libraries.optimizers.rst
 
 
jax.example_libraries.rst
jax.example_libraries.rst
 
 
jax.example_libraries.stax.rst
jax.example_libraries.stax.rst
 
 
jax.experimental.checkify.rst
jax.experimental.checkify.rst
 
 
jax.experimental.compilation_cache.rst
jax.experimental.compilation_cache.rst
 
 
jax.experimental.custom_partitioning.rst
jax.experimental.custom_partitioning.rst
 
 
jax.experimental.jet.rst
jax.experimental.jet.rst
 
 
jax.experimental.key_reuse.rst
jax.experimental.key_reuse.rst
 
 
jax.experimental.mesh_utils.rst
jax.experimental.mesh_utils.rst
 
 
jax.experimental.multihost_utils.rst
jax.experimental.multihost_utils.rst
 
 
jax.experimental.pallas.fuser.rst
jax.experimental.pallas.fuser.rst
 
 
jax.experimental.pallas.mosaic_gpu.rst
jax.experimental.pallas.mosaic_gpu.rst
 
 
jax.experimental.pallas.rst
jax.experimental.pallas.rst
 
 
jax.experimental.pallas.tpu.rst
jax.experimental.pallas.tpu.rst
 
 

README.md

To rebuild the documentation, see Update Documentation.

You are visitor 01402520 — GeezerVison does not store the pages you look up.
Best viewed with Netscape Navigator 4.7 or Internet Explorer 6 at 800×600. No cookies required.