docs: update examples from notebooks (#1154)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Pratyush Shukla <ps4534@nyu.edu>

Author: github-actions[bot]