[MAINTAIN] Standardise typing

[arnupretorius]

Please describe what needs to be maintained?

Unify the use of certain types across mava and remove our dependency on typing from other libraries such as acme as much as possible.

Describe the outcome you'd like

Consistent typing across all of Mava with sensible custom-defined types if needed.