autobound icon indicating copy to clipboard operation
autobound copied to clipboard

Published 20 hours ago • verified publisher icon google

Torch and TensorFlow support

Open legel opened this issue 1 year ago • 1 comments

Thanks to the authors and sponsors, this looks like it could be very useful for AI/ML.

Is there support for Torch and TensorFlow on the way?

legel avatar Apr 14 '23 16:04 legel

No immediate plans, but maybe after support for JAX is more mature (i.e., all the JAX primitives are supported) I'd consider adding this.

mstreeter avatar Apr 14 '23 23:04 mstreeter