mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-09 11:24:55 +08:00
Roman Knyazhitskiy
172b83b4fc
* Add a test for generators * Remove output annotations from decorators Also guarded torch imports for better compatibility with requirements.txt * Add flag to the main meta class to skip the typecheck * Return to the old solution * Make async tests work * Minor adjustments/fixing typos * Correct Python path for new tests * Remove some jax-dependent code * Implement equality for MetaArrays * Make all Dim variations frozen dataclasses * Shorten AbstractArray methods * Final touches * Removing get_origin use * Update tests with @jaxtyp
10 lines
89 B
Plaintext
10 lines
89 B
Plaintext
beartype
|
|
cloudpickle
|
|
equinox
|
|
IPython
|
|
jaxlib
|
|
pytest
|
|
pytest-asyncio
|
|
tensorflow
|
|
typeguard<3
|