Fix: Remove missing logging import and unused sample_with_grad export 54dadc1 chrisxx commited on Nov 11