πŸ“ˆ GH Trending Explorer 🎲 Surprise me

Search: tensor Β· Language: Python Β· clear

plotarmordev / drama-tensorfold

Fish Audio Drama 3 text-to-speech on the TensorFold native CUDA engine for DGX Spark (GB10): recipe, engine patches, benchmarks. Built with Fish Audio.

β–² 45/7d β˜… 45 Python 1d ago
pytorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

β˜… 105k Python πŸ”₯ today
ultralytics / yolov5

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

β˜… 58k Python today
ray-project / ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

β˜… 44k Python today
huggingface / datasets

πŸ€— The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

β˜… 22k Python 1d ago
onnx / onnx

Open standard for machine learning interoperability

β˜… 22k Python 1d ago
NVIDIA / TensorRT-LLM

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

β˜… 15k Python today
unifyai / ivy

Convert Machine Learning Code Between Frameworks

β˜… 14k Python 4d ago
apache / tvm

Open Machine Learning Compiler Framework

β˜… 14k Python today
wandb / wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

β˜… 11k Python today
ultralytics / yolov3

PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection with training, validation, inference, and multi-format export.

β˜… 11k Python 3d ago
fossasia / visdom

Tool for real-time visualization, monitoring and collaborative analysis of AI/ML experiments and live data. Supports Python, PyTorch/Torch, NumPy, TensorFlow/Keras https://visdom.dev

β˜… 10k Python today
pymc-devs / pymc

Bayesian Modeling and Probabilistic Programming in Python

β˜… 9.8k Python 1d ago
mikel-brostrom / boxmot

BoxMOT: Pluggable Python and C++ SOTA multi-object tracking modules with support for axis-aligned and oriented bounding boxes

β˜… 8.3k Python today
RapidAI / RapidOCR

πŸ“„ Awesome OCR multiple programing languages toolkits based on ONNX Runtime, OpenVINO, MNN, PaddlePaddle, TensorRT and PyTorch.

β˜… 8.2k Python today
mindee / doctr

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.

β˜… 6.4k Python 1d ago
aimhubio / aim

Aim πŸ’« β€” An easy-to-use & supercharged open-source experiment tracker.

β˜… 6.3k Python 1d ago
zenml-io / zenml

ZenML πŸ™: One AI Platform from Pipelines to Agents. https://zenml.io.

β˜… 5.6k Python today
xlite-dev / Awesome-LLM-Inference

πŸ“šA curated list of Awesome LLM/VLM Inference Papers with Codes: Flash-Attention, Paged-Attention, WINT8/4, Parallelism, etc.πŸŽ‰

β˜… 5.5k Python today
apple / coremltools

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

β˜… 5.4k Python 20d ago
NVIDIA / Model-Optimizer

A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.

β˜… 5.4k Python πŸ”₯ today
facebookincubator / AITemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

β˜… 4.7k Python 14d ago
PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

β˜… 4.6k Python 2d ago
tensorflow / datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

β˜… 4.6k Python 1d ago
SwanHubX / SwanLab

⚑️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

β˜… 4.3k Python today
google / deepvariant

DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.

β˜… 3.8k Python 20d ago
roflcoopter / viseron

Self-hosted, local only NVR and AI Computer Vision software. With features such as object detection, motion detection, face recognition and more, it gives you the power to keep an eye on your home, office or any other place you want to monitor.

β˜… 3.6k Python today
autorope / donkeycar

Open source hardware and software platform to build a small scale self driving car.

β˜… 3.5k Python 1d ago
PennyLaneAI / pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

β˜… 3.5k Python today
av / harbor

Stop configuring your AI stack. Start using it. One command brings a complete pre-wired LLM stack with hundreds of services to explore.

β˜… 3.2k Python today
pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

β˜… 3.0k Python today
tensorflow / graphics

TensorFlow Graphics: Differentiable Graphics Layers for TensorFlow

β˜… 2.8k Python 5d ago
intel / neural-compressor

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

β˜… 2.7k Python today
roboflow / inference

Turn any computer or edge device into a command center for your computer vision projects.

β˜… 2.5k Python today
tensorflow / quantum

An open-source Python framework for hybrid quantum-classical machine learning.

β˜… 2.2k Python 3d ago
tensorflow / tfx

TFX is an end-to-end platform for deploying production ML pipelines

β˜… 2.2k Python today
google / gin-config

Gin provides a lightweight configuration framework for Python

β˜… 2.2k Python 15d ago
deepmodeling / deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

β˜… 2.1k Python today
tensorflow / privacy

Library for training machine learning models with privacy for training data

β˜… 2.0k Python 8d ago
tensorflow / recommenders

TensorFlow Recommenders is a library for building recommender system models using TensorFlow.

β˜… 2.0k Python 14d ago
SemiAnalysisAI / InferenceX

Open Source AI Accelerator Research Platform Standard / εΌ€ζΊζŽ¨η†η ”η©ΆεΉ³ε°

β˜… 1.9k Python today

Load more β†’