Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Pytorch domain library for recommendation systems
Multilingual Automatic Speech Recognition with word-level timestamps
Superfast AI decision making and processing of multi-modal data
An open source implementation of CLIP
Optax is a gradient processing and optimization library for JAX
Sacred is a tool to help you configure, andorganize IDSIA experiments
An easy-to-use & supercharged open-source experiment tracker
A very simple framework for state-of-the-art NLP
Investment Research for Everyone, Everywhere
A simple but complete full-attention transformer
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
PyTorch extensions for fast R&D prototyping and Kaggle farming
Data science on data without acquiring a copy
Time series forecasting with PyTorch
MMEditing is a low-level vision toolbox based on PyTorch
Unified Model Serving Framework
Modular quant framework
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Scalable and user friendly neural forecasting algorithms.
Your open-source LLM evaluation toolkit
Training PyTorch models with differential privacy
High-level training, data augmentation, and utilities for Pytorch
High quality, fast, modular reference implementation of SSD in PyTorch
A lightweight vision library for performing large object detection