Toolbox of models, callbacks, and datasets for AI/ML researchers
Implementation of model parallel autoregressive transformers on GPUs
Hummingbird compiles trained ML models into tensor computation
NumPy aware dynamic Python compiler using LLVM
Easy-to-use LLM fine-tuning framework (LLaMA-2, BLOOM, Falcon
Spatial data processing for geomodeling
Library providing end-to-end GPU-accelerated recommender systems
A unified framework for scalable computing
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
The ultimate tool to automate custom telegram message forwarding
Synthetic data generators for structured and unstructured text
The data structure for multimodal data
OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
A library for deep learning end-to-end dialog systems and chatbots
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