Nadir (pronounced nay-di-ah) is derived from the arabic word nazir, and means "the lowest point of a space". In optimisation problems, it is equivalent to the point of minimum. If you are a machine learning enthusiast, a data scientist or an AI practitioner, you know how important it is to use the best optimization algorithms to train your models. The purpose of this library is to help optimize machine learning models and enable them to reach the point of nadir in the appropriate context.
Nadir follows the principles of Simplicity, Modularity and Composabilty. Read more in the Core Philosophy section.
You can either choose to install from the PyPI index, in the following manner:
$ pip install nadir
or install from source, in the following manner:
$ pip install git+https://github.com/OptimalFoundation/nadir.git
Note: Installing from source might lead to a breaking package. It is recommended that you install from PyPI itself.
import nadir as nd
# some model setup here...
model = ...
# set up your Nadir optimiser
config = nd.SGDConfig(lr=learning_rate)
optimizer = nd.SGD(model.parameters(), config)
# Call the optimizer step
optimizer.step()
Nadir
was built to provide a sense of uniformity and integration that might be lacking in the optimisation community, based on the simple idea that optimisers are not islands. They are usually inheriting characteristics from other optimisers and they provide inspiration to other optimisers. So why not make optimisers inheritable, composible and modular objects?
The core concepts that each optimiser in Nadir
follows are:
Simplicity is of key importance. We prefer readability and simplicity over performance. Experiment, test and verify what works and what does not with Nadir. Optimise and write custom fused kernels for your favorite optimisers after, for performance.
Modularity means that the each new optimiser should minimise on the extra new logic added by adding or editing only the parts that need editing. If you want to have a different momentum in Adam, you only change the function of Momentum after inheriting Adam. No need to write the entire code from scratch.
Composibility implies that we can take things from one optimiser and add them to another without much effort. You can build a optimiser that is the mix of RAdam and NAdam with the properties of AdaBelief, if you so desire! That's what makes this library really powerful.
We would like to thank all the amazing contributors of this project who spent so much effort making this repositary awesome! :heart:
You can use the Cite this repository button provided by Github or use the following bibtex:
@software{MinhasNadir,
title = {{Nadir: A Library for Bleeding-Edge Optimizers in PyTorch}},
author = {Minhas, Bhavnick and Kalathukunnel, Apsal},
year = 2023,
month = 3,
version = {0.0.2}
}