MichaelTMatthews/Craftax
A JAX-native reinforcement learning environment that reimplements and extends Crafter mechanics inspired by NetHack.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
Craftax is a research-grade RL environment written entirely in JAX, designed as a benchmark for training reinforcement learning agents. It extends the Crafter environment with roguelike mechanics inspired by NetHack, offering complex survival, crafting, and combat challenges. The project conforms to the gymnax interface for easy integration with existing JAX-based RL frameworks like PureJaxRL.