> JAX is just numpy (mostly)
Or you could say JAX is a DSL that looks like numpy to trick Python programmers into using a compiled functional programming language. ;)
Or you could say JAX is a DSL that looks like numpy to trick Python programmers into using a compiled functional programming language. ;)