CMD + K
njzjz/jaxlib

jaxlib

Community

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

Installation

To install this package, run one of the following:

Conda
$conda install njzjz::jaxlib

Usage Tracking

0.4.34
1 / 8 versions selected
Downloads (Last 6 months): 0

About

Summary

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

Last Updated

Nov 23, 2024 at 07:10

License

Apache-2.0

Supported Platforms

linux-64

Home

http://github.com/google/jax

GitHub Repository

http://github.com/google/jax