Qiumin Xu's repositories
GraphApplications
Graph Applications in CUDA
dm-haiku
JAX-based neural network library
Language:PythonApache-2.0000
examples
TensorFlow examples
Language:Jupyter NotebookApache-2.0000
flax
Flax is a neural network ecosystem for JAX that is designed for flexibility.
Language:PythonApache-2.0000
haiku
The Haiku operating system. (Pull requests will be ignored; patches may be sent to https://review.haiku-os.org).
Language:C++NOASSERTION000
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:PythonApache-2.0000
tensorboard
TensorFlow's Visualization Toolkit
tensorflow
Computation using data flow graphs for scalable machine learning