johnmarktaylor91 / torchlens

Related projects ⓘ

Alternatives and complementary repositories for torchlens

• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆325Updated last year
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆457Updated 8 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆449Updated this week
• BlackHC / neural_net_checklist
  ☆139Updated 3 months ago
• mert-kurttutan / torchview
  torchview: visualize pytorch models
  ☆831Updated 3 weeks ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆246Updated last year
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆840Updated this week
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆328Updated last month
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,113Updated 3 weeks 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 3 weeks ago
• google-deepmind / nanodo
  ☆197Updated 4 months ago
• srush / Autodiff-Puzzles
  ☆391Updated last month
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• soniajoseph / ViT-Prisma
  ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).
  ☆179Updated this week
• spfrommer / torchexplorer
  Interactively inspect module inputs, outputs, parameters, and gradients.
  ☆315Updated 5 months ago
• KellerJordan / cifar10-airbench
  94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds
  ☆177Updated last week
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆903Updated 10 months ago
• google / learned_optimization
  ☆751Updated this week
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆325Updated last week
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆394Updated 10 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
• EleutherAI / cookbook
  Deep learning for dummies. All the practical details and useful utilities that go into working with real models.
  ☆715Updated last month
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆252Updated 5 months ago
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆866Updated 7 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆469Updated 3 weeks ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆516Updated this week
• jxbz / agd
  Automatic gradient descent
  ☆206Updated last year
• 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
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆172Updated 2 months ago