jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

PythonStars 34318Forks 3302Watchers 34318Open issues 2274License Apache License 2.0

Details

仓库信息

Ownerjax-ml

Last pushed2025-12-15

Last updated2025-12-15

Issues fetched at—

Community at a glance