걍 파이토치 쓰셈
개인적으론 jax 내지는 julia의 flux도 관심은 음청 가지만 두개를 현업에서 직접 써본적은 없어서 뭐라 말하기 애매허네
jax는 연구소에서는 그래도 좀 쓰는 편인데, 실무에서 이거 쓰는걸 본적이 없음
jax는 inplace 연산 최소화하는 식으로 구현된건가
jax는 pytorch보다 속도는 더 빠른데 메모리 더 잡아먹는거로 알고있음.. GPU 한개쓰는거면 걍 토치쓰셈
걍 파이토치 쓰셈
개인적으론 jax 내지는 julia의 flux도 관심은 음청 가지만 두개를 현업에서 직접 써본적은 없어서 뭐라 말하기 애매허네
jax는 연구소에서는 그래도 좀 쓰는 편인데, 실무에서 이거 쓰는걸 본적이 없음
jax는 inplace 연산 최소화하는 식으로 구현된건가
jax는 pytorch보다 속도는 더 빠른데 메모리 더 잡아먹는거로 알고있음.. GPU 한개쓰는거면 걍 토치쓰셈