vwxyzjn / incubator

Collection of in-progress libraries for entity neural networks.

ENN Incubator

Collection of in-progress libraries for entity neural networks: Neural Network Architectures for Structured State

• Entity Gym: Abstraction over reinforcement learning environments that represent observations as lists of structured objects.
• ENN-PPO: PPO training loop for entity-gym environments.
• RogueNet: Ragged batch transformer implementation that accepts variable length lists of structured objects as inputs.
• ENN-Zoo: Collection of entity gym bindings for different environments.
• ENN-BC: Supervised training loop on behavioral cloning datasets collected from entity-gym policies.