yk / patch-torch-save
Patches the torch.save function with arbitrary code that gets executed upon torch.load.
☆71Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for patch-torch-save
- ☆48Updated 3 years ago
- Check if you have training samples in your test set☆64Updated 2 years ago
- Lite Inference Toolkit (LIT) for PyTorch☆161Updated 2 years ago
- [WIP] A 🔥 interface for running code in the cloud☆86Updated last year
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆185Updated 2 years ago
- ☆38Updated last year
- TorchFix - a linter for PyTorch-using code with autofix support☆103Updated this week
- Install PyTorch distributions with computation backend auto-detection☆223Updated last year
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆157Updated 6 months ago
- Smol but mighty language model☆62Updated last year
- Exploring finetuning public checkpoints on filter 8K sequences on Pile☆115Updated last year
- Resources from the EleutherAI Math Reading Group☆51Updated last month
- Drop in replacement for OpenAI, but with Open models.☆154Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- One stop shop for all things carp☆59Updated 2 years ago
- Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can gener…☆209Updated 2 years ago
- Train fastai models faster (and other useful tools)☆62Updated 5 months ago
- deep learning with pytorch lightning☆0Updated 3 weeks ago
- Scripts to prep PC for development use after OS installs☆37Updated last week
- HomebrewNLP in JAX flavour for maintable TPU-Training☆46Updated 10 months ago
- Python Research Framework☆107Updated 2 years ago
- ☆128Updated 2 years ago
- Functional deep learning☆106Updated last year
- Lightning HPO & Training Studio App☆18Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆58Updated 4 months ago
- mkdocs + material + cool stuff☆64Updated 2 years ago