Slicer / light-the-torch
Install PyTorch distributions with computation backend auto-detection
☆251Updated 2 weeks ago
Alternatives and similar repositories for light-the-torch
Users that are interested in light-the-torch are comparing it to the libraries listed below
Sorting:
- NumPy arrays, ready for human consumption☆69Updated 9 months ago
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- TorchFix - a linter for PyTorch-using code with autofix support☆141Updated 3 months ago
- Create powerful Hydra applications without the yaml files and boilerplate code.☆377Updated last week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆369Updated last month
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆56Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆228Updated 3 months ago
- Memory mapped numpy arrays of varying shapes☆298Updated 10 months ago
- aim-mlflow integration☆211Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆287Updated 8 months ago
- deep learning with pytorch lightning☆1Updated 6 months ago
- TensorDict is a pytorch dedicated tensor container.☆925Updated this week
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆116Updated 2 months ago
- ☆347Updated last week
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 5 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 3 years ago
- Scalable and Performant Data Loading☆258Updated this week
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆418Updated last year
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆436Updated 8 months ago
- For optimization algorithm research and development.☆513Updated this week
- Highly commented implementations of Transformers in PyTorch☆136Updated last year
- spock is a framework that helps manage complex parameter configurations during research and development of Python applications☆133Updated last year
- Clean up the public namespace of your package!☆55Updated 3 weeks ago
- Display progress as a pretty table in the command line.☆135Updated 2 months ago
- Tensors, for human consumption☆1,251Updated 5 months ago
- Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …☆230Updated last year
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆63Updated last month
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated last week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆472Updated this week