Open
Description
This issue focuses on giving users deeper insight into model resource consumption by adding utilities to inspect memory usage layer-by-layer in Lux.jl models.
Success when:
A utility function is added that returns memory usage per layer in Lux models during forward/backward pass.
Example use cases are provided.
Metadata
Metadata
Assignees
Labels
No labels