pytorch-optimizer

Optimizers

A collection of optimizers for PyTorch

torch-optimizer -- collection of optimizers for Pytorch

GitHub

3k stars
33 watching
299 forks
Language: Python
last commit: 8 months ago
accsgdadabeliefadaboundadamodapollodiffgradhacktoberfestlamblookaheadnovogradoptimizerpytorchsgdpshampooswatsyogi

Related projects:

Repository Description Stars
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 544
huggingface/optimum A toolkit providing optimization tools and hardware acceleration for training and inference of machine learning models 2,572
torch/optim A package that provides tools for finding and improving the parameters of a mathematical model to optimize its performance 197
ikostrikov/pytorch-trpo A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks 433
huggingface/accelerate A tool to simplify training and deployment of PyTorch models on various devices and configurations 7,947
pytorch/botorch A PyTorch-based library for Bayesian optimization, providing a modular interface for composing and optimizing probabilistic models. 3,102
clcarwin/convert_torch_to_pytorch Converts PyTorch models to their Torch equivalents. 539
moskomule/eve.pytorch An implementation of an optimization algorithm inspired by a 2016 research paper 33
rtqichen/torchdiffeq A PyTorch implementation of differentiable ODE solvers with GPU support and efficient backpropagation 5,588
pytorch-labs/torchfix A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. 102
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm 95
jiangoforit/yellowfin_pytorch An optimizer that automatically tunes momentum and learning rate using local quadratic approximation. 287
pytorch/audio A PyTorch module providing tools and functions for audio signal processing 2,538
alexis-jacq/pytorch-dppo A PyTorch implementation of Distributed Proximal Policy Optimization algorithm 180
microsoft/torchgeo A PyTorch library providing datasets, samplers, transforms, and pre-trained models for working with geospatial data in machine learning and remote sensing applications. 2,753