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

Image-to-Image Translation in PyTorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

100 numpy exercises (with solutions)

Minimal reproduction of DeepSeek R1-Zero

Trax — Deep Learning with Clear Code and Speed

Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and primitives.

Python framework for creating, editing, and running Noisy Intermediate-Scale Quantum (NISQ) circuits.

LPIPS metric. pip install lpips

Lenia - Mathematical Life Forms

A TensorFlow implementation of the Differentiable Neural Computer.

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Differentiable convex optimization layers

The python library for working with the ESPARGOS WiFi channel sounder

A platform for managing machine learning experiments

💬 Slides and supplementary codes for my talk 'Debugging Tips on TensorFlow' (2016)

chatbot does what you ask, like open Google search, post a Tweet, etc.

Easily convert common crawl to a dataset of caption and document. Image/text Audio/text Video/text, ...

Simple Tensorflow implementation of pre-activation ResNet18, 34, 50, 101, 152

a tool to visualize the time distribution of commits

the official repo for "D-AR: Diffusion via Autoregressive Models"

High-acc(>0.7) model(ResNet, ResNeXt, DenseNet, SENet, SE-ResNeXt) on TensorFlow.

Seamlessly integrate matplotlib figures as tensorflow summaries.

TF2.0 port for Augmix paper