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

The python library for working with the ESPARGOS WiFi channel sounder

Minimal reproduction of DeepSeek R1-Zero

The code for the Ensemble everything everywhere: Multi-scale aggregation for adversarial robustness paper

Replicating the basic adversarial multi-attack experimental results

Graphic notes on Gilbert Strang's "Linear Algebra for Everyone"

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

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

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

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

A platform for managing machine learning experiments

a cheat-sheet for mathematical notation in code form

!!Con West 2019 talk

Low-Cost Open Source Ventilator or PAPR

TF2.0 port for Augmix paper

Lenia - Mathematical Life Forms

Trax — Deep Learning with Clear Code and Speed

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

Standard basic quantum circuit simulator building blocks. (archived, for it is moved to Yao.jl)

Python bindings for the Flatiron Institute Non-Uniform Fast Fourier Transform library

ACT community resources

Differentiable convex optimization layers

"Neural Turing Machine" in Tensorflow

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

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

A Jupyter/python-based introduction to writing a finite-volume hydro code

A TensorFlow implementation of the Differentiable Neural Computer.

Seamlessly integrate matplotlib figures as tensorflow summaries.