|
|
|
|
|
|
by abhgh
814 days ago
|
|
|
I have found learning JAX to be rewarding. Of course this depends on the work you do - but if you want to occasionally code out problems which involve some kind of an optimization of a differentiable quantity (without you having to work out complex gradients), JAX is perfect. |
|
|