neverix

dietgpu

GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compression of numerical and other data types in HPC/ML applications.

alphafold_pytorch

An implementation of the DeepMind's AlphaFold based on PyTorch for research

Streetview-panorama-scraping

This module helps you scrape panoramas from Google's streetview

Human3D

We propose the first multi-human body-part segmentation model, called Human3D 🧑‍🤝‍🧑, that directly operates on 3D scenes. In an extensive analysis, we validate the benefits of training on synthetic data on multiple baselines and tasks.

FlowDiffuser

PyTorch implementation of FlowDiffuser: Advancing Optical Flow Estimation with Diffusion Models (CVPR-2024)

mobilestereonet

Lightweight stereo matching network based on MobileNet blocks

CircuitSim

Basic Circuit Simulator

optax-galore

Adds GaLore style projection wrappers to optax optimizers

yew

Rust / Wasm framework for creating reliable and efficient web applications

jafar

JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"

sophus-rs

2d and 3d geometry for Computer Vision and Robotics

mishax

OpenTPU

A open source reimplementation of Google's Tensor Processing Unit (TPU).

mdlm

Simplified Masked Diffusion Language Model

mpfb2

MPFB2 is a free and open source human generator for Blender

ELIZA-Transformer

Representing Rule-based Chatbots with Transformers

Python-WORLD

IC-Light

More relighting!

AQLM

Official Pytorch repository for Extreme Compression of Large Language Models via Additive Quantization https://arxiv.org/pdf/2401.06118.pdf and PV-Tuning: Beyond Straight-Through Estimation for Extreme LLM Compression https://arxiv.org/abs/2405.14852

bolt

10x faster matrix and vector operations

aqt

autograd

Efficiently computes derivatives of NumPy code.

FOLNet

This repository includes the code for First-Order Logic Network (FOLNet).

feature-dashboard

GeneralizedOptimalSparseDecisionTrees

Generalized Optimal Sparse Decision Trees

re-arc

Reverse Engineering the Abstraction and Reasoning Corpus

arc-dsl

Domain Specific Language for the Abstraction and Reasoning Corpus

ARC-solution

Code for 1st place solution to Kaggle's Abstraction and Reasoning Challenge

PIGEON

Code for the CVPR 2024 paper highlight and demo "PIGEON: Predicting Image Geolocations".

modula

Scalable neural net training via automatic normalization in the modular norm.