Based on some comments in the Rust code, Keccak implementation in Plonky2 is based on tiny-keccak crate which is also based on or related to the C-version keccak-tiny.
Our C version is taken from saarinen-keccak.
The CUDA version is a simple adaptation.