nvmath-python: Python에서 NVIDIA 수학 라이브러리 활용

2025-09-22
nvmath-python: Python에서 NVIDIA 수학 라이브러리 활용

nvmath-python은 NVIDIA 수학 라이브러리의 강력한 기능을 Python 생태계로 가져옵니다. 직관적인 Pythonic API를 제공하여 다양한 실행 공간에서 제공되는 NVIDIA 라이브러리의 모든 기능에 완벽하게 액세스할 수 있습니다. nvmath-python은 기존 Python 배열/텐서 프레임워크와 원활하게 통합되며, 해당 프레임워크에는 없는 기능을 제공하는 데 중점을 둡니다. 또한, 기본 cuBLASLt 라이브러리의 모든 매개변수에 액세스할 수 있으며, 다른 NVIDIA C-API 라이브러리 래퍼에서는 사용할 수 없는 매개변수도 포함됩니다. 뿐만 아니라 FFT 함수의 사용자 정의 프롤로그와 에필로그를 작성하고 이를 LTO-IR로 컴파일할 수도 있습니다.