coreweave / tensorizer
Module, Model, and Tensor Serialization/Deserialization
☆221Updated last month
Alternatives and similar repositories for tensorizer:
Users that are interested in tensorizer are comparing it to the libraries listed below
- ☆190Updated last week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆136Updated last week
- ☆30Updated 2 weeks ago
- CUDA checkpoint and restore utility☆322Updated 2 months ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆313Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆262Updated 6 months ago
- ☆301Updated 7 months ago
- NVIDIA NCCL Tests for Distributed Training☆88Updated this week
- PyTorch per step fault tolerance (actively under development)☆273Updated this week
- GPUd automates monitoring, diagnostics, and issue identification for GPUs☆330Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆358Updated last week
- ☆185Updated 6 months ago
- ☆241Updated this week
- 🏋️ A unified multi-backend utility for benchmarking Transformers, Timm, PEFT, Diffusers and Sentence-Transformers with full support of O…☆291Updated 2 months ago
- OpenAI compatible API for TensorRT LLM triton backend☆204Updated 8 months ago
- Google TPU optimizations for transformers models☆107Updated 2 months ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆59Updated 2 weeks ago
- High-performance safetensors model loader☆19Updated this week
- ☆205Updated 2 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆238Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 8 months ago
- Home for OctoML PyTorch Profiler☆112Updated last year
- A library to analyze PyTorch traces.☆361Updated last week
- Perplexity GPU Kernels☆185Updated this week
- ☆117Updated last year
- NVIDIA Inference Xfer Library (NIXL)☆242Updated this week
- Benchmark suite for LLMs from Fireworks.ai☆70Updated 2 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆788Updated 7 months ago
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆184Updated this week
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆273Updated last year