LokeshBadisa / txv

Related projects ⓘ

Alternatives and complementary repositories for txv

• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆325Updated last week
• JUSTSUJAY / ML-Research-Papers
  ☆100Updated 2 months ago
• BlackHC / neural_net_checklist
  ☆139Updated 3 months ago
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
  ☆154Updated this week
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆251Updated 7 months ago
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆111Updated last month
• ironjr / grokfast
  Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
  ☆515Updated 4 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆161Updated 9 months ago
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆207Updated this week
• jacobgil / confidenceinterval
  The long missing library for python confidence intervals
  ☆132Updated 5 months ago
• MathiasKraus / igann
  ☆42Updated 2 weeks ago
• p-giakoumoglou / pyssl
  Self-Supervised Learning in PyTorch
  ☆130Updated 8 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆451Updated this week
• FlorianDietz / comgra
  A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…
  ☆270Updated 3 weeks ago
• balthazarneveu / mva_pepites
  Access to free kaggle compute power from your command line
  ☆25Updated 10 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆172Updated 2 months ago
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆484Updated last week
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆1,898Updated 2 weeks ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆337Updated 3 weeks ago
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆285Updated 5 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆663Updated 6 months ago
• cabralpinto / modular-diffusion
  Python library for designing and training your own Diffusion Models with PyTorch.
  ☆266Updated 3 months ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• ThomasVrancken / info9023-mlops
  Repo centralising material for the ULiege course INFO9023 "Machine Learning Systems Design (MLOps)".
  ☆103Updated 2 weeks ago
• kozistr / pytorch_optimizer
  optimizer & lr scheduler & loss function collections in PyTorch
  ☆237Updated last week
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆584Updated 4 months ago
• thomasahle / tensorgrad
  Tensor Network Library with Autograd
  ☆148Updated 2 weeks ago
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆80Updated this week
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆203Updated last year
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,113Updated this week