mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/export/jax2tf.md
George Necula b58ff2ba20 [shape_poly] Add documentation for shape polymorphism 
This involved writing some new content and also moving and adapting
the documentation that existed as part of the jax2tf
README file:

https://github.com/google/jax/blob/main/jax/experimental/jax2tf/README.md#shape-polymorphic-conversion
2024-06-15 18:20:54 +03:00

153 B
Raw Blame History

(jax2tf)=

Interoperation with TensorFlow

See the JAX2TF documentation.