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.
☆34Updated last week
Alternatives and similar repositories for LitModels
Users that are interested in LitModels are comparing it to the libraries listed below
Sorting:
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆243Updated last month
- Create powerful Hydra applications without the yaml files and boilerplate code.☆419Updated last week
- A simple python package to solve all your problems with pythonpath, work dir, file paths, module imports and environment variables.☆172Updated 2 years ago
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆61Updated this week
- TensorDict is a pytorch dedicated tensor container.☆979Updated 2 weeks ago
- Speed up model training by fixing data loading.☆556Updated last week
- TorchFix - a linter for PyTorch-using code with autofix support☆148Updated 2 months ago
- Running Jax in PyTorch Lightning☆113Updated 10 months ago
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 11 months ago
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆136Updated last year
- Universal Notation for Tensor Operations in Python.☆447Updated 7 months ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆65Updated 2 weeks ago
- A tiny library for stochastic dataset caching in PyTorch.☆44Updated last year
- PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.☆25Updated last month
- Use Jax functions in Pytorch☆253Updated 2 years ago
- A flexible template, as a quick setup for deep learning projects in pytorch-lightning☆48Updated 2 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆223Updated 3 years ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 10 months ago
- Install PyTorch distributions with computation backend auto-detection☆255Updated 2 months ago
- Quickest way to share everything about your research within a single app☆37Updated last year
- Quickest way to share everything about your research within a single app☆16Updated last year
- Collection of Pytorch lightning tutorial form as rich scripts automatically transformed to ipython notebooks.☆318Updated 3 months ago
- The repository hosting example notebooks for 2024 CZII cryoET machine learning challenge☆38Updated last year
- Lightning-like training API for JAX with Flax☆44Updated 11 months ago
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆192Updated 4 months ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆288Updated last year
- ☆63Updated last year