Metadata-Version: 2.4
Name: mpadam
Version: 1.0.7
Summary: Mixed Precision Adam optimizer
Author-email: Alexander Lyulkov <alex.lyulkov@hotmail.com>
License-Expression: Apache-2.0
Project-URL: Homepage, https://github.com/alexlyulkov/MPAdam
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.9
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch
Requires-Dist: triton
Dynamic: license-file

# MPAdam - Mixed Precision Adam optimizer

MPAdam is a mixed precision Adam optimizer that allows to train up to 3x larger models on the same GPU. It supports Adam and AdamW algorithms and float16 and bfloat16 training types.

[Documentation](https://github.com/alexlyulkov/MPAdam)
