๐ฑ API Reference#
Efficiency methods for training. |
|
Callbacks that run at each training loop |
|
Central components used by other modules. |
|
Natively supported datasets. |
|
Functional API for applying algorithms in your own training loop. |
|
Loggers to store metrics and artifacts. |
|
The models module contains the |
|
Optimizers and learning rate schedulers. |
|
Performance profiling tools. |
|
Train models with flexible insertion of algorithms. |
|
Helper utilities. |