pip install torch-optimizer==0.3.0

pytorch-optimizer

Source
Among top 2% packages on PyPI.
Over 176.9K downloads in the last 90 days.

Commonly used with torch-optimizer

Based on how often these packages appear together in public requirements.txt files on GitHub.

pytorch-ranger

Ranger - a synergistic optimizer using RAdam (Rectified Adam) and LookAhead in one codebase

torch-optimizer

pytorch-optimizer

onnx-simplifier

Simplify your ONNX model

timm

(Unofficial) PyTorch Image Models

pytorch-lightning

PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.

albumentations

Fast image augmentation library and easy to use wrapper around other libraries

omegaconf

A flexible configuration library

onnxruntime

ONNX Runtime Python bindings

dacite

Simple creation of data classes from dictionaries.

coremltools

Community Tools for Core ML

tf-slim

TensorFlow-Slim: A lightweight library for defining, training and evaluating complex models in TensorFlow

tensorflow-addons

TensorFlow Addons.

onnx

Open Neural Network Exchange

tb-nightly

TensorBoard lets you watch Tensors Flow

rouge-score

Pure python implementation of ROUGE-1.5.5.

atari-py

Python bindings to Atari games

panoptes-utils

Astronomical utilities for PANOPTES

t5

Text-to-text transfer transformer

tokenizers

Fast and Customizable Tokenizers

Version usage of torch-optimizer

Proportion of downloaded versions in the last 3 months (only versions over 1%).

0.1.0

87.04%

0.3.0

4.51%

0.2.0

3.58%

0.0.1a15

3.35%