Giters

yinxx / jax

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

Home Page:http://jax.readthedocs.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This repository is not active

About

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

http://jax.readthedocs.io/

License:Apache License 2.0

Languages

Language:Python 91.5%Language:C++ 6.2%Language:Starlark 1.3%Language:Jupyter Notebook 0.7%Language:Shell 0.2%Language:C 0.1%Language:MAXScript 0.0%