잡다구리 너구리

필자는 기계공학과의 한 연구실에서 학부 연구생으로 일하고 있다. 요즘 워낙 인공지능이 대세이다 보니 기계공학과에서도 Physics-Informed Neural Network(PINN)을 비롯해 여러 신경망들을 기계공학에 맞게 사용하는 연구들이 진행되고 있다. 물론 나도 그중 하나... 각설하고 최근 Physics-Informed DeepONet(PIDON)이라는 Neural Operator에 대해 공부하게 되었는데, 해당 코드가 Jax가 주가 되어 코드가 돌아가 Jax에 대해 약간이나마 조사를 해보았다. 본문은 어디까지나 아직 Jax를 본격적으로 공부하지 않았고, 필자가 본 PIDON 코드 기준으로 작성이 되어서 많이 주관적일 수 있다는 점 참고 바란다. Jax는 최근 구글에서 만든 딥러닝을 위한 프레임..