Official repository for MemT-1.0
DeepSeek-V4.1-Flash-1017 EXL3 2.9bpw on 2x NVIDIA DGX Spark (GB10) with TensorFold: 2.1-2.3x the vLLM kit's decode speed.
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.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Models and examples built with TensorFlow
Deep Learning for humans
Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.
We write your reusable computer vision tools. π
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
NVR with realtime local object detection for IP cameras
100-Days-Of-ML-CodeδΈζη
π€ The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
Open standard for machine learning interoperability
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.
Convert Machine Learning Code Between Frameworks
Open Machine Learning Compiler Framework
NumPy & SciPy for GPU
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection with training, validation, inference, and multi-format export.
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
TensorFlow-based neural network library
Bayesian Modeling and Probabilistic Programming in Python
BoxMOT: Pluggable Python and C++ SOTA multi-object tracking modules with support for axis-aligned and oriented bounding boxes
π Awesome OCR multiple programing languages toolkits based on ONNX Runtime, OpenVINO, MNN, PaddlePaddle, TensorRT and PyTorch.
Flower: A Friendly Federated AI Framework
docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.
Aim π« β An easy-to-use & supercharged open-source experiment tracker.
ZenML π: One AI Platform from Pipelines to Agents. https://zenml.io.
πA curated list of Awesome LLM/VLM Inference Papers with Codes: Flash-Attention, Paged-Attention, WINT8/4, Parallelism, etc.π
Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.
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.
On-device AI across mobile, embedded and edge for PyTorch
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.
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.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Pre-trained Deep Learning models and demos (high quality and extremely fast)
A nearly-live implementation of OpenAI's Whisper.
β‘οΈ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.
Learning in infinite dimension with neural operators.
DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.
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.
Open source hardware and software platform to build a small scale self driving car.
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.
Stop configuring your AI stack. Start using it. One command brings a complete pre-wired LLM stack with hundreds of services to explore.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Lingvo
TensorFlow Graphics: Differentiable Graphics Layers for TensorFlow
SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime
Turn any computer or edge device into a command center for your computer vision projects.
An open-source framework for machine learning and other computations on decentralized data.
A library for training and deploying machine learning models on Amazon SageMaker
An open-source Python framework for hybrid quantum-classical machine learning.
TFX is an end-to-end platform for deploying production ML pipelines
Gin provides a lightweight configuration framework for Python
A deep learning package for many-body potential energy representation and molecular dynamics
RetinaFace: Deep Face Detection Library for Python
Library for training machine learning models with privacy for training data
TensorFlow Recommenders is a library for building recommender system models using TensorFlow.
Gaussian processes in TensorFlow
Open Source AI Accelerator Research Platform Standard / εΌζΊζ¨ηη η©ΆεΉ³ε°