acme VS taichi

I'm tired of having my 200m frames of Atari take 5 days to run with dopamine, so I'm looking for another framework to use. I haven't been able to find one that's fast and hackable, preferably distributed or with vectorized environments. Anybody have suggestions? seed-rl seems promising but is archived (and in TF2). sample-factory seems super fast but to the best of my knowledge doesn't work with replay buffers. I've been trying to get acme working but documentation is sparse and many of the features are broken.

I'm training a few algorithms from Deepmind's acme library on some MuJoCo models and I'm wondering how long this will take to train and what it's going to do to my electric bill. Is a 3090 or two enough to train something to keep its balance, or do a task, or do I need to wait for the 8090 to come out?

Recently dm-acme also added support for multi-agent environments. Acme: https://github.com/deepmind/acme

Can anyone speak to their experience using acme (https://github.com/deepmind/acme) and by extension dm_env (https://github.com/deepmind/dm_env)? I'm wondering if it would be worthwhile for me to invest the time into converting my custom environment (which loosely follows the standard RL setup) over to this format.

acme/acme/agents/tf/mpo at master · deepmind/acme · GitHub

after installing it in a clean env, I tried to run the example provided for solving the gym cartpole env: https://github.com/deepmind/acme/blob/master/examples/control/run_d4pg_gym.py

Acme has a slightly different one: https://github.com/deepmind/acme which includes specs for agents, buffers etc. It is very general. You can see their component description here: https://github.com/deepmind/acme/blob/master/docs/components.md

taichi – Productive, portable, and performant GPU programming in Python

ETH Zurich is using it for their physics sim courses, University of Utah is using it for simulations (SIGGRAPH 2022), OPPO (they make smart devices running Android), Kuaishou uses it for liquid and gas simulation on GPUs. Lots of GPU accelerated sim stuff.

https://www.taichi-lang.org/

https://www.researchgate.net/publication/337118128_Taichi_a_...

https://github.com/taichi-dev/taichi

Check our the realease note (https://github.com/taichi-dev/taichi/releases) for more improvements.

I've recently started using Taichi (https://taichi-lang.org/) for numerical codes and the fact it doesn't try to trick you into thinking it's numpy is a nice "feature". ;)

Taichi v1.4.0 is released! See what's new: - Taichi AOT, along with a native Taichi Runtime library: Native applications can now load compiled AOT modules and launch Taichi kernels without a Python interpreter. - Taichi ndarray: An array object that holds contiguous multi-dimensional data to allow easy data exchange with external libraries. - Dynamic index: Use variable indices whenever necessary on all backends without affecting the performance of those matrices with only constant indices. See deprecation and more improvements in the release note.

Oh, then if you're not already tied to any particular VFX software, I might as well recommend Taichi again.

It's an upgrade of a basic version. See changes to the source code here: https://github.com/taichi-dev/taichi/pull/6851