mayuyuace

Pinned Repositories

jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python00
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++00
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python30.1k2.8k
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++2.6k409
model-explorer
A modern model graph visualizer and debugger
Language:JavaScript1k76
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++00

mayuyuace's Repositories

mayuyuace/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
mayuyuace/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++