macio232

functa

Papers-in-100-Lines-of-Code

Implementation of papers in 100 lines of code.

cornell-cs5785-2024-applied-ml

Lecture materials for Cornell CS5785 Applied Machine Learning (Fall 2024)

resnet1d

PyTorch implementations of several SOTA backbone deep neural networks (such as ResNet, ResNeXt, RegNet) on one-dimensional (1D) signal/time-series data.

matplotlib-latex-bridge

Functions and shortcuts to create latex-ready images from matplotlib.

torch-fidelity

High-fidelity performance metrics for generative models in PyTorch

ANR

Released code for paper: Attention Beats Linear for Fast Implicit Neural Representation Generation. Attention-based localized implicit Neural Representation (ANR)

paramit

Parameterize Python scripts/notebooks all from the command line and run on cloud GPUs

corner.py

Make some beautiful corner plots

Imagenet32_Scripts

Scripts for Imagenet 32 dataset

penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

DeepSeek-Math

DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models

StructuredNNs

Structured Neural Networks

torchpairwise

Pairwise Metrics for PyTorch

theseus

A library for differentiable nonlinear optimization

Mixed-Curvature-GCN

Graph Convolutional Networks on Cartesian products of constant curvature manifolds.

TorchCP

A Python toolbox for conformal prediction research on deep learning models, using PyTorch.

pytorch-tensor-tracker

Flexibly track outputs and grad-outputs of torch.nn.Module.

convex-init

Code for Principled Weight Initialisation for Input-Convex Neural Networks

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

presentations

gatsbi

GATSBI: Generative Adversarial Training for Simulation-Based Inference

pymanopt

Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation

normalizing-flows

PyTorch implementation of normalizing flow models

nevergrad

A Python toolbox for performing gradient-free optimization

torchgfn

A modular, easy to extend GFlowNet library

AIRS

Artificial Intelligence Research for Science (AIRS)

PowerInfer

High-speed Large Language Model Serving on PCs with Consumer-grade GPUs

brevitas

Brevitas: neural network quantization in PyTorch

stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.