Tianjian Lu's repositories
google-research
Google Research
datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Language:PythonApache-2.0000
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:PythonApache-2.0000
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
Language:PythonMIT000
tensorflow
Computation using data flow graphs for scalable machine learning
Language:C++Apache-2.0000