5 citations · 12 across the 7 of their papers we have counts for
1 paper · 1 filter
Pierrick Pochelu
Among numerical libraries capable of computing gradient descent optimization, JAX stands out by offering more features, accelerated by an intermediate representation known as Jaxpr…