Switch to NP from Jax to improve attention manager performance#184

FanhaiLu1 merged 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/jetstream-pytorch:mainfrom

FanhaiLu1:pa_decode_checkin_3FanhaiLu1/jetstream-pytorch:pa_decode_checkin_3Copy head branch name to clipboard