Each file is run separately to avoid JAX out-of-memory'ing.
As such, run tests using python -m tests, not by just running pytest.
tests
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||
Each file is run separately to avoid JAX out-of-memory'ing.
As such, run tests using python -m tests, not by just running pytest.