enlite-ai/maze

Provides PyTorch-based Perception Modules for composing neural network architectures (dense, convolutional, graph, attention, recurrent) with built-in observation preprocessing and masking. Supports multi-step and multi-agent environments beyond standard Gym compatibility, alongside trainers for A2C, PPO, Impala, SAC, and Evolution Strategies with dictionary action/observation spaces. Configuration managed through Hydra, enabling imitation learning and policy fine-tuning workflows for complex industrial decision problems.

286 stars.