jndean / LossRider

Related projects ⓘ

Alternatives and complementary repositories for LossRider

• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆333Updated last week
• BlackHC / neural_net_checklist
  ☆139Updated 2 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆227Updated this week
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆106Updated last month
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆282Updated last week
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆417Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆832Updated this week
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆216Updated 4 months ago
• warner-benjamin / commented-transformers
  Highly commented implementations of Transformers in PyTorch
  ☆128Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆171Updated 2 months ago
• 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 2 weeks ago
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆479Updated 2 weeks ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆323Updated last year
• srush / Autodiff-Puzzles
  ☆388Updated 3 weeks ago
• pmeier / light-the-torch
  Install PyTorch distributions with computation backend auto-detection
  ☆220Updated last year
• apple / ml-sigma-reparam
  ☆292Updated 4 months ago
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆1,089Updated last month
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆161Updated 9 months ago
• LinearBoost / linearboost-classifier
  LinearBoost Classifier is a rapid and accurate classification algorithm that builds upon a very fast, linear classifier.
  ☆116Updated 5 months ago
• google-ai-edge / model-explorer
  A modern model graph visualizer and debugger
  ☆1,046Updated this week
• aai-institute / beyond-jupyter
  Software design principles for machine learning applications
  ☆285Updated 2 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆326Updated last month
• thomasahle / tensorgrad
  Tensor Network Library with Autograd
  ☆148Updated last week
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
  ☆153Updated this week
• SebChw / Actually-Robust-Training
  Actually Robust Training - Tool Inspired by Andrej Karpathy "Recipe for training neural networks". It allows you to decompose your Deep…
  ☆44Updated 7 months ago
• amakelov / mandala
  A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python
  ☆517Updated 2 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆252Updated 5 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆1,889Updated last week
• Lightning-AI / litdata
  Transform datasets at scale. Optimize datasets for fast AI model training.
  ☆362Updated this week
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆895Updated 10 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆84Updated 2 months ago