Radix/zml/zml.zig

//! Welcome to the ZML API documentation!
//! ZML provides tools to write high level code describing a neural network,
//! compiling it for various accelerators and targets, and executing it.
//!

// Namespaces
const std = @import("std");

pub const platform_specific = @import("c");
pub const tokenizer = @import("zml/tokenizer");

pub const aio = @import("aio.zig");
pub const Buffer = @import("buffer.zig").Buffer;
pub const Bufferized = @import("tensor.zig").Bufferized;
pub const callback = @import("callback.zig");
pub const CompilationOptions = @import("platform.zig").CompilationOptions;
pub const context = @import("context.zig");
pub const Context = @import("context.zig").Context;
pub const Data = @import("dtype.zig").Data;
pub const DataType = @import("dtype.zig").DataType;
pub const exe = @import("exe.zig");
pub const compile = exe.compile;
pub const compileWithPrefix = exe.compileWithPrefix;
pub const compileFn = exe.compileFn;
pub const compileModel = exe.compileModel;
pub const FnExe = exe.FnExe;
pub const ModuleExe = exe.ModuleExe;
pub const ModuleSignature = exe.ModuleSignature;
pub const floats = @import("floats.zig");
pub const helpers = @import("helpers.zig");
pub const HostBuffer = @import("hostbuffer.zig").HostBuffer;
pub const meta = @import("meta.zig");
pub const mlir = @import("mlirx.zig");
pub const module = @import("module.zig");
pub const nn = @import("nn.zig");
pub const ops = @import("ops.zig");
pub const call = ops.call;
pub const pjrt = @import("pjrtx.zig");
pub const platform = @import("platform.zig");
pub const Platform = @import("platform.zig").Platform;
pub const Shape = @import("shape.zig").Shape;
pub const ShapeOf = @import("tensor.zig").ShapeOf;
pub const Target = @import("platform.zig").Target;
pub const Tensor = @import("tensor.zig").Tensor;
pub const shapesOf = @import("tensor.zig").shapesOf;
pub const testing = @import("testing.zig");
pub const torch = @import("torch.zig");

pub const tools = struct {
    pub const Tracer = @import("tools/tracer.zig").Tracer;
};

pub const log = std.log.scoped(.zml);

test {
    // NOTE : testing entrypoint.
    // Don't forget to import your module if you want to declare tests declarations that will be run by //zml:test
    std.testing.refAllDecls(@This());
}