tshu-w / lightning-templateLinks
PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.
☆25Updated 3 months ago
Alternatives and similar repositories for lightning-template
Users that are interested in lightning-template are comparing it to the libraries listed below
Sorting:
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆66Updated last week
- TorchDR - PyTorch Dimensionality Reduction☆145Updated last week
- Running Jax in PyTorch Lightning☆111Updated 8 months ago
- ☆62Updated last year
- Lightning-like training API for JAX with Flax☆42Updated 8 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆100Updated last week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆124Updated last year
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆57Updated this week
- A repository to house some personal attempts to beat some state-of-the-art for medical datasets☆99Updated last year
- Convert scikit-learn models to PyTorch modules☆163Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- ☆150Updated last year
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆54Updated 2 years ago
- 🧼🔎 A holistic self-supervised data cleaning strategy to detect irrelevant samples, near duplicates and label errors (NeurIPS'24).☆36Updated 5 months ago
- Quickest way to share everything about your research within a single app☆37Updated 10 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆133Updated last year
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆190Updated 2 months ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆19Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated last year
- Create powerful Hydra applications without the yaml files and boilerplate code.☆394Updated last week
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 3 years ago
- Example of how to use Weights & Biases on Slurm☆117Updated 3 years ago
- ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"☆111Updated last month
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆236Updated last week
- Diffusion models in PyTorch☆107Updated 2 months ago
- Implementation of Infini-Transformer in Pytorch☆111Updated 7 months ago
- ☆49Updated last year
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated 3 months ago
- ☆131Updated 3 weeks ago
- Quadra: Effortless and reproducible deep learning workflows with configuration files.☆49Updated 3 months ago