✨ 팀 소개

<aside> 💡 JAX를 공부하며 활용하는 스터디입니다.

</aside>

🚩 빌더 소개

• 박우성
• 안녕하세요 강화학습을 공부하는 박우성입니다. JAX 공부에 대한 필요성을 느껴 스터디를 구성하게 되었습니다 잘 부탁드립니다!
• 가짜연구소 활동

JAX?

JAX는 고성능 수치 계산과 머신 러닝 연구를 위해 개발된 Python 라이브러리입니다. 이는 구글 리서치 팀에 의해 만들어졌으며, 주로 복잡한 수치 계산을 최적화하고 빠른 실험 및 연구를 가능하게 하기 위해 설계되었습니다. JAX는 NumPy와 유사한 API를 제공하지만 자동 미분 및 가속화된 컴퓨팅을 위한 추가 기능이 특징입니다. 이를 통해 사용자는 Python 코드를 효율적으로 GPU 및 TPU에서 실행할 수 있습니다.

JAX는 다양한 기술 및 연구 기관에서 사용됩니다. 특히 DeepMind에서는 JAX를 여러 프로젝트에 활용하여 AI 연구에 기여하고 있으며, 이를 통해 신속한 실험과 새로운 알고리즘 및 아키텍처의 개발을 지원하고 있습니다. 또한, Google Brain과 다른 기관들도 JAX를 사용하여 연구와 프로젝트를 진행하고 있다고 합니다. - ChatGPT

이미지 출처 : Why You Should (or Shouldn't) be Using Google's JAX in 2023

📢 스터디 공지

<aside> 📢 스터디 시간은 매주 금요일 21:00 입니다.

</aside>

⚽ Ground Rule

<aside> 📢 스터디 규칙

• 시간을 잘 지켜주세요.
• 출석체크를 위해 실명으로 참여 부탁드려요.
• 서로간의 Respect 부탁드립니다.
• 3회 이상 무단 불참시 제명 및 수료 불가합니다. (OT, 리뷰 제외) </aside>

👟 학습 활동

스터디/프로젝트 내용

• JAX 공식 문서(Tutorial, Advanced Tutorial, User Guide…)를 이해하여 JAX에 입문합니다.
• JAX 공식 문서를 읽어보며 문제를 내보고 서로의 문제를 풀어보며 JAX에 대한 더 깊은 이해를 목표로 합니다.

스터디/프로젝트 목적

• JAX 공식문서를 통해 JAX의 속도가 빠른 이유를 이해하고 실제 코드에 활용할 수 있도록 한다.
• JAX로 구현된 많은 오픈소스를 이해할(읽을) 수 있도록 한다.
• JAX 스타일의 코드에 익숙해질 수 있도록 한다.

모집 대상

<aside> 💡 JAX 코드를 이해하고 활용하고 싶으신 분

</aside>

• Pytorch 또는 Tensorflow에 대한 기본적인 경험이 있으신 분
• 파이썬에 대한 기본적인 이해가 있으신 분