Viewing a single comment thread. View all comments

r_linux_mod_isahoe t1_jclc6rp wrote

porting an existing codebase to jax? Using any existing algorithm that's implemented in numpy but on jax backend? The opportunities are massive

2

limpbizkit4prez t1_jcld70o wrote

Yeah, but if I have a code base written in numpy and want to use jax, wouldn't I need to do the same amount of refactoring to integrate this as I would with regular jax? Are there a lot of functions in numpy that don't exist in jax.numpy?

2