docs(example): add missing imports, device parameter, and correct API usage#2

mosure merged 3 commits intomainmosure/burn_depth:mainfrom

copilot/fix-readme-rust-examplemosure/burn_depth:copilot/fix-readme-rust-exampleCopy head branch name to clipboard