How ponseko/econojax works
EconoJax is a JAX-based economic simulation environment, trained in the supplied entry point with PPO. The workflow initializes the environment, trains agents through its environment interface, and simulates gathering/crafting, market trading, government taxation, and utility outcomes. The sampled source confirms these mechanics and their step-level calls; unsampled implementation details are not expanded.
Main entry points: econojax.py EconoJax · main.py Training entry · util.py Gini metric
Components 12
Training
- Training entry
main.py - PPO trainer
- Log wrapper
Environment
- EconoJax
econojax/econojax.py - Environment state
econojax/econojax.py - Observations and actions
econojax/econojax.py
Economic mechanics
- Gather and craft
econojax/econojax.py - Market trading
econojax/econojax.py - Government taxation
econojax/econojax.py
Outcomes
- Utility calculation
econojax/econojax.py - Gini metric
econojax/util.py
Other components
- Researcher
Connections 17
- Researcher → Training entry: starts training
- Training entry → EconoJax: creates environment
- Training entry → Log wrapper: wraps environment
- Training entry → PPO trainer: starts training
- Training entry → Log wrapper: passes to trainer
- PPO trainer → Log wrapper: trains through
- EconoJax → Environment state: initializes and updates
- EconoJax → Observations and actions: provides
- EconoJax → Gather and craft: dispatches
- EconoJax → Market trading: dispatches
- EconoJax → Government taxation: dispatches
- EconoJax → Utility calculation: calculates
- Gather and craft → Environment state: updates
- Market trading → Environment state: updates
- Government taxation → Environment state: updates
- Utility calculation → Environment state: updates
- Utility calculation → Gini metric: calculates equality with