Add support for Craftax#40

Open

rhaps0dy wants to merge 56 commits intomainfrom

tuphs28/craftax

Commits on Feb 21, 2025

Craftax DRC, with small changes
Adrià Garriga-Alonso
committed

Commits on Feb 22, 2025

Add craftax, use the network's norm
Adrià Garriga-Alonso
committed
Fix some of the jax madness in environment
Adrià Garriga-Alonso
committed
Reintroduce obs_flat
Adrià Garriga-Alonso
committed
set env_params, transform to class(cfg) style
Adrià Garriga-Alonso
committed
Runs but get locked
Adrià Garriga-Alonso
committed
Make env api conform
Adrià Garriga-Alonso
committed

Commits on Feb 23, 2025

Upgrade for jax 5
Adrià Garriga-Alonso
committed
Upgrade dockerfile
Adrià Garriga-Alonso
committed
Build envpool with new bazel for new python
Adrià Garriga-Alonso
committed
Use uv to be FAST
Adrià Garriga-Alonso
committed
Use MIG and vast by default
Adrià Garriga-Alonso
committed
Deal with updated gymnasium
Adrià Garriga-Alonso
committed
Copy hyperparameters from transformer PPO
Adrià Garriga-Alonso
committed
Change details of where things are stored
Adrià Garriga-Alonso
committed
Set the cache from config
Adrià Garriga-Alonso
committed

Commits on Feb 24, 2025

Use tanh, orthogonal network
Adrià Garriga-Alonso
committed
Cache tune data
Adrià Garriga-Alonso
committed
fix network bug
Adrià Garriga-Alonso
committed
Refactor for PPO
Adrià Garriga-Alonso
committed

Commits on Feb 25, 2025

Downgrade gymnasium, fix test_convlstm
Adrià Garriga-Alonso
committed
Fix more tests
Adrià Garriga-Alonso
committed
Calculate and store the last value
Adrià Garriga-Alonso
committed
PPO seems correct according to the tests
Adrià Garriga-Alonso
committed
I think PPO is correct
Adrià Garriga-Alonso
committed
Add all requirements
Adrià Garriga-Alonso
committed
oopsie, wrong PPO value length
Adrià Garriga-Alonso
committed
update for profile
Adrià Garriga-Alonso
committed
No more block until ready
Adrià Garriga-Alonso
committed

Commits on Feb 26, 2025

Move from step_async to step, modify deps
Adrià Garriga-Alonso
committed
Fewer things error
Adrià Garriga-Alonso
committed
Use NamedTuple to prevent dumb errors
Adrià Garriga-Alonso
committed
Fix stuff
Adrià Garriga-Alonso
committed
Fix more stuff
Adrià Garriga-Alonso
committed
Make input devices consistent
Adrià Garriga-Alonso
committed

Commits on Feb 27, 2025

Squeeze a bit more speed
Adrià Garriga-Alonso
committed
Even a little faster
Adrià Garriga-Alonso
committed
Fix _state bug
Adrià Garriga-Alonso
committed
Embarrassingly it's only fixed now
Adrià Garriga-Alonso
committed
Reasonably correct and performant
Adrià Garriga-Alonso
committed
Improve queue timeout behavior in cleanba_impala.py

Adrià Garriga-Alonso
and
claude
committed
Delete unused cartpole tests
Adrià Garriga-Alonso
committed
Tell Claude what to look at
Adrià Garriga-Alonso
committed
Centralize logging
Adrià Garriga-Alonso
committed
Fix problems
Adrià Garriga-Alonso
committed
Perhaps running item at the top helps
Adrià Garriga-Alonso
committed
Include achievements, prune unused metrics
Adrià Garriga-Alonso
committed
Ignore for pyright
Adrià Garriga-Alonso
committed
Try logging achievements
Adrià Garriga-Alonso
committed
Craftax experiments
Adrià Garriga-Alonso
committed

Commits on Feb 28, 2025

Save checkpoints
Adrià Garriga-Alonso
committed
Avoid collecting last value in Impala (not needed)
Adrià Garriga-Alonso
committed

Commits on Mar 1, 2025

Fix loading fenceless checkpoints
Adrià Garriga-Alonso
committed

Commits on Mar 5, 2025

Experiments with checkpoints
Adrià Garriga-Alonso
committed

Commits on Apr 8, 2025

fix compile command
Adrià Garriga-Alonso
committed

Commits on Oct 13, 2025

add noscroll craftax
Adrià Garriga-Alonso
committed