add vmapped version of generate_trace #6

Closed gyoge0 closed 3 months ago