An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A library that allows for inference on probabilistic models

VIP cheatsheets for Stanford's CS 230 Deep Learning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Code for our NeurIPS 2022 paper

Fast and Easy Infinite Neural Networks in Python

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well a…

Flax is a neural network library for JAX that is designed for flexibility.

A playbook for systematically maximizing the performance of deep learning models.

Implementation of Muse: Text-to-Image Generation via Masked Generative Transformers, in Pytorch

tree is a library for working with nested data structures

Examples and guides for using the OpenAI API

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

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

Echo State Networks in Python

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

AMD-SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution

Deep Active Learning

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

PyBoolNet is a Python package for the generation, modification and analysis of Boolean networks.

Evals is a framework for evaluating LLMs and LLM systems, and an open-source registry of benchmarks.

Code release for "Dropout Reduces Underfitting"

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

CLU lets you write beautiful training loops in JAX.

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation

Code and experiments for "Deep Neural Networks for Rank Consistent Ordinal Regression based on Conditional Probabilities"

An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.