BasaltBasalt

activations

Functions

initializers

Functions

loss

Functions

model

Aliases

DEBUG = env_get_int[stdlib::builtin::string_literal::StringLiteral,stdlib::builtin::int::Int]["DEBUG", 0]()

Structs

Parameters
  • Parameters
    • g - Graph
  • Fields
    • params - Collection
    • grads - Collection
  • Functions
  • Aliases
    • trainable_parameters = apply_result_slot(:!lit.signature<[2]("g": !lit.ref<@basalt::@autograd::@graph::@Graph, mut #lit.lifetime> borrow_in_mem, ?, "__result__": !lit.ref<@stdlib::@collections::@list::@List<:trait<@stdlib::@builtin::@value::@CollectionElement> [@basalt::@autograd::@symbol::@Symbol : anystruct<@basalt::@autograd::@symbol::@Symbol>, {"__copyinit__" : !lit.signature<[2]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, "existing": !lit.ref<@basalt::@autograd::@symbol::@Symbol, imm *[0,1]> borrow_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[2](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, !lit.ref<@basalt::@autograd::@symbol::@Symbol, imm *[0,1]> borrow_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__copyinit__(basalt::autograd::symbol::Symbol=&,basalt::autograd::symbol::Symbol)_thunk"), "__del__" : !lit.signature<[1]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> owned_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[1](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> owned_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__del__(basalt::autograd::symbol::Symbol)_thunk"), "__moveinit__" : !lit.signature<[2]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, "existing": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,1]> owned_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[2](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,1]> owned_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__moveinit__(basalt::autograd::symbol::Symbol=&,basalt::autograd::symbol::Symbol)_thunk")}]>, mut #lit.lifetime> byref_result) -> !kgen.none> rebind(:!lit.signature<[2]("g": !lit.ref<@basalt::@autograd::@graph::@Graph, imm *[0,0]> borrow_in_mem, ?, "__result__": !lit.ref<@stdlib::@collections::@list::@List<:trait<@stdlib::@builtin::@value::@CollectionElement> [@basalt::@autograd::@symbol::@Symbol : anystruct<@basalt::@autograd::@symbol::@Symbol>, {"__copyinit__" : !lit.signature<[2]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, "existing": !lit.ref<@basalt::@autograd::@symbol::@Symbol, imm *[0,1]> borrow_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[2](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, !lit.ref<@basalt::@autograd::@symbol::@Symbol, imm *[0,1]> borrow_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__copyinit__(basalt::autograd::symbol::Symbol=&,basalt::autograd::symbol::Symbol)_thunk"), "__del__" : !lit.signature<[1]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> owned_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[1](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> owned_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__del__(basalt::autograd::symbol::Symbol)_thunk"), "__moveinit__" : !lit.signature<[2]("self": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, "existing": !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,1]> owned_in_mem, |) -> !kgen.none> = rebind(:!lit.signature<[2](!lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,0]> init_self, !lit.ref<@basalt::@autograd::@symbol::@Symbol, mut *[0,1]> owned_in_mem, |) -> !kgen.none> @basalt::@autograd::@symbol::@Symbol::@"__moveinit__(basalt::autograd::symbol::Symbol=&,basalt::autograd::symbol::Symbol)_thunk")}]>, mut *[0,1]> byref_result) -> !kgen.none> @basalt::@nn::@model::@"collect_trainable_parameters(basalt::autograd::graph::Graph)"), store_to_mem(g))
Model
  • Parameters
    • g - Graph
    • n_inference_nodes - Optional[Int]
  • Fields
    • parameters - Parameters[g]
    • perf_metrics - PerfMetrics
  • Functions

Functions

optim

Structs

Adam
  • Parameters
    • g - Graph
  • Fields
    • lr - SIMD[f32, 1]
    • beta1 - SIMD[f32, 1]
    • beta2 - SIMD[f32, 1]
    • epsilon - SIMD[f32, 1]
    • iter - Int
    • rms_grads - Collection
    • momentum_grads - Collection
  • Functions

Functions

tensor

Aliases

MAX_RANK = 8

Structs

TensorShape
  • Functions
Tensor
  • Parameters
    • dtype - DType
  • Functions