I have a multi-agent setting with two agents. I would like the two agents to share the same network except for the last layer. I know that custom RL modules can help with this (RL Modules (Alpha) — Ray 2.9.0). Does custom RL modules support GTrXLNet (i.e. I would like two agents to share the same GTrXLNet)?