Predator-Prey Flock Environment

[zombie-einstein]

Hi, I am the maintainer/creator of Esquilax a JAX large scale sim and RL environment library.

As a first use case of the library I've been working on a predator-prey training environment:

• Two sets of agents aim to pursue/evade each other
• It's intended for larger scale use cases (i.e. 100s of agents)
• Each agent has a simple view model of the surrounding environment, allowing for observations space to independent of the number of agents
• All implemented in native JAX

Would this be something you'd be interested in porting into Jumanji? I'd be happy to work on implementation. The version on the linked repo is an initial prototype, with lots of scope to add different setups (e.g. communication between predators, multiple competing agent types).

[sash-a]

Hi @zombie-einstein

Thanks for opening the issue, I remember chatting to you about this env on reddit 👋

Personally I think it would be a nice addition, although this is supposed to be an industry focused benchmark suite, so I'll wait for @clement-bonnet to give his opinion.

If you are happy to implement this I'm happy to review it, but I can't devote too much other time to this

[sash-a]

Hi @zombie-einstein sorry for leaving this hanging for a while if you are still interested in adding this to jumanji we'd be happy to accept the addition 🙏

[zombie-einstein]

No worries @sash-a , great I'll look to make a start on this!

[zombie-einstein]

Closing this issue. Will open a PR with implementation.

[zombie-einstein]

Hi @sash-a just wanted to check what the best tactic was to get this reviewed. I've got the core functionality on a branch of my fork here, but it's still missing docs, some more in-depth tests, etc.
I was trying to avoid opening too large a PR, so what was best to do?

• Open a PR for the added functionality and then additional PRs to finalise?
• Use the Fork as a staging area and review PRs into the Fork main, before merging the whole fork?
• Just finish up the whole thing in one big PR?

[sash-a]

Good question, so we usually use option 3 and just review the whole PR at once. Option two sounds best but honestly I'm quite used to option 3, so just go ahead and create a big PR for the whole thing, it ends up taking a while and getting lots of comments but I think it's relatively well tried and tested

[sash-a]

Re-opening until this env is merged 😄

[zombie-einstein]

Thanks, yeah makes sense.