File tree Expand file tree Collapse file tree 1 file changed +20
-0
lines changed
Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -27,3 +27,23 @@ or [TensorFlow Hub](https://tensorflow.org/hub). Refer to the
2727``` @docs
2828Reactant.Serialization.export_as_tf_saved_model
2929```
30+
31+ ## Exporting to JAX via EnzymeAD
32+
33+ !!! note "No Dependencies Required"
34+
35+ Unlike TensorFlow SavedModel export, exporting to JAX via EnzymeAD does not require any
36+ Python dependencies at build time. It generates standalone files that can be used with
37+ EnzymeAD/JAX in Python.
38+
39+ This export functionality generates:
40+ 1 . A ` .mlir ` file containing the StableHLO representation of your Julia function
41+ 2 . Example input ` .npy ` files with properly transposed arrays (column-major → row-major)
42+ 3 . A Python script that wraps the function for use with ` enzyme_ad.jax.hlo_call `
43+
44+ The generated Python script can be immediately used with JAX and EnzymeAD without any
45+ additional Julia dependencies.
46+
47+ ``` @docs
48+ Reactant.Serialization.export_to_enzymeax
49+ ```
You can’t perform that action at this time.
0 commit comments