An extremely fast Python linter and code formatter, written in Rust.

Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc

PyTorch implementation of normalizing flow models

Normalizing flows in PyTorch

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

EasyTPP: Towards Open Benchmarking Temporal Point Processes

Trax — Deep Learning with Clear Code and Speed

Hackable and optimized Transformers building blocks, supporting a composable construction.

🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.

One of the "BEST" markdown preview extensions for Visual Studio Code

Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch

A modular graph-based Retrieval-Augmented Generation (RAG) system

Graph Neural PDEs

Code for CRATE (Coding RAte reduction TransformEr).

PyTorch native quantization and sparsity for training and inference

GraphAny: A foundation model for node classification on any graph.

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

☗ lishogi.org: the forever free, adless and open source shogi server forked from lichess.org ☗

♞ lichess.org: the forever free, adless and open source chess server ♞

Take potentially dangerous PDFs, office documents, or images and convert them to safe PDFs

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT (CVPR'24), OA-CNNs (CVPR'24), MSC (CVPR'23)

Distribute and run LLMs with a single file.

Geometric kernels on manifolds, meshes and graphs

Open-Source Battle Simulator, Rankings & Team Building for Pokemon GO PvP

Contains code for the NeurIPS 2019 paper "Practical Deep Learning with Bayesian Principles"

IVON optimizer for neural networks based on variational learning.

oneAPI Math Kernel Library (oneMKL) Interfaces