代码库

patrick-kidger/equinox

2.9k

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Python

deep-learningequinoxjaxneural-networks

patrick-kidger/jaxtyping

1.8k

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Python

jaxpython-typingtyping