rai-core-0.10.0
- rai-core 0.10.0
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- cksac
- Dependencies
- candle-core ^0.4.1 normal optional
  
  candle-flash-attn ^0.4.1 normal optional
  
  colored ^2.1.0 normal
  
  dyn-clone ^1.0.16 normal
  
  half ^2.3.1 normal
  
  num-traits ^0.2.17 normal
  
  once_cell ^1.19.0 normal
  
  paste ^1.0 normal
  
  safetensors ^0.4.1 normal
  
  thiserror ^1.0.51 normal
  
  tracing ^0.1.40 normal
- Versions
- 10.58% of the crate is documented
Go to latest version
Platform
Feature flags

docs.rs

Rust

rai_core0.10.0

All Items

Re-exports
Modules
Macros
Structs
Enums
Traits
Functions
Type Aliases

Crate rai_core

Re-exports§

pub use primitives::Primitive;

Modules§

dispatch
nn
ops
primitives
ty_kind
utils

Macros§

non_differentiable

Structs§

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

Enums§

Error

Traits§

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

Functions§

eval
grad
jvp
linearize
raiexpr
value_and_grad
vjp

Type Aliases§

Result