trax

Expand experiments with fixed-base position encodings; current config gets 90%+ seq-level acc on copy and 5%+ on addition.

Pass in rng to policy & value net initialization.