tang 0.2.0

Math library for physical reality — geometry, spatial algebra, tensor, training, GPU compute, and 3D gaussian splatting
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! Hand-optimized compute kernels for performance-critical operations.
//!
//! Each submodule contains kernel source code as `const &str` for a specific
//! dialect (MSL or CUDA). These are compiled at runtime and cached by the
//! respective backend.

#[cfg(feature = "metal")]
pub mod matmul_msl;

#[cfg(feature = "metal")]
pub mod reduce_msl;

#[cfg(feature = "metal")]
pub mod attention_msl;

#[cfg(feature = "metal")]
pub mod backward_msl;

#[cfg(feature = "cuda")]
pub mod matmul_cuda;

#[cfg(feature = "cuda")]
pub mod reduce_cuda;

#[cfg(feature = "cuda")]
pub mod attention_cuda;

#[cfg(feature = "cuda")]
pub mod backward_cuda;

#[cfg(feature = "cuda")]
pub mod adamw_cuda;