Here is a list of all modules:
[detail level 1234]
▼binaries | Binaries Module consists of C++ backend exposed via pybind11 to rlpack via rlpack._C. These modules are optimized to perform heavier workloads |
grad_accumulator | Memory module is the C++ backend for rlpack._C.grad_accumulator.GradAccumulator class. Heavier workloads have been optimized with multithreading with OpenMP |
▼memory | Memory module is the C++ backend for rlpack._C.memory.Memory class. Heavier workloads have been optimized with multithreading with OpenMP and CUDA (if CUDA compatible device is found) |
▼offload | Template class to offload some heavier computation to specialised hardware. Functions will be executed in CUDA if CUDA device is available else OpenMP routines will be used to execute the functions on CPU |
cpu | CPU optimized implementation for Offload |
utils | Utils Module provides generic utilities to be used by all binaries in rlpack |