Alternatives and similar repositories for TensorHue:

Users that are interested in TensorHue are comparing it to the libraries listed below

• BlackHC / neural_net_checklist
  ☆149Updated 5 months ago
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
  ☆191Updated this week
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆119Updated 6 months ago
• nikhilvyas / SOAP
  ☆149Updated last month
• modula-systems / modula
  🧱 Modula software package
  ☆134Updated this week
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆178Updated 4 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆412Updated last month
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆344Updated 2 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆486Updated last week
• apple / ml-sigma-reparam
  ☆296Updated 7 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…
  ☆279Updated last month
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆210Updated this week
• KellerJordan / cifar10-airbench
  94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds
  ☆197Updated 2 months ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆122Updated 3 weeks ago
• KellerJordan / Muon
  Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead
  ☆220Updated 3 weeks ago
• apoorvkh / academic-pretraining
  $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources
  ☆118Updated 2 weeks ago
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated 5 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆603Updated this week
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆470Updated 11 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆65Updated last week
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆539Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆868Updated this week
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆75Updated last month
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆331Updated last year
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆75Updated 3 weeks ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆89Updated 3 months ago
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆154Updated this week
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆203Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• google-deepmind / nanodo
  ☆203Updated 6 months ago