search

sotetsuk / pgx

♟️ Vectorized RL game environments in JAX (NeurIPS23)
http://sotets.uk/pgx/
Apache License 2.0
404 stars 25 forks source link

Convert Haiku baseline models into Flax #1200

Open sotetsuk opened 3 months ago

sotetsuk commented 3 months ago

Or add new Flax baseline and deprecate Haiku baselines