Expand description
Module for data type related traits and structs. Contains things like Unit, Dtype, and AMP.
When the f16
feature is enabled, this exports the [f16] type.
AMP
AMP is a technique for mixed precision training.
This is a data type in dfdx, you can use it like any normal dtype like AMP<f16>
or AMP<bf16>
.
Structs
- Wrapper type around the storage type. Use like
AMP<f16>
orAMP<bf16>
.