Fix the Order of GatedAct to be (act, linear) when Pax2TE. - Githubissues

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

210 stars 35 forks source link

Fix the Order of GatedAct to be (act, linear) when Pax2TE. #892

Open mingxu1067 opened 1 month ago