ribs.discount_models

Discount models and related utilities.

MLP(layer_specs, activation)

PyTorch multi-layer perceptron model.

DiscountModelManager(model, optimizer, ...)

Wraps a PyTorch model so it can be used as a discount model.