Implementation of model parallel autoregressive transformers on GPUs
Hummingbird compiles trained ML models into tensor computation
Easy-to-use LLM fine-tuning framework (LLaMA-2, BLOOM, Falcon
Spatial data processing for geomodeling
A unified framework for scalable computing
Toolbox of models, callbacks, and datasets for AI/ML researchers
Build multi-modal Agents with memory, knowledge, tools and reasoning
The data structure for multimodal data
Library providing end-to-end GPU-accelerated recommender systems
Optax is a gradient processing and optimization library for JAX
Fast Forward Computer Vision (and other ML workloads!)
A repo for distributed training of language models with Reinforcement
State-of-the-art Parameter-Efficient Fine-Tuning
Synthetic data generators for structured and unstructured text
OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
A library for deep learning end-to-end dialog systems and chatbots
NumPy aware dynamic Python compiler using LLVM
The ultimate tool to automate custom telegram message forwarding
Experiment tracking, ML developer tools
Tools to help users inter-operate among deep learning frameworks
Library of deep learning models and datasets
Easily train your own text-generating neural network
Python Robotic Process Automation Library
Aims to enable researcher to tap in to mobile computing capability