News for hackers. Stuff that matters.

A comment on Swift for TensorFlow – A system for deep learning and differentiable computing. Parent comment

Re:Swift for TensorFlow – A system for deep learning and dif... by brilee on Sunday September 20, 2020 @12:02PM (#24534737)
(googler)

Not quite sure why you picked those particular examples... JAX also requires usage of lax.cond, lax.while_loop, and ops.segment_sum. Only gather has been improved with slice notation support. IMO, TF has landed on a pretty nice solution to cond/while_loop via AutoGraph.

[ Parent | on HN ]
Re:Swift for TensorFlow – A system for deep learning and dif... by joaogui1 on Sunday September 20, 2020 @08:18PM (#24538561)
While jax has those operations you don't always need them, it depends on what transformations you want to do (JIT or grad) and they have been working on making normal control structures compatible with all transformations
[ Parent | on HN ]