Graph Neural Network Library for PyTorch
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Qlib is an AI-oriented quantitative investment platform
Fast image augmentation library and an easy-to-use wrapper
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
AI discovers 520000 stable inorganic crystal structures for research
The most intuitive, flexible, way for researchers to build models
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A library for audio and music analysis, feature extraction
A library for deep learning end-to-end dialog systems and chatbots
Superfast AI decision making and processing of multi-modal data
Uniform Manifold Approximation and Projection
PyTorch version of Stable Baselines
Beta Machine Learning Toolkit
OpenMMLab Model Deployment Framework
A toolkit to optimize ML models for deployment for Keras & TensorFlow
Usable Implementation of "Bootstrap Your Own Latent" self-supervised