FastAPI framework, high performance, easy to learn, fast to code, ready for production

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Python packaging and dependency management made easy

Practical and document place for DevOps toolchain

Fork of the Triton language and compiler for Windows support

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

The Simple Chemistry Balancer is an open-source Python project designed to balance chemical equations. It provides users with the ability to input unbalanced chemical formulas and receive balanced …

Code for paper "DeepEMD: Few-Shot Image Classification with Differentiable Earth Mover's Distance and Structured Classifiers", CVPR2020

Retrieval Augmented Generation (RAG) chatbot powered by Weaviate

Integrate the DeepSeek API into popular softwares

A Pytorch image classification training system, easy to use (relatively 😅). Supports Tensorboard logging and techniques like "Gradient centralization", "Adaptive gradient clipping" and etc...

optimizer & lr scheduler & loss function collections in PyTorch

The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)

A FAT filesystem with SPI driver for SD card on Raspberry Pi Pico

A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model

MiniTorch is an educational library that recreates core PyTorch features using only Python and NumPy. For more information, please read the README.md file.

Uniform Manifold Approximation and Projection

✨ Magical shell history

A todo list app with HTML, CSS & JavaScript

Official repository of paper titled "CAS-ViT: Convolutional Additive Self-attention Vision Transformers for Efficient Mobile Applications"

Model interpretability and understanding for PyTorch

An open-source, cross-platform terminal for seamless workflows

functorch is JAX-like composable function transforms for PyTorch.

PyTurboJPEG is a highly optimized Python wrapper of libjpeg-turbo (TurboJPEG API) which supports x86 and ARM architecture.

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Dataframes powered by a multithreaded, vectorized query engine, written in Rust

EfficientNetV2 pytorch (pytorch lightning) implementation with pretrained model

SVG JSON Visualizer

Pytorch to Keras/Tensorflow/TFLite conversion made intuitive