brentyi

tyro

Zero-effort CLI interfaces & config objects, from types

jaxlie

Rigid transforms + Lie groups in JAX

jaxls

Factor graphs and nonlinear least squares for JAX

tilted

Canonical Factors for Hybrid Neural Fields @ ICCV 2023

dfgo

Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021

jax_dataclasses

Pytrees + dataclasses ❤️

tensorf-jax

Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)

pips-jax

JAX port of Persistent Independent Particles

dotfilesp

Configuration files & setup scripts

fifteen

jaxfg

new version: https://github.com/brentyi/jaxls

NeMF

[NeurIPS 2022] Official implementation of "NeMF: Neural Motion Fields for Kinematic Animation"

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.

jaxtyping-export-repro

jekyll-prettier

K-Planes

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.

optax

Optax is a gradient processing and optimization library for JAX.

pips

Particle Video Revisited

projectaria_tools-stubs

server

server for localtunnel.me

SimpleParsing

Simple, Elegant, Typed Argument Parsing with argparse

slahmr

trl

Train transformer language models with reinforcement learning.

unimatch

[TPAMI'23] Unifying Flow, Stereo and Depth Estimation