Is your feature request related to a problem.
any way to get rid of the TensorFlow dependency?

Describe alternatives you've considered
See discussion in #265.

Additional context

• https://gist.github.com/wchargin/d3ab370042a3a2c66c495ef814c36354
• https://gist.github.com/gyglim/1f8dfb1b5c82627ae3efcfbbadb9f514#gistcomment-3010311