tensorflow/tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

해결하고자 하는 문제

TensorFlow.js는 개발자가 브라우저나 Node.js에서 직접 머신러닝 모델을 학습·배포할 수 있게 해 주어, 각 ML 작업마다 전용 백엔드 서버를 구축할 필요를 없앱니다. 센서 데이터와 같은 클라이언트 측 데이터를 서버로 전송하지 않고도 모델을 재학습할 수 있습니다.

작동 방식

저수준 선형대수 라이브러리 또는 Keras와 유사한 고수준 Layers API를 사용해 처음부터 모델을 구축할 수 있는 API 세트를 제공합니다. WebGL, WebGPU, WASM, Node.js용 네이티브 C++ 어댑터 등 여러 하드웨어 가속 백엔드를 지원해 성능을 보장합니다. 또한 기존 TensorFlow 및 Keras 모델을 JavaScript 환경에서 사용할 수 있도록 변환·임포트하는 도구도 포함되어 있습니다.

대상 사용자

웹 애플리케이션, Node.js 서버‑사이드 애플리케이션, 혹은 React Native 모바일 앱에 머신러닝을 통합하고자 하는 JavaScript 개발자를 위한 것입니다.

주요 특징

  • 멀티플랫폼 지원: 브라우저(WebGL, WebGPU, WASM)와 Node.js 모두에서 동작합니다.
  • 유연한 API: 저수준 수치 연산과 고수준 모델 구축을 모두 제공합니다.
  • 모델 변환: 사전 학습된 TensorFlow SavedModel 및 Keras 모델을 임포트하고 실행할 수 있습니다.
  • 클라이언트‑사이드 학습: 로컬 센서 또는 클라이언트‑사이드 데이터를 사용한 재학습을 지원합니다.
  • 시각화: 브라우저 내 모델 시각화를 위한 전용 라이브러리를 포함합니다.