[INVESTIGATION] Mava composability

[DriesSmit]

What do you want to investigate?

Determine a better Mava standard that would make the executors and trainers more composable. We want to make it easy to add new functionality to them with the constraint that the entire trainer and executor forward pass must be jittable (if applicable). This would require that we define jittable functions inside components, when building them, and execute one function when running them. We would therefore like to find a way to standardise these functions.