PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
sgl-project
sglang

SGLang is a high-performance serving framework for large language models and multimodal models.

287.7M 27K 6K
flashinfer-ai
flashinfer-python

FlashInfer: Kernel Library for LLM Serving

4M 6K 948
flashinfer-ai
flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

2.7M 6K 948
sgl-project
sglang-kernel

SGLang is a high-performance serving framework for large language models and multimodal models.

264K 27K 6K
sgl-project
sgl-kernel

SGLang is a high-performance serving framework for large language models and multimodal models.

256K 27K 6K
thu-ml
sageattention

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

156K 3K 403
lucidrains
dreamer4

Implementation of Danijar's latest iteration for his Dreamer line of work

19K 185 17
lucidrains
performer-pytorch

An implementation of Performer, a linear attention-based transformer, in Pytorch

10K 1K 150
CyberZHG
keras-transformer

Transformer implemented in Keras

8K 368 94
CyberZHG
keras-position-wise-feed-forward

Feed forward layer implemented in Keras

7K 8 5
leondgarse
keras-cv-attention-models

Keras beit,caformer,CMT,CoAtNet,convnext,davit,dino,efficientdet,edgenext,efficientformer,efficientnet,eva,fasternet,fastervit,fastvit,flexivit,gcvit,ghostnet,gpvit,hornet,hiera,iformer,inceptionnext,lcnet,levit,maxvit,mobilevit,moganet,nat,nfnets,pvt,swin,tinynet,tinyvit,uniformer,volo,vanillanet,yolor,yolov7,yolov8,yolox,gpt2,llama2, alias kecam

7K 625 97
wnsgus00114-droid
lightning-core

Lightning Core: macOS-first CUDA-style runtime with Metal backend

7K 1 0
densechen
activations

AReLU: Attention-based-Rectified-Linear-Unit

5K 62 8
ai4co
rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

4K 867 146
sgl-project
sglang-kt

SGLang is a high-performance serving framework for large language models and multimodal models.

4K 27K 6K
kyegomez
qwen

My personal implementation of the model from "Qwen-VL: A Frontier Large Vision-Language Model with Versatile Abilities", they haven't released model code yet sooo...

4K 12 2
lucidrains
transfusion-pytorch

Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI

4K 1K 71
lucidrains
native-sparse-attention-pytorch

Native Sparse Attention

3K 802 52
labmlai
labml-nn

🧑‍🏫 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit), optimizers (adam, radam, adabelief), gans(dcgan, cyclegan, stylegan2), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, diffusion, etc. 🧠

3K 66K 7K
lucidrains
spear-tts-pytorch

Implementation of Spear-TTS - multi-speaker text-to-speech attention network, in Pytorch

2K 277 20
kyegomez
longnet

Implementation of plug in and play Attention from "LongNet: Scaling Transformers to 1,000,000,000 Tokens"

2K 718 61
ddbourgin
numpy-ml

Machine learning, in numpy

1K 16K 4K
lucidrains
genie2-pytorch

Genie2

1K 156 13
lucidrains
h-transformer-1d

Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning

1K 166 22
    • Data from PyPI, GitHub, ClickHouse, and BigQuery