I cannot generate the API reference for the NLinear class because its definition, located in chronax/models/nlinear/nlinear_model.py, was not provided in the input.
I can only confirm the module summary:
chronax.models.nlinear
NLinear forecasting model (univariate, JAX/Flax-NNX port of neuralforecast.NLinear).