quantax.utils.apply_updates

quantax.utils.apply_updates(model: PyTree, updates: PyTree) → PyTree

Similar to equinox.apply_updates, but the original data type of the model is kept unchanged.