Docs.rs
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
i686-pc-windows-msvc
i686-unknown-linux-gnu
x86_64-apple-darwin
x86_64-pc-windows-msvc
x86_64-unknown-linux-gnu
Feature flags
Rust
About docs.rs
Privacy policy
Rust website
The Book
Standard Library API Reference
Rust by Example
The Cargo Guide
Clippy Documentation
rai_core
0.10.0
Module primitives
Structs
Traits
In crate rai_core
?
Module
rai_core
::
primitives
source
·
[
−
]
Structs
§
Abs
Add
Arange
ArgMax
ArgMin
AvgPool1d
AvgPool2d
Broadcast
Concatenate
Conv1d
Conv2d
ConvTranspose1d
ConvTranspose2d
Cos
Div
Equal
Erf
Exp
FlashAttention
FromArray
Full
Gather
Greater
GreaterEqual
IndexAdd
IndexSelect
Less
LessEqual
Log
Log2
Log10
LogSoftmax
MatMul
MaxPool1d
MaxPool2d
Maximum
Minimum
Mul
Narrow
Negative
Normal
NotEqual
Permute
PowerFloat
Random
ReduceMax
ReduceMin
ReduceSum
Reshape
Rsqrt
ScatterAdd
Sign
Sin
Softmax
Sqrt
Square
Sub
Tanh
ToContiguous
ToDType
ToDevice
Transpose
UpsampleNearest1d
UpsampleNearest2d
Where
Traits
§
Primitive