Note that using this as your global allocator requires Rust 1.68 or later. (With earlier versions, you need the unstable feature #![feature(default_alloc_error_handler)]) ...
As an example, you could write your own GELU activation function with the high level tensor api (see Rust code snippet below). Then, at runtime, a custom low-level kernel will be automatically created ...