brohrer / sharpened-cosine-similarity
An alternative to convolution in neural networks
☆254Updated 11 months ago
Alternatives and similar repositories for sharpened-cosine-similarity:
Users that are interested in sharpened-cosine-similarity are comparing it to the libraries listed below
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- My implementation of DeepMind's Perceiver☆63Updated 3 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Memory mapped numpy arrays of varying shapes☆295Updated 9 months ago
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆126Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month
- ☆376Updated last year
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆206Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 10 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.☆156Updated 4 years ago
- Probing the representations of Vision Transformers.☆321Updated 2 years ago
- Lite Inference Toolkit (LIT) for PyTorch☆161Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆161Updated 10 months ago
- Check if you have training samples in your test set☆64Updated 2 years ago
- Minimal standalone example of diffusion model☆156Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆411Updated 2 weeks ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated 9 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆211Updated 2 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- Lightweight Hyperparameter Optimization 🚂☆145Updated 6 months ago
- ☆201Updated 2 years ago
- Functional deep learning☆108Updated 2 years ago