Enabling PyTorch on XLA Devices (e.g. Google TPU)

Source code for the userspace level runtime driver for Coral.ai devices.

Ecosystem of libraries and tools for writing and executing fast GPU code fully in Rust.

Extracting Tables from Document Images using a Multi-stage Pipeline for Table Detection and Table Structure Recognition

NeRF Meta-Learning with PyTorch

ArduPlane, ArduCopter, ArduRover, ArduSub source

Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"

Plenoxels: Radiance Fields without Neural Networks

Code release for NeRF (Neural Radiance Fields)

A library for efficient similarity search and clustering of dense vectors.

A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.

GLIDE: a diffusion-based text-conditional image synthesis model

Implementation of "Gated Context Aggregation Network for Image Dehazing and Deraining"

Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)

Code release for SLIP Self-supervision meets Language-Image Pre-training

Rayon: A data parallelism library for Rust

An efficient single/multi-agent trajectory planner for multicopters.

Offline speech recognition API for Android, iOS, Raspberry Pi and servers with Python, Java, C# and Node

SIMD-accelerated operations

Reproduce results and replicate training fo T0 (Multitask Prompted Training Enables Zero-Shot Task Generalization)

Real-time LIDAR-based Urban Road and Sidewalk detection for Autonomous Vehicles 🚗

Chess + Rust <3. Chess engine written in Rust.

Codebase for the solution that won first place and was awarded the most human-like agent in the 2021 NeurIPS Competition MineRL BASALT Challenge.

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

Instant neural graphics primitives: lightning fast NeRF and more

Code release for ConvNeXt model