tshu-w / lightning-templateLinks
PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.
☆23Updated last month
Alternatives and similar repositories for lightning-template
Users that are interested in lightning-template are comparing it to the libraries listed below
Sorting:
- ☆46Updated 10 months ago
- Quickest way to share everything about your research within a single app☆37Updated 8 months ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 7 months ago
- An annotated implementation of the Hyena Hierarchy paper☆33Updated 2 years ago
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆53Updated last year
- ☆61Updated last year
- TorchDR - PyTorch Dimensionality Reduction☆130Updated this week
- ☆8Updated last year
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆51Updated 3 months ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- Utilities for PyTorch distributed☆24Updated 4 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆63Updated this week
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆97Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated 2 years ago
- Implementation of Infini-Transformer in Pytorch☆111Updated 5 months ago
- ☆74Updated 2 years ago
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated 11 months ago
- ☆53Updated 8 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆48Updated 8 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆124Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆122Updated 11 months ago
- Quickest way to share everything about your research within a single app☆16Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated last year
- TabDPT: Scaling Tabular Foundation Models☆30Updated 2 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆66Updated 9 months ago
- Diffusion models in PyTorch☆104Updated last week
- Running Jax in PyTorch Lightning☆102Updated 6 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Research on Tabular Foundation Models☆52Updated 6 months ago