Viewing a single comment thread. View all comments

limpbizkit4prez t1_jclabza wrote

What's value in using this instead of "jax.numpy as np"?

8

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