melissawm / jax

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

http://jax.readthedocs.io/

Repository from Github https://github.commelissawm/jax

This repository is not active

About

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

http://jax.readthedocs.io/

Apache License 2.0

Languages

Language:Python 84.7%Language:C++ 12.6%Language:Starlark 1.9%Language:Jupyter Notebook 0.4%Language:Shell 0.3%Language:C 0.1%Language:MAXScript 0.0%Language:Makefile 0.0%Language:Linker Script 0.0%