mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
5a340a9781
Description: - Disable second order vjp tests in RunStateHypothesisTest.test_vjp if JAX_SKIP_SLOW_TESTS=true to reduce the test execution time - especially for TSAN CI job where this test takes ~700 seconds to pass with the recent 3.13 cpython - Removed optional deps for 3.14
22 lines
330 B
Plaintext
22 lines
330 B
Plaintext
--pre
|
|
--extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
|
|
numpy
|
|
|
|
--pre
|
|
--extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
|
|
scipy
|
|
|
|
absl-py==2.1.0
|
|
|
|
attrs==24.3.0
|
|
|
|
hypothesis==6.123.9
|
|
|
|
sortedcontainers==2.4.0
|
|
|
|
flatbuffers==24.12.23
|
|
|
|
ml-dtypes==0.5.1
|
|
|
|
opt-einsum==3.4.0
|