Lightning-AI / LitModelsLinks
Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-grade access controls.
☆30Updated last week
Alternatives and similar repositories for LitModels
Users that are interested in LitModels are comparing it to the libraries listed below
Sorting:
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆56Updated this week
- TorchFix - a linter for PyTorch-using code with autofix support☆143Updated 4 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆122Updated 11 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 11 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆233Updated 5 months ago
- deep learning with pytorch lightning☆1Updated 7 months ago
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- VIT inference in triton because, why not?☆29Updated last year
- Implementation of Infini-Transformer in Pytorch☆111Updated 5 months ago
- Scalable and Performant Data Loading☆278Updated this week
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆41Updated 4 months ago
- ☆61Updated last year
- For optimization algorithm research and development.☆520Updated this week
- This is a port of Mistral-7B model in JAX☆32Updated 11 months ago
- A repository to house some personal attempts to beat some state-of-the-art for medical datasets☆99Updated last year
- Additional multi-backend functionality for Keras 3.☆16Updated last year
- Running Jax in PyTorch Lightning☆102Updated 6 months ago
- ☆12Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆91Updated 2 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- ☆133Updated last year
- PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.☆23Updated last month
- Automate issue discovery for your projects against Lightning nightly and releases.☆46Updated last month
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise☆36Updated 9 months ago
- ☆150Updated 10 months ago
- ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"☆96Updated this week
- 🧼🔎 A holistic self-supervised data cleaning strategy to detect irrelevant samples, near duplicates and label errors (NeurIPS'24).☆35Updated 3 months ago
- ☆55Updated 7 months ago
- A functional training loops library for JAX☆88Updated last year