Brent Yi's repositories
jax_dataclasses
Pytrees + dataclasses ❤️
tensorf-jax
Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)
accelerate
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
Adafruit_CircuitPython_BLE
Bluetooth Low Energy (BLE) library for CircuitPython
body_visualizer
Visualization code for SMPL body model Family
einops
Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
jaxtyping
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
LGM
LGM: Large Multi-View Gaussian Model for High-Resolution 3D Content Creation.
nanorl
A tiny reinforcement learning codebase for continuous control, built on top of JAX.
nerfstudio_MSI
A collaboration friendly studio for NeRFs
optax
Optax is a gradient processing and optimization library for JAX.
pips
Particle Video Revisited
server
server for localtunnel.me
SimpleParsing
Simple, Elegant, Typed Argument Parsing with argparse
trl
Train transformer language models with reinforcement learning.
unimatch
[TPAMI'23] Unifying Flow, Stereo and Depth Estimation