Skip to content

Craftax prep for 5.0 release - #671

Merged
jsuarez5341 merged 12 commits into
PufferAI:5.0from
daphne-cornelisse:craftax_release
Aug 31, 2026
Merged

Craftax prep for 5.0 release#671
jsuarez5341 merged 12 commits into
PufferAI:5.0from
daphne-cornelisse:craftax_release

Conversation

@daphne-cornelisse

@daphne-cornelisse daphne-cornelisse commented Aug 29, 2026

Copy link
Copy Markdown
Contributor

Description

  • Delete old profiling code [minor]
  • Rendering: Make enchanted armour visible.
  • Replace remaining Jaxisms with C native equivalents; do a pass to improve code readability
  • Delete bloated craftax env code (total of 8132 lines)
  • Delete old craftax files and deprecated tests
  • Ensure craftax is playable (press shift to interrupt)

Parity test is at tests/craftax_parity.py

Evaluate using

bash build.sh crafax
./puffer eval

-> checked on macOS (with and without cpt) ❗️ note: build.sh and puffercpu need a couple adaptations to make eval work on my mac (different pr)

Next PR

  • Separate craftax.h into one clean version and craftax_parity.h containing all the treefry rng logic.

Run parity test to original (output)

daphne@daphne:~/code/PufferLib$ .venv/bin/python tests/craftax_parity.py
An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
Compiling craftax_clean worldgen harness in /home/daphne/code/PufferLib
Compiling craftax_clean replay harness
Loading JAX Craftax-Symbolic-v1
Reference env: CraftaxSymbolicEnvNoAutoReset
PASS seed=0 dungeon_dark=1397 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=1 dungeon_dark=1416 dungeon_chests=7 boss_light=2.9 steps=200
PASS seed=2 dungeon_dark=1248 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=3 dungeon_dark=1366 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=4 dungeon_dark=1282 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=5 dungeon_dark=1214 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=6 dungeon_dark=1295 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=7 dungeon_dark=1222 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=8 dungeon_dark=1421 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=9 dungeon_dark=1283 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=10 dungeon_dark=1123 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=11 dungeon_dark=1505 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=12 dungeon_dark=1375 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=13 dungeon_dark=1287 dungeon_chests=7 boss_light=2.9 steps=200
PASS seed=14 dungeon_dark=1376 dungeon_chests=8 boss_light=2.9 steps=200
PASS seed=15 dungeon_dark=1221 dungeon_chests=7 boss_light=2.9 steps=200
PASS craftax_clean vs JAX Craftax: seeds=16 steps=200 atol=1e-05 dungeon_chests=8 dungeon_darkness=1397 sewer_ice_table=1 vault_fire_table=1 fire_light=255.0 boss_light=2.9

@jsuarez5341
jsuarez5341 merged commit ffc8a4f into PufferAI:5.0 Aug 31, 2026
0 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants