brohrer / sharpened-cosine-similarity

Related projects: ⓘ

• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆247Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• louislva / deepmind-perceiver
  My implementation of DeepMind's Perceiver
  ☆63Updated 3 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆150Updated 2 years ago
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆199Updated last year
• OATML / RHO-Loss
  ☆182Updated last year
• lucidrains / attention
  ☆192Updated this week
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆276Updated 3 months ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆379Updated 2 months ago
• lkhphuc / lightning-hydra-template
  Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.
  ☆153Updated 3 years ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆234Updated last year
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆173Updated 2 years ago
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆419Updated 3 weeks ago
• rwightman / efficientnet-jax
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆126Updated 8 months ago
• ml-jku / hopular
  Hopular: Modern Hopfield Networks for Tabular Data
  ☆299Updated 2 years ago
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆311Updated last year
• unixpickle / sk2torch
  Convert scikit-learn models to PyTorch modules
  ☆157Updated 4 months ago
• sayakpaul / probing-vits
  Probing the representations of Vision Transformers.
  ☆311Updated last year
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆103Updated 3 years ago
• mle-infrastructure / mle-hyperopt
  Lightweight Hyperparameter Optimization 🚂
  ☆145Updated 3 weeks ago
• HomebrewNLP / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆123Updated 2 years ago
• IssamLaradji / sps
  Official code for the Stochastic Polyak step-size optimizer
  ☆137Updated 2 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆215Updated last year
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 2 years ago
• infocusp / diffusion_models
  Minimal standalone example of diffusion model
  ☆154Updated 2 years ago
• brohrer / parameter_efficiency_leaderboard
  The most parameter efficient machine learning models on a few popular benchmarks
  ☆42Updated 2 years ago
• KeremTurgutlu / self_supervised
  Implementation of popular SOTA self-supervised learning algorithms as Fastai Callbacks.
  ☆318Updated last year
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago