Google Research

Instruct-tune LLaMA on consumer hardware

Official inference library for Mistral models

A scikit-learn compatible neural network library that wraps PyTorch

Debugging, monitoring and visualization for Python Machine Learning and Data Science

Bayesian optimization in PyTorch

DeepFashion2 Dataset https://arxiv.org/pdf/1901.07973.pdf

Fast and Easy Infinite Neural Networks in Python

🎨 Semantic segmentation models, datasets and losses implemented in PyTorch.

Anatomy of Matplotlib -- tutorial developed for the SciPy conference

PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks

A simple probabilistic programming language.

CLIP-like model evaluation

Finetuning InstructLLaMA with portuguese data

Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"

Code repository for the paper - "Matryoshka Representation Learning"

A library for ML benchmarking. It's powerful.

An implementation of our CVPR 2018 work 'Domain Adaptive Faster R-CNN for Object Detection in the Wild'

A pytorch implementation of MINE(Mutual Information Neural Estimation)

Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'

Source code for ClimateLearn

A collection of deep learning frameworks ported to Keras for face analysis.

Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learning, ICLR 2020

[TMLR23] Official implementation of UnIVAL: Unified Model for Image, Video, Audio and Language Tasks.

This is the repo for the paper Shepherd -- A Critic for Language Model Generation

LLM-Merging: Building LLMs Efficiently through Merging

An awesome semantic segmentation model that runs in real time

Official PyTorch implementation of "Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup" (ICML'20)