rocm_jax/docs/requirements.txt

absl-py
ipython>=8.8.0  # 8.7.0 has ipython3 lexer error
sphinx>=7.3.2  # 7.3.0 breaks sphinx-book-theme
sphinx-book-theme>=1.0.1  # Older versions fail to pin pydata-sphinx-theme
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
sphinx-design
myst-nb>=1.0.0

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

# Packages used for notebook execution
matplotlib
scikit-learn
numpy
rich[jupyter]
.[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: pin sphinx to >=7.3.2 2024-04-17 09:51:27 -07:00			`sphinx>=7.3.2 # 7.3.0 breaks sphinx-book-theme`
DOC: pin newest sphinx-book-theme 2023-03-31 08:42:34 -07:00			`sphinx-book-theme>=1.0.1 # Older versions fail to pin pydata-sphinx-theme`
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: 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`
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]`
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.`