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-project d9d
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