Needs to be actor_mean instead of embedding, right?

[jforkel]

https://github.com/FLAIROx/JaxMARL/blame/63226c5fbe5385d67d4b3b97511495a0f2c75bb9/baselines/IPPO/ippo_cnn_overcooked.py#L85

[alexunderch]

yes, it's a typo, consider a pr

[amacrutherford]

@jforkel a PR would be fab when you have a moment