Repository Details
Shared by
HelloGitHub Rating
0 ratings
Free•Apache-2.0
Claim
Discuss
Collect
Share
30.4k
Stars
No
Chinese
Python
Language
Yes
Active
704
Contributors
2k
Issues
Yes
Organization
jax-0.4.35
Latest
3k
Forks
Apache-2.0
License
More
This is a Python library for numerical computation that combines Just-In-Time (JIT) compilation, automatic differentiation (Autograd), and a linear algebra compiler (XLA), with a usage pattern similar to NumPy. However, JAX is faster, more memory-efficient, and supports automatic differentiation, automatic vectorization, parallel computing, and more.
Comments
Rating:
No comments yet