DeepMind's software stack for physics-based simulation
Deep learning library
Open source framework for deep learning satellite and aerial imagery
Chemcrow
A modular, primitive-first, python-first PyTorch library
A Python package for segmenting geospatial data with the SAM
An open source implementation of CLIP
Time series forecasting with PyTorch
A Pythonic framework to simplify AI service building
GPU environment management and cluster orchestration
Implementation of Imagen, Google's Text-to-Image Neural Network
AutoGluon: AutoML for Image, Text, and Tabular Data
Create HTML profiling reports from pandas DataFrame objects
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
A library for accelerating Transformer models on NVIDIA GPUs
Trainable models and NN optimization tools
Medical imaging toolkit for deep learning
High-level training, data augmentation, and utilities for Pytorch
Optax is a gradient processing and optimization library for JAX
Helps scientists define testable, modular, self-documenting dataflow
Fast, flexible and easy to use probabilistic modelling in Python
An easy-to-use & supercharged open-source experiment tracker
A very simple framework for state-of-the-art NLP
Python library for defining and optimizing mathematical expressions