Docs.rs
forge-ml-0.1.0
Platform
x86_64-unknown-linux-gnu
Feature flags
docs.rs
About docs.rs
Badges
Builds
Metadata
Shorthand URLs
Download
Rustdoc JSON
Build queue
Privacy policy
Rust
Rust website
The Book
Standard Library API Reference
Rust by Example
The Cargo Guide
Clippy Documentation
forge-ml 0.1.0
A WebGPU-native machine learning runtime in Rust: train and run GPT-2 on any GPU wgpu reaches, with no CUDA and no Python
Crate
Source
Builds
Feature flags
Documentation
..
adamw.wgsl
add.wgsl
ce_bwd.wgsl
dropout.wgsl
embedding.wgsl
gather_nll.wgsl
gelu.wgsl
gelu_bwd.wgsl
kv_append.wgsl
layernorm.wgsl
layernorm_bwd_dp.wgsl
layernorm_bwd_dx.wgsl
matmul.wgsl
merge_heads.wgsl
scale.wgsl
scatter_add.wgsl
softmax.wgsl
softmax_bwd.wgsl
split_heads.wgsl
sum_rows.wgsl
sumsq.wgsl
unmerge_heads.wgsl
unsplit_heads.wgsl