Open caelum02 opened 10 months ago
purejaxrl - 학습 로직까지 jax로 구현해서 아예 cpu-gpu 통신을 최소하 하는 듯 throughput이랑 코드 우리한테 적용 가능한지 확인 필요 colab tutorial
RLAX - rlax
purejaxrl - 학습 로직까지 jax로 구현해서 아예 cpu-gpu 통신을 최소하 하는 듯 throughput이랑 코드 우리한테 적용 가능한지 확인 필요 colab tutorial
RLAX - rlax