BlackHC / neural_net_checklist
☆150Updated 8 months ago
Alternatives and similar repositories for neural_net_checklist:
Users that are interested in neural_net_checklist are comparing it to the libraries listed below
- An implementation of PSGD Kron second-order optimizer for PyTorch☆89Updated last month
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆137Updated last month
- Diffusion models in PyTorch☆101Updated last month
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆367Updated 3 weeks ago
- TorchDR - PyTorch Dimensionality Reduction☆108Updated last week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 9 months ago
- Scalable and Performant Data Loading☆247Updated this week
- 🧱 Modula software package☆188Updated last month
- Exca - Execution and caching tool for python☆80Updated this week
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 7 months ago
- Efficient optimizers☆190Updated this week
- supporting pytorch FSDP for optimizers☆80Updated 4 months ago
- ☆217Updated 9 months ago
- ☆177Updated 5 months ago
- Uncertainty quantification with PyTorch☆354Updated 3 weeks ago
- σ-GPT: A New Approach to Autoregressive Models☆63Updated 8 months ago
- ☆301Updated 10 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆271Updated 10 months ago
- Highly commented implementations of Transformers in PyTorch☆136Updated last year
- WIP☆93Updated 8 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆101Updated 3 weeks ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆99Updated 4 months ago
- ☆43Updated last month
- Getting crystal-like representations with harmonic loss☆182Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆78Updated 11 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆115Updated 2 months ago
- ☆105Updated this week
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- Run PyTorch in JAX. 🤝☆239Updated 2 months ago
- The boundary of neural network trainability is fractal☆198Updated last year