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: 10 months ago
accsgdadabeliefadaboundadamodapollodiffgradhacktoberfestlamblookaheadnovogradoptimizerpytorchsgdpshampooswatsyogi

Related projects:

Repository Description Stars
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 554
huggingface/optimum A toolkit for optimizing and accelerating the training and inference of machine learning models on various hardware platforms. 2,618
torch/optim A package that provides tools for finding and improving the parameters of a mathematical model to optimize its performance 196
ikostrikov/pytorch-trpo A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks 435
huggingface/accelerate A tool to simplify training and deployment of PyTorch models on various devices and configurations 8,056
pytorch/botorch A PyTorch-based library for Bayesian optimization, providing a modular interface for composing and optimizing probabilistic models. 3,126
clcarwin/convert_torch_to_pytorch Converts PyTorch models to their Torch equivalents. 541
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,655
pytorch-labs/torchfix A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. 111
ahmedfgad/torchga Trains PyTorch models using a genetic algorithm 96
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,561
alexis-jacq/pytorch-dppo A PyTorch implementation of Distributed Proximal Policy Optimization algorithm 180
microsoft/torchgeo Provides tools and pre-trained models for working with geospatial data in machine learning applications 3,083