Skip to main content

Crate axonml_jit

Crate axonml_jit 

Source
Expand description

JIT Compilation for Axonml

§File

crates/axonml-jit/src/lib.rs

§Author

Andrew Jewell Sr - AutomataNexus

§Updated

March 8, 2026

§Disclaimer

Use at own risk. This software is provided “as is”, without warranty of any kind, express or implied. The author and AutomataNexus shall not be held liable for any damages arising from the use of this software.

Re-exports§

pub use cache::FunctionCache;
pub use codegen::CompiledFunction;
pub use codegen::JitCompiler;
pub use compile::Backend;
pub use compile::CompileConfig;
pub use compile::CompileStats;
pub use compile::CompiledModel;
pub use compile::LazyCompiled;
pub use compile::Mode;
pub use compile::compile_fn;
pub use compile::compile_fn_with_config;
pub use compile::compile_graph;
pub use compile::compile_graph_with_config;
pub use error::JitError;
pub use error::JitResult;
pub use ir::DataType;
pub use ir::Graph;
pub use ir::Node;
pub use ir::NodeId;
pub use ir::Op;
pub use ir::Shape;
pub use optimize::OptimizationPass;
pub use optimize::Optimizer;
pub use trace::TracedValue;
pub use trace::Tracer;
pub use trace::trace;

Modules§

cache
Function Cache
codegen
Code Generation
compile
torch.compile Equivalent - High-Level Compilation API
error
JIT Error Types
ir
Intermediate Representation
optimize
Graph Optimization
trace
Operation Tracing