6 projects
s2vit
Spatial Shift Vision Transformer
x-mlps
Configurable MLPs built on JAX and Haiku
acni
A Jax/Optax implementation of Anticorrelated Noise Injection for Improved Generalization
pypk
An opinionated Python package template generator
ctorch
Complex Neural Networks for PyTorch
libpyrite
PyTorch utility library