Rename export_to_enzymeax to export_to_enzymejax

Co-authored-by: wsmoses <[email protected]>

Author: Copilot

docs/src/api/serialization.md

@@ -45,5 +45,5 @@ The generated Python script can be immediately used with JAX and EnzymeAD withou
 additional Julia dependencies.
 
 ```@docs
-Reactant.Serialization.export_to_enzymeax
+Reactant.Serialization.export_to_enzymejax
 ```

src/serialization/EnzymeJAX.jl

@@ -3,7 +3,7 @@ module EnzymeJAX
 using ..Reactant: Reactant, Compiler, MLIR, Serialization
 
 """
-    export_to_enzymeax(
+    export_to_enzymejax(
         f,
         args...;
         output_dir::String=".",
@@ -50,7 +50,7 @@ x = Reactant.to_rarray(Float32[1, 2, 3])
 y = (; x=Reactant.to_rarray(Float32[4, 5, 6]), y=Reactant.to_rarray(Float32[7, 8, 9]))
 
 # Export to EnzymeJAX
-python_file_path = Reactant.Serialization.export_to_enzymeax(my_function, x, y)
+python_file_path = Reactant.Serialization.export_to_enzymejax(my_function, x, y)
 ```
 
 Then in Python:
@@ -62,7 +62,7 @@ import jax
 result = jax.jit(run_my_function)(*inputs)
 ```
 """
-function export_to_enzymeax(
+function export_to_enzymejax(
     f, args...; output_dir::Union{String,Nothing}=nothing, function_name::String=string(f)
 )
     if output_dir === nothing
@@ -208,7 +208,7 @@ function _generate_python_script(
     \"\"\"
     Auto-generated Python script for calling exported Julia/Reactant function via EnzymeJAX.
 
-    This script was generated by Reactant.Serialization.export_to_enzymeax().
+    This script was generated by Reactant.Serialization.export_to_enzymejax().
     \"\"\"
 
     from enzyme_ad.jax import hlo_call

src/serialization/Serialization.jl

@@ -128,7 +128,7 @@ function export_as_tf_saved_model(
 end
 
 """
-    export_to_enzymeax(
+    export_to_enzymejax(
         f,
         args...;
         output_dir::String=".",
@@ -137,10 +137,10 @@ end
 
 Export a Julia function to EnzymeJAX format for use in Python/JAX.
 
-See [`EnzymeJAX.export_to_enzymeax`](@ref) for details.
+See [`EnzymeJAX.export_to_enzymejax`](@ref) for details.
 """
-function export_to_enzymeax(f, args...; kwargs...)
-    return EnzymeJAX.export_to_enzymeax(f, args...; kwargs...)
+function export_to_enzymejax(f, args...; kwargs...)
+    return EnzymeJAX.export_to_enzymejax(f, args...; kwargs...)
 end
 
 end

test/export_enzymeax.jl

@@ -16,7 +16,7 @@ using Test
         y = Reactant.to_rarray(Float32[4, 5, 6])
 
         # Export to EnzymeJAX
-        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymeax(
+        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymejax(
             simple_add, x, y;
             output_dir=tmpdir,
             function_name="simple_add"
@@ -46,7 +46,7 @@ using Test
             @test filesize(input_path) > 0
         end
 
-        println("✓ All export_to_enzymeax tests passed!")
+        println("✓ All export_to_enzymejax tests passed!")
         println("  - MLIR file created: $(mlir_path)")
         println("  - Python file created: $(python_path)")
         println("  - Input files created: $(length(input_paths))")

test/export_enzymeax_comprehensive.jl

@@ -15,7 +15,7 @@ using Test
         y = Reactant.to_rarray(Float32[7 8; 9 10; 11 12])  # 3x2 matrix
 
         # Export to EnzymeJAX
-        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymeax(
+        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymejax(
             matrix_multiply, x, y;
             output_dir=tmpdir,
             function_name="matrix_multiply"
@@ -56,7 +56,7 @@ end
         y = Reactant.to_rarray(rand(Float32, 28, 28, 1, 4))
 
         # Export to EnzymeJAX
-        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymeax(
+        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymejax(
             add_3d, x, y;
             output_dir=tmpdir,
             function_name="add_3d"
@@ -86,7 +86,7 @@ end
 
         x = Reactant.to_rarray(Float32[1.0, 2.0, 3.0, 4.0])
 
-        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymeax(
+        mlir_path, python_path, input_paths = Reactant.Serialization.export_to_enzymejax(
             simple_fn, x;
             output_dir=tmpdir,
             function_name="test_fn"