minihack/agent/polybeast/models/base.py
minihack/agent/rllib/models.py

minihack/agent/polybeast/models/intrinsic.py
minihack/agent/rllib/models.py