Time series forecasting with PyTorch

A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-Modal Model. Powered by Z...

🐍 | Python library for Runpod API and serverless worker SDK.

Efficient CPU/GPU ML Runtimes for VapourSynth (with built-in support for waifu2x, DPIR, RealESRGANv2/v3, Real-CUGAN, RIF...

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras,...

DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome

深度学习框架

A Pythonic framework to simplify AI service building

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and eff...

Resource scheduling and cluster management for AI

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, DeepSeek, Mixtral, Gemma, Phi, MiniCPM, Qw...

OpenCL 1.2 implementation for Tensorflow

A library for high performance deep learning inference on NVIDIA GPUs.

Implementation of a Tensor Processing Unit for embedded systems and the IoT.

Neural network-based chess engine capable of natural language commentary