Lightning-AI / lightning-Habana
Lightning support for Intel Habana accelerators.
☆25Updated 3 weeks ago
Alternatives and similar repositories for lightning-Habana:
Users that are interested in lightning-Habana are comparing it to the libraries listed below
- 삼각형의 실전! Triton☆15Updated 11 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"☆58Updated 3 months ago
- ☆41Updated last year
- ☆85Updated 8 months ago
- some common Huggingface transformers in maximal update parametrization (µP)☆79Updated 2 years ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆32Updated 9 months ago
- Implementation of Infini-Transformer in Pytorch☆109Updated 3 weeks ago
- ☆37Updated 9 months ago
- A library for unit scaling in PyTorch☆122Updated 2 months ago
- ☆47Updated 5 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆122Updated 3 weeks ago
- Efficient CUDA kernels for training convolutional neural networks with PyTorch.☆38Updated this week
- Randomized Positional Encodings Boost Length Generalization of Transformers☆79Updated 10 months ago
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated last year
- Triton Implementation of HyperAttention Algorithm☆46Updated last year
- Train, tune, and infer Bamba model☆80Updated 2 weeks ago
- Code for studying the super weight in LLM☆72Updated last month
- Implementation of Hyena Hierarchy in JAX☆10Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- ☆75Updated 6 months ago
- Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.☆155Updated 9 months ago
- ☆57Updated this week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆119Updated 6 months ago
- Torch Distributed Experimental☆115Updated 5 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Code for NeurIPS LLM Efficiency Challenge☆54Updated 9 months ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆71Updated last month
- Utilities for Training Very Large Models☆57Updated 4 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆153Updated last month