xla
297f5de6 - [call_jax] support returning PyTree from the JAX function (#8957)

Commit

293 days ago

[call_jax] support returning PyTree from the JAX function (#8957)

References

#8957 - [call_jax] support returning PyTree from the JAX function

Author

tengyifei

tengyifei

Parents

Loading