Pretrained (Language) Models for Probabilistic Time Series Forecasting
Explainability and Interpretability to Develop Reliable ML models
We write your reusable computer vision tools
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
A refreshing functional take on deep learning
Library for OCR-related tasks powered by Deep Learning
DeepVariant is an analysis pipeline that uses a deep neural networks
Geometric deep learning extension library for PyTorch
Deep learning library
Fast image augmentation library and an easy-to-use wrapper
Fast and memory-efficient exact attention
An on-premises, OCR-free unstructured data extraction
A modular, primitive-first, python-first PyTorch library
A Python package for segmenting geospatial data with the SAM
Build MLOps Pipelines in Minutes
Hub of ready-to-use datasets for ML models
AI discovers 520000 stable inorganic crystal structures for research
A Python package for extending the official PyTorch
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Running large language models on a single GPU
A library for accelerating Transformer models on NVIDIA GPUs
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Clean, Robust, and Unified PyTorch implementation