Convert codebase to JAX using Pallas for CUDA.#30
Open
sguada wants to merge 1 commit intoFelix-Petersen:mainfrom
Open
Convert codebase to JAX using Pallas for CUDA.#30sguada wants to merge 1 commit intoFelix-Petersen:mainfrom
sguada wants to merge 1 commit intoFelix-Petersen:mainfrom
![google-labs-jules[bot]](https://avatars.githubusercontent.com/in/842251?v=4&size=32){kind=avatar}