rocm_jax/docs/requirements.txt

absl-py
ipython>=8.8.0  # 8.7.0 has ipython3 lexer error
pydata-sphinx-theme==0.14.4  # v0.15 breaks sidebar toggling
sphinx>=7.3.2,<8.0  # 7.3.0 breaks sphinx-book-theme; 8.0 breaks myst-nb 1.1
sphinx-book-theme==1.1.1  # v1.1.2 requires pydata-sphinx-theme v0.15
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
sphinx-design
sphinxext-rediraffe
myst-nb>=1.0.0

# Packages used for CI tests.
flatbuffers
pytest
pytest-xdist

# Packages used for notebook execution
matplotlib
scikit-learn
pooch
numpy
rich[jupyter]
cmake
.[ci]  # Install jax from the current directory; jaxlib from pypi.
CI: add absl-py to docs/requirements.txt 2022-11-04 11:28:15 -07:00			`absl-py`
DOC: change requirements pinnings to prevent timeout 2023-01-09 08:32:03 -08:00			`ipython>=8.8.0 # 8.7.0 has ipython3 lexer error`
DOC: update main landing page style Co-authored-by: Jake VanderPlas <jakevdp@google.com> 2024-11-15 13:44:31 -08:00			`pydata-sphinx-theme==0.14.4 # v0.15 breaks sidebar toggling`
CI: pin sphinx to avoid build errors on 8.0 2024-08-06 09:16:41 -07:00			`sphinx>=7.3.2,<8.0 # 7.3.0 breaks sphinx-book-theme; 8.0 breaks myst-nb 1.1`
DOC: update main landing page style Co-authored-by: Jake VanderPlas <jakevdp@google.com> 2024-11-15 13:44:31 -08:00			`sphinx-book-theme==1.1.1 # v1.1.2 requires pydata-sphinx-theme v0.15`
Add copy button to code snippets in documentation 2022-02-22 13:49:48 +04:00			`sphinx-copybutton>=0.5.0`
CI: upgrade doc requirements 2022-07-11 14:08:53 -07:00			`sphinx-remove-toctrees`
Update doc landing page Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> Co-authored-by: Jake VanderPlas <jakevdp@google.com> 2023-01-13 09:59:40 -08:00			`sphinx-design`
DOC: replace old tutorials with new content 2024-04-18 13:11:25 -07:00			`sphinxext-rediraffe`
DOC: update package requirements 2023-11-20 11:22:29 -08:00			`myst-nb>=1.0.0`
Update docs requirements. (#3969) 2020-08-07 11:47:05 -07:00
DOC: update doc build requirements 2021-02-23 10:31:26 -08:00			`# Packages used for CI tests.`
[export] Add support for serialization and deserialization of Exported At the moment we can export a JAX function into an Exported and we can invoke an Exported from another JAX function, but there is no way to serialize an Exported to be able to use it in another process. Exported now has all the features we had in mind, so it is a reasonable time to add a serialization method. The intention is for the serialization to have backwards compatibility guarantees, meaning that we can deserialize an Exported that has been serialized with an older version of JAX. This PR does not add explicit support for versioning, nor backwards compatibility tests. Those will follow. Here we add serialization and deserialization to bytearray, using the flatbuffers package. We use flatbuffers because it is simple, it has backwards and forwards compatibility guarantees, it is a lightweight dependency that does not require additional build steps, and it is fast (deserialization simply indexes into the bytearray rather than creating a Python structure). In the process of implementing this we have done some small cleanup of the Exported structure: * renamed serialization_version to mlir_module_serialization_version * renamed disabled_checks to disabled_safety_checks This code is tested by changing export_test.py to interpose a serialization followed by a deserialization every time we export.export. There is a known bug with the serialization of effects, so I disabled one of the export tests. Will fix in a subsequent PR. PiperOrigin-RevId: 590078785 2023-12-11 23:22:16 -08:00			`flatbuffers`
DOC: update doc build requirements 2021-02-23 10:31:26 -08:00			`pytest`
			`pytest-xdist`
Update docs requirements. (#3969) 2020-08-07 11:47:05 -07:00
DOC: update doc build requirements 2021-02-23 10:31:26 -08:00			`# Packages used for notebook execution`
Fixed the readthedocs documentation build Had to extend the docs/requirements.txt file to install matplotlb (needed by the Gotchas notebook) and ".", needed by everything. This results in a reduction of the sphinx warnings from 3300 to 1200! 2019-10-09 13:44:13 +02:00			`matplotlib`
Use scikit-learn rather than sklearn in docs/requirements.txt 2021-06-21 12:03:44 +02:00			`scikit-learn`
fix the doc error: module 'scipy.misc' has no attribute 'face' 2025-01-05 23:02:17 -08:00			`pooch`
Update minimum jaxlib version to 0.1.69. 2021-07-15 16:39:18 -04:00			`numpy`
Upgrade JAX Gradient Checkpointing doc 2023-12-21 23:13:29 +00:00			`rich[jupyter]`
Add ffi_call tutorial Building on #21925, this tutorial demonstrates the use of the FFI using `ffi_call` with a simple example. I don't think this should cover all of the most advanced use cases, but it should be sufficient for the most common examples. I think it would be useful to eventually replace the existing CUDA tutorial, but I'm not sure that it'll get there in the first draft. As an added benefit, this also runs a simple test (akin to `docs/cuda_custom_call`) which actually executes using a tool chain that open source users would use in practice. 2024-06-25 14:08:49 -04:00			`cmake`
Release jax and jaxlib 0.3.0 as per the new release process. PiperOrigin-RevId: 427809845 2022-02-10 11:58:30 -08:00			`.[ci] # Install jax from the current directory; jaxlib from pypi.`