apex?

boingboing·2024년 3월 12일

apex

apex: pytorch의 extension. Pytorch에서 혼합 정밀도 및 분산(distributed) 트레이닝을 간소화하기 위해 NVIDIA가 관리하는 유틸리티가 포함되어 있음.
여기에 있는 코드 중 일부는 향후 업스트림 Pytorch에 포함될 예정입니다. Apex의 의도는 사용자가 최신 유틸리티를 최대한 빠르게 사용할 수 있도록 하는 것입니다.

https://github.com/NVIDIA/apex

0개의 댓글