Backends#

Core#

base.BaseDevice(config, device_config[, state])

cpu.CPUDevice(config, device_config[, state])

cuda.CUDADevice(config, device_config[, state])

cuda.Kernel(code, dev, block_size, n_sms)

cuda.Launcher(fn, dev, block_size, n_sms)

buffers.FitBuffers()

Arrays the fit kernels read and write, kept on the device for a whole epoch.

buffers.PackedClauses()

Clauses in compact form for evaluation.

toolchain.Toolchain([openmp])

Builds C code into shared libraries for the cpu backend.

toolchain.compiler_cmd()

Tsetlin Machine#

cpu.CPUDevice(config, device_config[, state])

cuda.CUDADevice(config, device_config[, state])

buffers.TMFitBuffers()

Fit buffers of the TM.

Loss-Guided TM#

cpu.CPUDevice(config, device_config[, state])

cuda.CUDADevice(config, device_config[, state])

buffers.GuidedFitBuffers()

Fit buffers of the guided TM.