neuralqx.optimizer package

A module which handles creating optax based optimisers for the minimisation process

class Optimizer(optimizer_name, learning_rate=None, scheduler_type=None, scheduler_kwargs=None, **optimizer_kwargs)

Bases: object

Subpackages

Submodules