229 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Tensors and Dynamic neural networks in Python with strong GPU acceleration | 83.9M | |
| Fast, Flexible and Portable Structured Generation | 6.4M | |
| Robust Speech Recognition via Large-Scale Weak Supervision | 2.3M | |
| Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt... | 1.9M | |
| Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt... | 1.4M | |
| WhisperX: Automatic Speech Recognition with Word-level Timestamps (& Diarizatio... | 1.1M | |
| Efficient Triton Kernels for LLM Training | 792K | |
| Code for cut cross entropy, a memory efficient implementation of linear-cross-en... | 658K | |
| cortical is the framework for building fabric architectures | 325K | |
| Mamba SSM architecture | 125K | |
| A toolset for compressing, deploying and serving LLM | 123K | |
| AutoAWQ implements the AWQ algorithm for 4-bit quantization with a 2x speedup du... | 122K | |
| Efficient Triton Kernels for LLM Training | 63K | |
| Memory-Efficient PyTorch optimizers | 62K | |
| Sparse Toolkit | 32K | |
| JAX + OpenAI Triton integration | 29K | |
| Go ahead and axolotl questions | 20K | |
| TensorRT LLM provides users with an easy-to-use Python API to define Large Langu... | 16K | |
| Toward High-Accuracy Open-Source Biomolecular Structure Prediction. | 13K | |
| Model compression toolkit engineered for enhanced usability, comprehensiveness, ... | 7K | |
| A tool to find homologous interactions and speed up AlphaFold-based structural m... | 7K | |
| Text2Text Language Modeling Toolkit | 6K | |
| VoxCPM inference engine based on Nano-vLLM | 6K | |
| A unified inference and post-training framework for accelerated video generation... | 4K | |
| Triton kernels and PyTorch ops for Block Attention Residuals (AttnRes) | 4K | |
| (EasyDel Former) is a utility library designed to simplify and enhance the devel... | 4K | |
| Web apps to inspect & engineer NN activations. | 4K | |
| Pythonic AI generation of images and videos | 4K | |
| KT-Kernel: High-performance kernel operations for KTransformers (AMX/AVX/KML opt... | 4K | |
| d9d - d[istribute]d - distributed training framework based on PyTorch that tries... | 3K | |
| An implementation of PSGD Kron second-order optimizer for PyTorch | 3K | |
| Use bacpipe to streamline the process of generating embeddings and analysing you... | 3K | |
| Vox box | 3K | |
| Fast kernel for triangle self attetion. | 3K | |
| Triton kernels and utilities from Kernelize AI | 3K | |
| Shape Analysis Explainability And Interpretability | 2K | |
| SOTA Omni-Modal Personal AI Orchestrator & Engine | 2K | |
| Structural heterogeneous cryoEM reconstruction: https://github.com/Gabriel-Ducro... | 2K | |
| A PyTorch library for multi-modal image translation with diffusion bridges, GANs... | 2K | |
| A production-grade, multi-backend compiler for Triton kernels (ONNX & TensorRT) | 2K | |
| Triton kernels for MASE | 2K | |
| Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt... | 1K | |
| A research-focused modular generative modeling library built on JAX/Flax NNX | 1K | |
| A flexible linear operator abstraction in pytorch. | 1K | |
| GPU-accelerated training for CALM (Catastrophically Abridged Language Models) | 1K | |
| Sparsity Kit for Structured Sparsity Specification | 1K | |
| Retention-based next-generation kernels for long-context models | 1K | |
| Memory-efficient RF-DETR with RFDETRSegIntermediate and xformers attention suppo... | 1K | |
| A domain-specific language (DSL) based on Triton but providing higher-level abst... | 1K | |
| FeatureFlex: efficient and lightweight library for feature selection, designed t... | 1K |