728x90 반응형 framework2 JAX (Just After eXecution)소개 JAX는수학적 함수(pure function)를 개발자가 작성하면,이를 인자로 받아 자동 미분, JIT 컴파일, 벡터화·병렬화를 적용한 새로운 함수를 반환해 주는 function들을 제공.이를 고성능 수치 계산을 가능하게 하는 함수 변환 라이브러리임ㅣ 딥러닝 프레임워크의 기반이 되어주는 라이브러리로 볼 수 있음.JAX는기존 딥러닝 프레임워크와 사고방식 자체가 다른 도구이며,특히 이론·연구·수식 중심 작업에 매우 잘 맞는 도구로 알려짐.1. JAX (Just After eXecution) 소개JAX의 fully qualified name은 다음과 같음:Just After eXecution이는 “코드를 실행한 직후에 필요한 변환(미분, 컴파일 등)을 수행한다”는 철학을 반영.JAX는 Google에서 개발·유.. 2026. 1. 16. [DL] GPU Acceleration 기술 소개 1. 대표적 GPU 가속 기술현재 가장 많이 사용되고 있는 GPU기반의 가속기술은 다음과 같음:Compute Unified Device Architecture (CUDA): NVIDIAMetal Performance Shaders (MPS): AppleRadeon Open Compute (ROCm): AMDDirect Machine Learning (DirectML): MicrosoftGPU의 대안으로는 TPU(Tensor Processing Unit)도 존재하며, TPU에선 XLA(Accelerated Linear Algebra) 가 필수 인터페이스로 동작함: 참고로 XLA는 GPU, CPU에선 일종의 연산그래프를 최적화하는 Compiler 역할로 동작함.XLA는 TPU, GPU, CPU 모두에서 사용.. 2025. 2. 28. 이전 1 다음 728x90 반응형
