Top 9 Python Vulkan Projects

tvm

15 11,156 9.9 Python

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Project mention: Making AMD GPUs competitive for LLM inference | news.ycombinator.com | 2023-08-09

Yes, this is coming! Myself and others at OctoML and in the TVM community are actively working on multi-gpu support in the compiler and runtime. Here are some of the merged and active PRs on the multi-GPU (multi-device) roadmap:
Support in TVM’s graph IR (Relax) - https://github.com/apache/tvm/pull/15447

video2x

108 8,630 5.5 Python

A lossless video/GIF/image upscaler achieved with waifu2x, Anime4K, SRMD and RealSR. Started in Hack the Valley II, 2018.

Project mention: Upscayl – Free and Open Source AI Image Upscaler | news.ycombinator.com | 2024-03-31

InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
wgpu-py

5 370 8.6 Python

Next generation GPU API for Python

Project mention: Pygfx/wgpu-py: Next generation GPU API for Python | news.ycombinator.com | 2024-02-20

pygfx

3 352 8.8 Python

A python render engine running on wgpu.

Project mention: Emerging Rust GUI libraries in a WASM world | news.ycombinator.com | 2023-04-26

https://github.com/kushalkolar/fastplotlib
Alternatively, try pygfx for ThreeJS graphics in Python leveraging wgpu. It works great in Notebooks through notebook-rfb. https://github.com/pygfx/pygfx
If you're adventurous, figure out how to make pygfx work with webgpu via wasm

fastplotlib

2 324 8.9 Python

Next-gen fast plotting library running on WGPU using the pygfx rendering engine

Project mention: Emerging Rust GUI libraries in a WASM world | news.ycombinator.com | 2023-04-26

https://github.com/kushalkolar/fastplotlib
Alternatively, try pygfx for ThreeJS graphics in Python leveraging wgpu. It works great in Notebooks through notebook-rfb. https://github.com/pygfx/pygfx
If you're adventurous, figure out how to make pygfx work with webgpu via wasm

GPU-Viewer

4 318 8.5 Python

A front-end to glxinfo, vulkaninfo, clinfo and es2_info - Linux
StableDiffusionUI

9 125 8.6 Python

Stable Diffusion UI: Diffusers (CUDA/ONNX)

Project mention: I have created Olive model. But what is the next step? Who knows? | /r/StableDiffusion | 2023-05-24

There’s this UI that runs on ONNX models. Not sure if it will work though.

WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
lava

1 15 0.0 Python

A Highlevel Python Wrapper for Vulkan's Compute API (by osanj)
SPIRVSmith

3 14 0.0 Python

A differential testing tool targeting SPIRV based on structured fuzzing techniques

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).