A library for accelerating Transformer models on NVIDIA GPUs
A lightweight 3D Morphable Face Model library in modern C++
Machine Learning Pipelines for Kubeflow
Training PyTorch models with differential privacy
The open-source tool for building high-quality datasets
DeepVariant is an analysis pipeline that uses a deep neural networks
Graph Neural Network Library for PyTorch
Fast image augmentation library and an easy-to-use wrapper
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Qlib is an AI-oriented quantitative investment platform
AI discovers 520000 stable inorganic crystal structures for research
The most intuitive, flexible, way for researchers to build models
Machine Learning Systems: Design and Implementation
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
Hyperparameter Optimization for TensorFlow, Keras and PyTorch
A simple but complete full-attention transformer
TensorFlow-based neural network library
A library for deep learning end-to-end dialog systems and chatbots
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A fast image processing library with low memory needs
Superfast AI decision making and processing of multi-modal data