Anna Shors's repositories
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:PythonApache-2.0000
JAX-Toolbox
JAX-Toolbox
Language:ShellApache-2.0000
Language:PythonApache-2.0000
Language:PythonApache-2.0000
Language:PythonApache-2.0000
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++Apache-2.0000