juntang-zhuang

ShelfNet

implementation for paper "ShelfNet for fast semantic segmentation"

LadderNet

torch_ACA

repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE

ShelfNet-lw-cityscapes

ShelfNet-lightweight for paper [ShelfNet for fast semantic segmentation]

explain_invertible

repo for "Decision explanation and feature importance for invertible networks"

fairseq-adabelief

rainbow-adabelief

SAC-Adabelief

transformer-adabelief

EAdam-optimizer

Some improvements on Adam

adabelief-toy-examples-pytorch

numerical experiments in https://arxiv.org/pdf/2010.07468.pdf

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

federated

A collection of Google research projects related to Federated Learning and Federated Analytics.

machine_learning_refined

Notes, examples, and Python demos for the textbook "Machine Learning Refined" (published by Cambridge University Press).

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

adahessian

ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning

interviews_prep

Coding Interview Preparation

Linear-Algebra-Basics

Linear Algebra Fundamentals for Machine Learning

machine-learning-basics

Basics for Machine Learning

Probability-and-Information-Theory

Probability and Information Theory for Machine Learning

pytorch_geometric

Geometric Deep Learning Extension Library for PyTorch

RAdam

On the Variance of the Adaptive Learning Rate and Beyond

RMSprop

The official implementation of the paper "RMSprop can converge with proper hyper-parameter"

shallow-linear-net