JAX

https://github.com/jax-ml/jax

JAX is a Python library for accelerator-oriented array computation and program transformation,...

Tokens:607,684
Snippets:6,017
Trust Score:8.8
License:Apache-2.0
Update:3 days ago
Tokens:
Raw