Описание тега jax

Описание тега Вопросы с тегом
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

Документация: https://jax.readthedocs.io/

Репо проекта: https://github.com/google/jax