Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
30.35k
stars
2.78k
forks
source link
Get wrong structure when using jax2tf to convert nnx.module into tflite file #24497
Open
noahzhy opened 4 days ago
Description
But the converted tflite file structure is terrible.
System info (python version, jaxlib version, accelerator, etc.)