ZeWang95 / scs_pytorch
PyTorch implementation of 2D Sharpened Cosine Similarity layer
☆17Updated 2 years ago
Related projects: ⓘ
- ☆33Updated last year
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- A convolution-free, transformer-only version of the CycleGAN framework☆32Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 2 years ago
- FID computation in Jax/Flax.☆23Updated 2 months ago
- ☆35Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆50Updated 2 years ago
- Tensorflow 2.x implementation of Gradient Origin Networks☆13Updated 4 years ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated last year
- ☆40Updated this week
- A simple Transformer where the softmax has been replaced with normalization☆17Updated 4 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 2 years ago
- Convolutions and more as einsum for PyTorch☆12Updated 3 months ago
- ☆21Updated last year
- Official repository for MaGNET, ICLR 2022☆26Updated last year
- High performance pytorch modules☆18Updated last year
- ☆21Updated this week
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- ☆16Updated 3 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Updated 2 years ago
- ☆16Updated 2 years ago
- A JAX nn library☆20Updated 6 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- ☆12Updated this week
- Implementation of the ALI-G algorithm (PyTorch, Tensorflow)☆22Updated 3 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year