AI Agent Skills

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Rigid transforms + Lie groups for JAX

JAX-based neural network library

PIX is an image processing library in JAX, for JAX.

Optax is a gradient processing and optimization library for JAX.

Jax Official Implementation of T-RO Paper: Songyuan Zhang*, Oswin So*, Kunal Garg, Chuchu Fan: "GCBF+: A Neural Graph Control Barrier Function Framework for Distributed…