Crate rai_core

Source

Re-exports§

pub use primitives::Primitive;
pub use device::AsDevice;
pub use device::Cpu;
pub use device::Cuda;
pub use device::Device;
pub use device::Metal;

Modules§

device
dispatch
nn
ops
primitives
ty_kind
utils

Macros§

non_differentiable

Structs§

Aux
BF16
CandleBackend
F16
F32
F64
I64
Tensor
U8
U32

Enums§

Error

Traits§

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

Functions§

eval
grad
jvp
linearize
optimize
raiexpr
value_and_grad
vjp

Type Aliases§

Result