Так проблема в том что экшен - это не стрим, а аналог события, он единовременный.
Но ты абсолютно тоже самое можешь пистаь так
const lhs = declareAtom(0)
const rhs = declareAtom(0)
const sumAtom = declareAtom(($, state = 0) => {
return $(lhs) + $(rhs)
})
store.dispatch(lhs.update(2))