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

A high-throughput and memory-efficient inference and serving engine for LLMs

9.4M 79K 16K
skypilot-org
skypilot

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

1.8M 10K 1K
skypilot-org
skypilot-nightly

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

482K 10K 1K
google
torchax

torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.

291K 218 29
vllm-project
vllm-tpu

A high-throughput and memory-efficient inference and serving engine for LLMs

143K 79K 16K
skypilot-org
trainy-skypilot-nightly

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

17K 10K 1K
tensorflow
tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

8K 17K 4K
aphrodite-engine
aphrodite-engine

Large-scale LLM inference engine

7K 2K 194
Kohulan
decimer

DECIMER Image Transformer is a deep-learning-based tool designed for automated recognition of chemical structure images. Leveraging transformer architectures, the model converts chemical images into SMILES strings, enabling the digitization of chemical data from scanned documents, literature, and patents.

7K 361 76
google
xpk

xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.

7K 180 88
google
google-jetstream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

4K 432 64
hhk7734
yolov4

YOLOv4 Implemented in Tensorflow 2.

2K 135 74
clashluke
revlib

Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload

1K 132 6
clashluke
tpucare

Automatically take good care of your preemptible TPUs

970 37 1
andreped
gradient-accumulator

Package for gradient accumulation in TensorFlow

955 53 11
tensorflow
adanet

Fast and flexible AutoML with learning guarantees.

794 3K 525
skypilot-org
nextai-prism

Prism: An intercloud broker for the clouds

600 10K 1K
skypilot-org
ac-scheduler

SkyPilot: Run AI on Any Infra — Unified, Faster, Cheaper. (Fork by Jason-AC)

441 10K 1K
vllm-project
vllm-hust

A high-throughput and memory-efficient inference and serving engine for LLMs

437 79K 16K
birkhoffg
cfnet

Recourse Explanation Library in JAX

437 2 1
sascha-kirch
deepsaki

DeepSaki is an add-on to TensorFlow. It provides a variaty of custom classes ranging from activation functions to entire models, helper functions to facilitate connectiong to your, compute HW and many more!

434 10 0
vllm-project
wxy-test

A high-throughput and memory-efficient inference and serving engine for LLMs

375 2K 1K
vllm-project
vllm-xft

A high-throughput and memory-efficient inference and serving engine for LLMs

345 79K 16K
vllm-project
ai-dynamo-vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

344 79K 16K
    • Data from PyPI, GitHub, ClickHouse, and BigQuery