open_spiel icon indicating copy to clipboard operation
open_spiel copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Added JAX implementation of CFR

Open kubicon opened this issue 1 year ago • 3 comments

Implementation of CFR that uses JAX. This allows running CFR with GPU acceleration. Speed up over python CFR implementation is ~10-times on CPU only.

The goal was to make it interchangable with the python CFR.

kubicon avatar Feb 05 '24 12:02 kubicon