평소처럼 NumPy 코드를 작성하고, 미분 및 배치를 자동 처리하여 GPU와 TPU에서 실행하세요:

요약

이 기술은 사용자가 익숙한 NumPy 코드를 작성하는 것만으로도 미분 및 배치 처리를 자동으로 수행합니다. 이를 통해 GPU나 TPU 같은 가속기에서 고성능으로 코드를 실행할 수 있으며, grad, jit, vmap 등을 조합하여 활용 범위를 확장할 수 있습니다.

핵심 포인트

• 일반적인 NumPy/Python 코드로 작성 가능
• 미분(derivatives) 및 배치 처리 자동화 지원
• GPU와 TPU 등 가속기에서 고성능 실행
• grad, jit, vmap 조합으로 확장성 확보