구글 딥마인드, 펜자이 출시: 신경망 구축, 편집 및 시각화를 위한 JAX 라이브러리 출시

구글 딥마인드는 최근 연구자들이 신경망을 구성, 시각화 및 변경하는 방식을 혁신할 수 있는 새로운 JAX 라이브러리인 Penzai, 을 출시했습니다. 이 혁신적인 도구는 Google Colab 및 JAX 에코시스템과 원활하게 통합되도록 설계되었으며, 이는 AI 모델의 접근성과 조작성에서 중요한 진전입니다.

펜자이 은 투명성과 기능을 강조하는 신경망 개발에 대한 새로운 접근 방식입니다. 이를 통해 사용자는 모델을 읽기 쉬운 파이트리 데이터 구조로 보고 편집할 수 있으므로 모델의 내부 작동을 그 어느 때보다 쉽게 살펴볼 수 있습니다. 이 기능은 모델이 어떻게 작동하는지에 대한 인사이트를 제공하고 원하는 결과를 달성하는 데 도움이 되는 수정을 허용하므로 모델을 학습한 후에 특히 유용합니다.

펜자이 는 사전 학습된 신경망을 수정하는 과정을 간소화하여 연구자들이 AI 연구에 더 쉽게 접근할 수 있도록 하는 것을 목표로 합니다. 이를 통해 더 많은 연구자들이 기존 AI 기술을 실험하고 혁신할 수 있으며, 이는 이 분야를 발전시키고 새로운 AI 애플리케이션을 발견하는 데 매우 중요합니다. 펜자이의 사용자 친화적인 인터페이스는 AI 연구의 장벽을 허물고 누구나 쉽게 이 기술의 혜택을 누릴 수 있게 해줍니다.

펜자이는 포괄적인 모듈식 도구를 제공하는 탁월한 플랫폼으로, 독립적 또는 집단적 활용이 가능한 것이 특징입니다. 이러한 제품 중에는 신경망 라이브러리를 위한 선구적인 조합 방법론을 제시하며 Keras나 하이쿠와 같은 기존 프레임워크와 차별화되는 pzenn(pz.nn)이 있습니다. pz.nn 모듈은 사용자가 모델의 전진 진행 과정 전체에 자유롭게 액세스할 수 있도록 하여 각 구성 요소를 면밀히 검토하고 그에 따라 자발적으로 수정할 수 있게 해줍니다. 이러한 특성은 신경망 아키텍처 내에서 높은 수준의 정밀도와 적응성을 요구하는 전문가에게 없어서는 안 될 리소스입니다.

또 다른 주목할 만한 측면은 모델과 데이터 배열의 묘사를 향상시키는 penzai.treescope(pz.ts)입니다. 이 Python 포맷터는 기본 IPython/Colab 디스플레이어를 대체하도록 특별히 설계되었으며, 다차원 배열 구성의 시각적 표현을 용이하게 하는 동시에 JAX Pytrees 아키텍처에 대한 포괄적인 이해를 제공합니다.

펜자이 코어 셀렉터라고도 불리는 pz.select 모듈은 파이트리를 매우 정교하게 조작하고 패치할 수 있는 매우 강력한 기능을 제공합니다. 이 모듈의 고급 기능을 사용하면 기본 설정 조정을 뛰어넘는 복잡한 변경 작업을 손쉽게 실행할 수 있습니다.또한, 일반적으로 penzai.core.named\_axes로 표시되는 pz.nx 모듈은 네임드 기반에서 포지션 프로그래밍 패러다임으로의 전환을 간소화하는 네임드 축 프레임워크를 도입합니다. 후자의 특성은 사용자가 대체 API에 익숙해지지 않아도 된다는 점에서 특히 유리합니다. 결론적으로, 이 두 가지 리소스는 노력의 실용성을 크게 높이는 동시에 더 효율적이고 효과적인 목표 달성을 촉진합니다.

펜자이는 광범위한 기능 외에도 트리 탐색 과정에서 사이드 인수, 무작위로 생성된 값, 변경 가능한 상태 변수와 같은 추가 입력 파라미터를 효율적으로 관리할 수 있는 혁신적인 도구인 penzai.data\_effects(pz.de)를 제공합니다. 이 사용자 중심 접근 방식은 개인이 자신의 모델에 대해 더 큰 자율성을 부여하여 데이터 분석 작업의 다양성을 강화합니다.

주요 요점:

Penzai 플랫폼은 신경망에 대한 투명성과 제어 기능을 제공하여 사용자가 더 나은 이해와 조작을 위해 구조와 동작을 수정할 수 있도록 합니다. Google Colab 및 JAX와 같은 인기 플랫폼과 원활하게 통합되어 기존 워크플로우와의 호환성을 보장합니다. 또한 향상된 접근성을 제공하여 고급 AI 모델 작업에 대한 진입 장벽을 낮춥니다. 이 플랫폼은 독립적으로 사용하거나 더 큰 시스템에 통합할 수 있는 확장 가능한 모듈식 도구를 갖추고 있어 연구 및 개발 과정에서 유연성을 제공합니다. 마지막으로 펜자이는 복잡한 데이터 구조에 대한 이해와 상호작용을 개선하는 혁신적인 시각화 기법을 도입했습니다.

설치

pip install penzai

가져오기 사용…

import penzai from penzai import pz

Google, NVIDIA, Meta, 스탠포드 대학, 매사추세츠 공과대학, Microsoft 등 저명한 기관의 저명한 학자들이 열독하는 빠르게 확장 중인 AI 연구 뉴스레터를 구독하고, 이 분야의 수많은 선도 기관을 비롯한 여러 기관의 전문가들과 함께하세요.