vmap

JAX에서 제공하는 자동 벡터화 도구로, 단일 데이터에 대해 작성된 함수를 배치 데이터에 대해 병렬로 실행되도록 변환한다. 루프 없이 고성능 병렬 연산을 가능케 한다.