raymondyeh07

Raymond Yeh's repositories

[CVPR2022] Total Variation Optimization Layers for Computer Vision

Language:CudaMIT43 4 4

[NeurIPS2019] Chirality Nets for Human Pose Regression

Language:PythonMIT23 40

[NeurIPS 2022] Learnable Polyphase Sampling for Shift Invariant and Equivariant Convolutional Networks

Language:PythonMIT11 40

[AISTATS2022] Equivariance Discovery by Learned Parameter-Sharing

Language:PythonMIT2 20

Flax is a neural network library for JAX that is designed for flexibility.

010

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Language:PythonApache-2.0010

Language:Python010

020