Automated Machine Learning on Kubernetes
TensorLy: Tensor Learning in Python.
Add bisenetv2. My implementation of BiSeNet
Hyperparameter Experiments with TensorFlow and Keras
High-quality implementations of standard and SOTA methods on a variety of tasks.
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.
Qualcomm® AI Hub Models is our collection of state-of-the-art machine learning models optimized for performance (latency, memory etc.) and ready to deploy on Qualcomm® devices.
Neural Network Compression Framework for enhanced OpenVINO™ inference
A scalable, agentic-first, and HuggingFace-native RL framework for research (9k lines).
XuanCe: A Comprehensive and Unified Deep Reinforcement Learning Library
Fast & Simple Resource-Constrained Learning of Deep Network Structure
TensorDict is a pytorch dedicated tensor container.
:zap: TensorFlowASR: Almost State-of-the-art Automatic Speech Recognition in Tensorflow 2. Supported languages that can use characters or subwords
Training neural models with structured signals.
Pretrained model hub for Keras 3.
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
AI agent identity, signed credentials, delegated authority and portable governance evidence. Python core and offline verifier SDKs.
Python Computer Vision & Video Analytics Framework With Batteries Included
Code for paper "Computation Offloading Optimization for UAV-assisted Mobile Edge Computing: A Deep Deterministic Policy Gradient Approach"
Predicting stock prices using a TensorFlow LSTM (long short-term memory) neural network for times series forecasting
Python Causal Impact Implementation Based on Google's R Package. Built using TensorFlow Probability.
A python library for quantum information and many-body calculations including tensor networks.
Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
Build high-performance AI models with modular building blocks
High-performance, light-weight C++ LLM and VLM Inference Software for Physical AI
AI Tensor Engine for ROCm
Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.
A robust, and flexible open source User & Entity Behavior Analytics (UEBA) framework used for Security Analytics. Developed with luv by Data Scientists & Security Analysts from the Cyber Security Industry. [BETA]
Tensor Network Python (TeNPy)
GLM-5.3-Flash EXL3 on 2x DGX Spark with TensorFold
Hand-written NVFP4 W4A16 CUDA kernels for Volta
Local JavaScript + FastAPI video restoration studio for SeedVR2 with TensorRT acceleration, previews, synchronized comparisons, resumable renders, and post-processing.
Imagination Studio by keys+Wasserman powered by TensorFold: an AI film studio for Mac and NVIDIA DGX Spark. Plan a film like a crew (Slate workflow by Sam Wasserman), render Qwen-Image-2.1 / FLUX3-K pictures and MiniMax H3 / FastH3 video with sound on TensorFold.
GPU 性能与 AI Infra 学习项目:CUDA/Triton 算子、NCU/NSYS、vLLM/SGLang/TRT-LLM/ms-swift、PyTorch/DeepSpeed/ms-swift 训练、并行架构
GLM-5.3-Flash (abliterated EXL3) on 2x NVIDIA DGX Spark with the TensorFold engine: 1.8x faster decode than vLLM, 4x256k concurrent threads, byte-exact speculative decoding. Work in progress.
WeeLLM runs large diffusion models with as little as 4 GB of VRAM, without any quantization. It dynamically determines how many layers can fit within the available VRAM and streams the text encoder and transformer layers to the GPU layer by layer, enabling inference on hardware with limited VRAM. It supports both safetensors and GGUF models.
GLM-5.3-Flash EXL3 (320B MoE) on 2x NVIDIA DGX Spark — production serving kit, 1M context, 97%+ multi-session prefix caching, DFlash2 spec decode
A custom "Hey Donut" wake word for Home Assistant — 62 KB, runs on-device, zero false accepts per hour. Includes the full training pipeline so you can teach your assistant any name. CC0.
Chatpin puts your AI’s chat template under inspection and lock. Compare it with the publisher’s original, flag suspicious conditional logic, and pin the version you trust so unexpected changes fail CI before deployment. Built for GGUF and safetensors workflows. Your model has a chat template. Give it a tamper alarm.
TensorRT-first robot foundation model inference for edge GPUs
Static reproducibility checks, experiment replay, and verification for ML pipelines.
Switchless RoCE collectives, NCCL profiles and fabric tooling for DGX Spark inference recipes
BertholomusAI recipe: DeepSeek-V4.1-Flash on 2x NVIDIA DGX Spark (GB10), our own TensorFold TP2 engine. Weights: Mia-AiLab EXL3 2.9bpw (MIT).
A local model discovery and download manager for Hugging Face, CivitAI, CivitAI Red, ModelScope, TensorHub Art, and SeaArt.
High-throughput batched ASR for NVIDIA GPUs. Run Zipformer and NVIDIA NeMo Parakeet models through one Python API with TensorRT, custom CUDA plugins, GPU beam search, and word timestamps - up to 25,000 RTFx on B300.
Enables the ComfyUI MiniMax H3 reference-to-video workflow to use a stronger FL2VA model and validated Ref2VA tensor-overlay modes to unlock MiniMax H3 capabilities.
MiniMax H3 (video + audio) on TensorFold 0.6.5 with an H3 family, Mac Studio M5 Ultra: 5 s 864x480 clip in ~50 s, 40 s at 768x448 — the fastest H3 we have measured on Apple Silicon; int8 tensor-unit kernels + Turbo adapter; one-shot + GHCR carrier
Serving large LLMs on a single DGX Spark (121 GiB): Qwen3.8-Flash-Next on TensorFold/SparkFold EXL3 and vLLM NVFP4, Nemotron 3.5 Super VL on vLLM; what ran, what it measured, what did not work
GLM-5.3 EXL3 on DGX Sparks with TensorFold
Native ComfyUI nodes for SenseNova U1.5 image generation and multi-reference image editing.