pmeier / light-the-torch
Install PyTorch distributions with computation backend auto-detection
☆216Updated last year
Related projects: ⓘ
- NumPy arrays, ready for human consumption☆63Updated 2 months ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆316Updated last month
- Memory mapped numpy arrays of varying shapes☆277Updated 3 months ago
- ☆97Updated this week
- Multidimensional indexing for tensors☆108Updated last year
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆49Updated this week
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆74Updated this week
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆102Updated last week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆318Updated this week
- Highly commented implementations of Transformers in PyTorch☆126Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆469Updated 2 years ago
- Lightning HPO & Training Studio App☆17Updated last year
- ☆137Updated last month
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆173Updated 2 years ago
- Create powerful Hydra applications without the yaml files and boilerplate code.☆317Updated this week
- ☆288Updated 2 months ago
- Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …☆189Updated 9 months ago
- ☆192Updated this week
- For optimization algorithm research and development.☆240Updated last week
- Clean up the public namespace of your package!☆51Updated 5 months ago
- TensorDict is a pytorch dedicated tensor container.☆808Updated this week
- Patches the torch.save function with arbitrary code that gets executed upon torch.load.☆70Updated 2 years ago
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆174Updated 4 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆73Updated last week
- 📈 Log and track ML metrics, parameters, models with Git and/or DVC☆162Updated this week
- Display progress as a pretty table in the command line.☆90Updated last week
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- Run PyTorch in JAX. 🤝☆187Updated 10 months ago
- git extension for {collaborative, communal, continual} model development☆202Updated 3 months ago