Alternatives and similar repositories for RHO-Loss:

Users that are interested in RHO-Loss are comparing it to the libraries listed below

• lucidrains / Mega-pytorch
  Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena
  ☆204Updated last year
• ctlllll / SGConv
  ☆163Updated 2 years ago
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆203Updated last year
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆209Updated 9 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• google-research / diffstride
  TF/Keras code for DiffStride, a pooling layer with learnable strides.
  ☆124Updated 2 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆251Updated 2 years ago
• frederick0329 / TracIn
  Implementation of Estimating Training Data Influence by Tracing Gradient Descent (NeurIPS 2020)
  ☆226Updated 2 years ago
• MadryLab / trak
  A fast, effective data attribution method for neural networks in PyTorch
  ☆189Updated 2 months ago
• libffcv / ffcv-imagenet
  Train ImageNet *fast* in 500 lines of code with FFCV
  ☆136Updated 8 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆153Updated 2 years ago
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆97Updated last year
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆210Updated last year
• AnanyaKumar / transfer_learning
  Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…
  ☆147Updated 2 years ago
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆311Updated last year
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆80Updated 3 years ago
• nikhilvyas / SOAP
  ☆149Updated last month
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• facebookresearch / mega
  Sequence modeling with Mega.
  ☆297Updated 2 years ago
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆257Updated last year
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆66Updated 2 years ago
• AminRezaei0x443 / memory-efficient-attention
  Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
  ☆180Updated 2 years ago
• HomebrewNLP / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆125Updated 2 years ago
• themrzmaster / git-re-basin-pytorch
  Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch
  ☆73Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆154Updated last month
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆61Updated 2 years ago
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆165Updated 6 months ago
• Felix-Petersen / diffsort
  Differentiable Sorting Networks
  ☆109Updated last year
• lucidrains / nystrom-attention
  Implementation of Nyström Self-attention, from the paper Nyströmformer
  ☆123Updated last year