Lists (1)
Stars
OME is a Kubernetes operator for enterprise-grade management and serving of Large Language Models (LLMs)
Repository hosting code for "Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendations" (https://arxiv.org/abs/2402.17152).
slime is a LLM post-training framework aiming at scaling RL.
An Efficient and User-Friendly Scaling Library for Reinforcement Learning with Large Language Models
Official repository for the Boltz biomolecular interaction models
LightLLM is a Python-based LLM (Large Language Model) inference and serving framework, notable for its lightweight design, easy scalability, and high-speed performance.
Flax is a neural network library for JAX that is designed for flexibility.
SpatialLM: Training Large Language Models for Structured Indoor Modeling
VILA is a family of state-of-the-art vision language models (VLMs) for diverse multimodal AI tasks across the edge, data center, and cloud.
Scenic: A Jax Library for Computer Vision Research and Beyond
Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Jeo: Jax model training lib for Earth Observation
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Supercharge Your LLM with the Fastest KV Cache Layer
DORA (Dataflow-Oriented Robotic Architecture) is middleware designed to streamline and simplify the creation of AI-based robotic applications. It offers low latency, composable, and distributed dat…
Scalable toolkit for efficient model reinforcement
My learning notes/codes for ML SYS.
Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels
Distributed Compiler based on Triton for Parallel Systems
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
🐍 Geometric Computer Vision Library for Spatial AI
Free, open-source VHS effect. Standalone application + plugin (After Effects, Premiere, and OpenFX).
Multi-Joint dynamics with Contact. A general purpose physics simulator.
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.