Module Tolk_frontend

module Creation : sig ... end

Tensor creation.

module Device_metal : sig ... end
module Dtype_ops : sig ... end

Type conversions.

module Elementwise : sig ... end

Element-wise operations.

module Jit : sig ... end

Capture-and-replay JIT for tensor functions.

module Movement : sig ... end

Shape movement.

module Op : sig ... end

Composed operations.

module Rand : sig ... end

Random number generation.

module Reduce : sig ... end

Reductions.

module Run : sig ... end

Realization: turning a lazy tensor graph into computed values.

module Tensor : sig ... end

Tensor shell over the unified IR.