BlackHC / neural_net_checklistLinks
☆150Updated last year
Alternatives and similar repositories for neural_net_checklist
Users that are interested in neural_net_checklist are comparing it to the libraries listed below
Sorting:
- An implementation of PSGD Kron second-order optimizer for PyTorch☆96Updated last month
- Getting crystal-like representations with harmonic loss☆194Updated 5 months ago
- Diffusion models in PyTorch☆110Updated 2 months ago
- The boundary of neural network trainability is fractal☆215Updated last year
- ☆279Updated last year
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆146Updated 3 months ago
- ☆210Updated 9 months ago
- Efficient optimizers☆261Updated last month
- Exca - Execution and caching tool for python☆100Updated last week
- ☆307Updated last year
- 🧱 Modula software package☆237Updated 3 weeks ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆120Updated 6 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆180Updated last week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆124Updated last year
- Scalable and Performant Data Loading☆299Updated this week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆291Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆408Updated 5 months ago
- supporting pytorch FSDP for optimizers☆84Updated 9 months ago
- Highly commented implementations of Transformers in PyTorch☆136Updated 2 years ago
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- For optimization algorithm research and development.☆534Updated last week
- ☆65Updated 10 months ago
- ☆118Updated 3 months ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- Annotated version of the Mamba paper☆489Updated last year
- Easily run PyTorch on multiple GPUs & machines☆47Updated 2 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 8 months ago
- Because we don't have enough time to read everything☆89Updated 11 months ago