Added AIS for RBM-like models.