xtrain

Files

Gahow Wang c36cdf74d1 Merge t18-dropout into main

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

# Conflicts:
#	README.md
#	crates/xtrain-autodiff/tests/autograd.rs
#	crates/xtrain-model/src/model.rs
#	crates/xtrain-train/src/bin/train.rs
#	crates/xtrain-train/src/train_loop.rs
#	docs/evolution.md

2026-06-18 00:41:41 +08:00

attention.cu

autograd: batch dim for ops (flatten linears, batched attention)

2026-06-16 00:44:15 +08:00

cast.cu

cuda: bf16 cuBLAS GemmEx (16BF in/out, fp32 accum) + cast kernels

2026-06-16 14:14:39 +08:00

dropout.cu

dropout: device RNG kernel + Tensor fwd/bwd (T18)

2026-06-18 00:05:18 +08:00

elementwise.cu

tensor: add scale elementwise CUDA kernel + FFI