Crate rai

Source

Modules§

device
dispatch
ext
nn
ops
opt
primitives
ty_kind
utils

Macros§

non_differentiable

Structs§

Aux
BF16
CandleBackend
Cpu
Cuda
F16
F32
F64
I64
Metal
Tensor
U8
U32

Enums§

Error

Traits§

AsDType
AsDevice
Backend
DType
Device
Dim
Dims
ElemType
Eval
FloatElemType
Func
GenericValue
Primitive
Shape
TensorIter
Type
Value
ValueSpec

Functions§

eval
grad
jvp
linearize
optimize
raiexpr
value_and_grad
vjp

Type Aliases§

Result

Derive Macros§

Module