Amir Gholami's repositories
adahessian
ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
ai_and_memory_wall
AI and Memory Wall
ada-hessian
Easy-to-use AdaHessian optimizer (PyTorch)
pytorch-optimizer
torch-optimizer -- collection of optimizers for Pytorch
convnet-burden
Memory consumption and FLOP count estimates for convnets
Language:MATLABMIT000
DCP
Code for “Discrimination-aware-Channel-Pruning-for-Deep-Neural-Networks”
Language:PythonBSD-3-Clause000
Language:Python000
Language:PythonApache-2.0000
torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.