#pytorch

No repository description published.

State-of-the-art 2D and 3D Face Analysis Project

A convenient CLI to streamline LiteRT related development workflows, including converting, quantizing, compiling, managing, running, benchmarking and visualizing LiteRT (TFLite) models on various hardwares (CPU / GPU / NPU) across platforms (desktop, mobile or cloud).

Training-free KV cache compression via E8 lattice VQ. 2-bit KV that preserves retrieval (30/30 NIAH vs TurboQuant 0/30). Calibration-free, 9 architectures validated.

An open-source deep learning framework for data mining of protein-protein interfaces or single-residue variants.

🛸 Use pretrained transformers like BERT, XLNet and GPT-2 in spaCy

A configurable, tunable, and reproducible library for CTR prediction https://fuxictr.github.io

High-fidelity performance metrics for generative models in PyTorch

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

Model-agnostic multi-modality generation server with plugin discovery: OpenAI-compatible TTS, embeddings, and image generation. Drop a Python file in ~/.muse/models/ to add a backend.

depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.

Fused CUDA kernels for LLM inference - RMSNorm / RoPE / SwiGLU

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)

High-performance RDMA for Python and PyTorch (ibverbs, efa, NVMe-oF)

A configurable, tunable, and reproducible library for CTR prediction https://fuxictr.github.io

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

CnSTD: 基于 PyTorch/MXNet 的 中文/英文 场景文字检测（Scene Text Detection）、数学公式检测（Mathematical Formula Detection, MFD）、篇章分析（Layout Analysis）的Python3 包

PyTorch Extension Library of Optimized Scatter Operations

Audio processing by using pytorch 1D convolution network

A 3D Slicer integration to Meta's SAM.