NeuralABM is a Python framework that combines agent-based modeling with neural networks. It enables developers to define agent behaviors via differentiable neural modules, train agents using gradient-based optimization, and simulate multi-agent environments for research or game development. With built-in tools for data collection, visualization, and customization, NeuralABM streamlines the process of creating intelligent, learnable agents that adapt and evolve over time.
NeuralABM is a Python framework that combines agent-based modeling with neural networks. It enables developers to define agent behaviors via differentiable neural modules, train agents using gradient-based optimization, and simulate multi-agent environments for research or game development. With built-in tools for data collection, visualization, and customization, NeuralABM streamlines the process of creating intelligent, learnable agents that adapt and evolve over time.
NeuralABM is an open-source Python library that leverages PyTorch to integrate neural networks into agent-based modeling. Users can specify agent architectures as neural modules, define environment dynamics, and train agent behaviors using backpropagation across simulation steps. The framework supports custom reward signals, curriculum learning, and synchronous or asynchronous updates, enabling the study of emergent phenomena. With utilities for logging, visualization, and dataset export, researchers and developers can analyze agent performance, debug models, and iterate on simulation designs. NeuralABM simplifies combining reinforcement learning with ABM for applications in social science, economics, robotics, and AI-driven game NPC behaviors. It provides modular components for environment customization, supports multi-agent interactions, and offers hooks for integrating external datasets or APIs for real-world simulations. The open design fosters reproducibility and collaboration through clear experiment configuration and version control integration.
Who will use NeuralABM?
Researchers in agent-based modeling
Game developers
AI/ML practitioners
Academic educators
Economists and social scientists
How to use the NeuralABM?
Step1: Install NeuralABM via pip: pip install neuralabm
Step2: Import NeuralABM and define neural network modules for agents
Step3: Configure environment dynamics and reward functions
Step4: Initialize the training pipeline and train agents using backpropagation
Step5: Monitor simulation metrics and collect data
Step6: Visualize results and analyze agent behaviors