linean

val gridOffsets = remember(columns, rows, itemSize) {
    (0 until columns).map { column ->
        (0 until rows).map { row ->
            DpOffset(
                x = itemSize.width * row,
                y = itemSize.height * column,
            )
        }
    }.flatten()
}

linean

typealias ItemOffset = Animatable<DpOffset, AnimationVector2D>
fun ItemOffset(offset: DpOffset): ItemOffset = Animatable(offset, DpOffset.VectorConverter)

linean

var itemsOffsets by remember { mutableStateOf(mapOf<KEY, ItemOffset>()) }
key(itemKeys) {
   itemsOffsets = items.mapIndexed { index, item ->
       val key = itemKey(item)
       key to when {
           itemsOffsets.containsKey(key) -> itemsOffsets.getValue(key)
           else -> ItemOffset(gridOffsets[index])
       }
   }.toMap()
}

linean

items.forEach { item ->
    val offset = itemsOffsets.getValue(itemKey(item)).value
    Box(
        modifier = Modifier
            .size(itemSize)
            .offset(offset.x, offset.y)
    ) {
        itemContent(item)
    }
}

linean

LaunchedEffect(itemKeys) {
   items.forEachIndexed { index, item ->
       val newOffset = gridOffsets[index]
       val itemOffset = itemsOffsets.getValue(itemKey(item))
       launch { itemOffset.animateTo(newOffset, animationSpec) }
   }
}

linean

// Keep in mind this is just simplified example
class Example {

    private val scope = CoroutineScope(Job())

    private val debouncingChannel = Channel<UserEvent>(
        capacity = RENDEZVOUS,
        onBufferOverflow = BufferOverflow.DROP_OLDEST,
    )