Backend 설정 방법 세 가지: (1) env var KERAS_BACKEND=jax python script.py, (2) Python 코드에서 os.environ['KERAS_BACKEND'] = 'jax' 를 *import keras 전*에, (3) ~/.keras/keras.json 에 {"backend": "jax"} 로 영구 설정. 추천 순서: env var > config 파일 > 코드 내 설정.
가장 흔한 실수: import keras 다음에 env 바꾸기 — 안 먹어. import 시점에 backend 가 freeze 돼. 디버깅하다 backend 바꾸려면 process 재시작.
백엔드 노트:
⚙️ Backend Note
Code
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras # Must come AFTER setting the env var