Shypd.aiThe largest open index of AI toolsDI-engine is an open-source AI Frameworks & Infra tool that provides a comprehensive reinforcement learning framework. It supports various deep reinforcement learning algorithms and decision intelligence environments for both academic research and applications.
Trending
About
DI-engine is a generalized decision intelligence engine built for PyTorch and JAX, offering a comprehensive framework for reinforcement learning. It features python-first and asynchronous-native task and middleware abstractions, integrating key decision-making concepts like Env, Policy, and Model. The framework supports a wide array of deep reinforcement learning algorithms, including DQN, PPO, SAC, and multi-agent, imitation, offline, and model-based RL. Beyond algorithms, DI-engine aims to standardize decision intelligence environments and applications, catering to academic research and prototype development. It also includes highly re-usable modules for RL optimization, PyTorch utilities, and system optimizations for efficient large-scale RL training.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
