[tab] /home/franz.srambical/.cache/huggingface/hub/datasets--p-doom--crowd-code-0.1/snapshots/8bec3975ff8d70c5255efed57e06265ea182033c/1f15334ab7e6820c9fda17c961659882ef9853cc80f7356b9a9b22f286fd7389/crowd-code-df82217c-210f-4b7f-897f-6f53642cf7c81754033699441-2025_08_01-09.35.08.848/source.csv (plaintext)
Sequence,Time,File,RangeOffset,RangeLength,Text,Language,Type
1,3,"sample.py",0,0,"from dataclasses import dataclass
import time
import os
import optax
import math

import dm_pix as pix
import einops
import jax
import jax.numpy as jnp
import flax.linen as nn
import numpy as np
import orbax.checkpoint as ocp
from PIL import Image, ImageDraw
import tyro
from flax import nnx

from genie import Genie
from utils.dataloader import get_dataloader


@dataclass
class Args:
    # Experiment
    seed: int = 0
    seq_len: int = 16
    image_channels: int = 3
    image_height: int = 90
    image_width: int = 160
    data_dir: str = ""data/coinrun_episodes""
    checkpoint: str = """"
    # Sampling
    batch_size: int = 1
    maskgit_steps: int = 25
    temperature: float = 1.0
    sample_argmax: bool = True
    start_frame: int = 0
    # Tokenizer checkpoint
    tokenizer_dim: int = 512
    tokenizer_ffn_dim: int = 2048
    latent_patch_dim: int = 32
    num_patch_latents: int = 1024
    patch_size: int = 4
    tokenizer_num_blocks: int = 4
    tokenizer_num_heads: int = 8
    # LAM checkpoint
    lam_dim: int = 512
    lam_ffn_dim: int = 2048
    latent_action_dim: int = 32
    num_latent_actions: int = 6
    lam_patch_size: int = 16
    lam_num_blocks: int = 4
    lam_num_heads: int = 8
    # Dynamics checkpoint
    dyna_type: str = ""maskgit""
    dyna_dim: int = 512
    dyna_ffn_dim: int = 2048
    dyna_num_blocks: int = 6
    dyna_num_heads: int = 8
    param_dtype = jnp.float32
    dtype = jnp.bfloat16
    use_flash_attention: bool = True


args = tyro.cli(Args)

if __name__ == ""__main__"":
    """"""
    Dimension keys:
        B: batch size
        T: number of input (conditioning) frames
        N: number of patches per frame
        S: sequence length
        H: height
        W: width
        E: B * (S - 1)
    """"""
    jax.distributed.initialize()

    rng = jax.random.key(args.seed)

    # --- Load Genie checkpoint ---
    rngs = nnx.Rngs(rng)
    genie = Genie(
        # Tokenizer
        in_dim=args.image_channels,
        tokenizer_dim=args.tokenizer_dim,
        tokenizer_ffn_dim=args.tokenizer_ffn_dim,
        latent_patch_dim=args.latent_patch_dim,
        num_patch_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        tokenizer_num_blocks=args.tokenizer_num_blocks,
        tokenizer_num_heads=args.tokenizer_num_heads,
        # LAM
        lam_dim=args.lam_dim,
        lam_ffn_dim=args.lam_ffn_dim,
        latent_action_dim=args.latent_action_dim,
        num_latent_actions=args.num_latent_actions,
        lam_patch_size=args.lam_patch_size,
        lam_num_blocks=args.lam_num_blocks,
        lam_num_heads=args.lam_num_heads,
        lam_co_train=False,
        # Dynamics
        dyna_type=args.dyna_type,
        dyna_dim=args.dyna_dim,
        dyna_ffn_dim=args.dyna_ffn_dim,
        dyna_num_blocks=args.dyna_num_blocks,
        dyna_num_heads=args.dyna_num_heads,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        decode=True,
        rngs=rngs,
    )

    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeSave, ocp.handlers.PyTreeCheckpointHandler
    )
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )
    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    checkpoint_manager = ocp.CheckpointManager(
        args.checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )

    dummy_tx = optax.adamw(
        learning_rate=optax.linear_schedule(0.0001, 0.0001, 10000),
        b1=0.9,
        b2=0.9,
        weight_decay=1e-4,
        mu_dtype=args.dtype,
    )
    dummy_optimizer = nnx.Optimizer(genie, dummy_tx)

    abstract_optimizer = nnx.eval_shape(lambda: dummy_optimizer)
    abstract_optimizer_state = nnx.state(abstract_optimizer)
    restored = checkpoint_manager.restore(
        checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(abstract_optimizer_state),  # type: ignore
        ),
    )
    restored_optimizer_state = restored[""model_state""]
    nnx.update(dummy_optimizer, restored_optimizer_state)

    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame + 1]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)

    # --- Sample + evaluate video ---
    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)

    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
    recon_video_BSHWC = recon_video_BSHWC.astype(jnp.float32)
    gt = gt_video[:, : recon_video_BSHWC.shape[1]].clip(0, 1).reshape(-1, *gt_video.shape[2:])
    recon = recon_video_BSHWC.clip(0, 1).reshape(-1, *recon_video_BSHWC.shape[2:])
    ssim = jnp.asarray(
        pix.ssim(gt[:, args.start_frame + 1 :], recon[:, args.start_frame + 1 :])
    ).mean()
    print(f""SSIM: {ssim}"")

    # --- Construct video ---
    true_videos = (gt_video * 255).astype(np.uint8)
    pred_videos = (recon_video_BSHWC * 255).astype(np.uint8)
    video_comparison = np.zeros((2, *recon_video_BSHWC.shape), dtype=np.uint8)
    video_comparison[0] = true_videos[:, : args.seq_len]
    video_comparison[1] = pred_videos
    frames = einops.rearrange(video_comparison, ""n b t h w c -> t (b h) (n w) c"")

    # --- Save video ---
    imgs = [Image.fromarray(img) for img in frames]
    # Write actions on each frame, on each row (i.e., for each video in the batch, on the GT row)
    B, S, _, _, _ = video_batch_BSHWC.shape
    action_batch_BSm11 = jnp.reshape(action_batch_E, (B, S-1, 1))
    for t, img in enumerate(imgs[1:]):
        d = ImageDraw.Draw(img)
        for row in range(action_batch_BSm11.shape[0]):
            action = action_batch_BSm11[row, t, 0]
            y_offset = row * video_batch_BSHWC.shape[2] + 2
            d.text((2, y_offset), f""{action}"", fill=255)
    imgs[0].save(
        f""generation_{time.time()}.gif"",
        save_all=True,
        append_images=imgs[1:],
        duration=250,
        loop=0,
    )
",python,tab
2,92,"anysphere.remote-ssh.Remote - SSH",0,0,"2025-07-31 20:25:37.423 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:25:37.467 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][606dd439-5e7c-498d-a9b5-2a244a1f6aca] socks forwarding established
2025-07-31 20:25:37.511 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][606dd439-5e7c-498d-a9b5-2a244a1f6aca] socks connection closed
2025-07-31 20:25:37.511 [info] [command][b8220620-fffb-4818-8513-61402933bff8] Process exited with code 0
2025-07-31 20:25:37.511 [info] [command][b8220620-fffb-4818-8513-61402933bff8] Socket close event received
2025-07-31 20:25:37.550 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53679 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:26:37.515 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:26:37.516 [info] [command][41c56497-62b6-4c5f-aba9-379a90eea401] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""41c56497-62b6-4c5f-aba9-379a90eea401""}
2025-07-31 20:26:37.517 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bdb8ed23-3c2c-4958-a1f8-556ac0b5d104] received connection request
2025-07-31 20:26:37.517 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:26:37.557 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bdb8ed23-3c2c-4958-a1f8-556ac0b5d104] socks forwarding established
2025-07-31 20:26:37.605 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bdb8ed23-3c2c-4958-a1f8-556ac0b5d104] socks connection closed
2025-07-31 20:26:37.605 [info] [command][41c56497-62b6-4c5f-aba9-379a90eea401] Process exited with code 0
2025-07-31 20:26:37.605 [info] [command][41c56497-62b6-4c5f-aba9-379a90eea401] Socket close event received
2025-07-31 20:26:37.644 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53718 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:27:37.608 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:27:37.610 [info] [command][3545dc5e-5aea-49cd-9fbf-e0d754087aa3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3545dc5e-5aea-49cd-9fbf-e0d754087aa3""}
2025-07-31 20:27:37.611 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][69018b60-829f-44fe-85a5-a4f47937dd43] received connection request
2025-07-31 20:27:37.611 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:27:37.661 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69018b60-829f-44fe-85a5-a4f47937dd43] socks forwarding established
2025-07-31 20:27:37.705 [info] [command][3545dc5e-5aea-49cd-9fbf-e0d754087aa3] Process exited with code 0
2025-07-31 20:27:37.706 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69018b60-829f-44fe-85a5-a4f47937dd43] socks connection closed
2025-07-31 20:27:37.706 [info] [command][3545dc5e-5aea-49cd-9fbf-e0d754087aa3] Socket close event received
2025-07-31 20:27:37.745 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53755 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:28:37.711 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:28:37.713 [info] [command][cee17ab6-39d0-4210-861f-79729de31f38] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cee17ab6-39d0-4210-861f-79729de31f38""}
2025-07-31 20:28:37.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b9c9afc0-b1ce-47e1-8761-d42b67cfa57e] received connection request
2025-07-31 20:28:37.715 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:28:37.759 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9c9afc0-b1ce-47e1-8761-d42b67cfa57e] socks forwarding established
2025-07-31 20:28:37.804 [info] [command][cee17ab6-39d0-4210-861f-79729de31f38] Process exited with code 0
2025-07-31 20:28:37.804 [info] [command][cee17ab6-39d0-4210-861f-79729de31f38] Socket close event received
2025-07-31 20:28:37.804 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9c9afc0-b1ce-47e1-8761-d42b67cfa57e] socks connection closed
2025-07-31 20:28:37.843 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53788 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:29:37.808 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:29:37.810 [info] [command][cbd81d94-733c-4c5f-b796-10c8c739f10c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cbd81d94-733c-4c5f-b796-10c8c739f10c""}
2025-07-31 20:29:37.810 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6c8fde26-f847-4c25-8584-d356c289d421] received connection request
2025-07-31 20:29:37.811 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:29:37.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c8fde26-f847-4c25-8584-d356c289d421] socks forwarding established
2025-07-31 20:29:37.899 [info] [command][cbd81d94-733c-4c5f-b796-10c8c739f10c] Process exited with code 0
2025-07-31 20:29:37.900 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c8fde26-f847-4c25-8584-d356c289d421] socks connection closed
2025-07-31 20:29:37.900 [info] [command][cbd81d94-733c-4c5f-b796-10c8c739f10c] Socket close event received
2025-07-31 20:29:37.941 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53825 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:30:37.903 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:30:37.906 [info] [command][dad2520f-9bcd-4a88-807e-20c7cbaeb8ce] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dad2520f-9bcd-4a88-807e-20c7cbaeb8ce""}
2025-07-31 20:30:37.906 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7783fc21-cba4-47e3-8e2f-b84d0dfc9a15] received connection request
2025-07-31 20:30:37.907 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:30:38.025 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7783fc21-cba4-47e3-8e2f-b84d0dfc9a15] socks forwarding established
2025-07-31 20:30:38.071 [info] [command][dad2520f-9bcd-4a88-807e-20c7cbaeb8ce] Process exited with code 0
2025-07-31 20:30:38.071 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7783fc21-cba4-47e3-8e2f-b84d0dfc9a15] socks connection closed
2025-07-31 20:30:38.071 [info] [command][dad2520f-9bcd-4a88-807e-20c7cbaeb8ce] Socket close event received
2025-07-31 20:30:38.112 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53843 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:31:38.072 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:31:38.074 [info] [command][62ec089e-e371-47e1-acda-6c8b644feeda] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""62ec089e-e371-47e1-acda-6c8b644feeda""}
2025-07-31 20:31:38.075 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ef4ad9c1-6cad-4db7-8fef-3fd4dc87ae39] received connection request
2025-07-31 20:31:38.075 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:31:38.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ef4ad9c1-6cad-4db7-8fef-3fd4dc87ae39] socks forwarding established
2025-07-31 20:31:38.163 [info] [command][62ec089e-e371-47e1-acda-6c8b644feeda] Process exited with code 0
2025-07-31 20:31:38.163 [info] [command][62ec089e-e371-47e1-acda-6c8b644feeda] Socket close event received
2025-07-31 20:31:38.165 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ef4ad9c1-6cad-4db7-8fef-3fd4dc87ae39] socks connection closed
2025-07-31 20:31:38.204 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53882 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:32:38.165 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:32:38.168 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f4e31df0-7974-4ef7-91f6-c9df3dafb762] received connection request
2025-07-31 20:32:38.168 [info] [command][56404101-583b-482b-b1d5-e9dc758e1e29] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""56404101-583b-482b-b1d5-e9dc758e1e29""}
2025-07-31 20:32:38.168 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:32:38.209 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4e31df0-7974-4ef7-91f6-c9df3dafb762] socks forwarding established
2025-07-31 20:32:38.254 [info] [command][56404101-583b-482b-b1d5-e9dc758e1e29] Process exited with code 0
2025-07-31 20:32:38.254 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4e31df0-7974-4ef7-91f6-c9df3dafb762] socks connection closed
2025-07-31 20:32:38.254 [info] [command][56404101-583b-482b-b1d5-e9dc758e1e29] Socket close event received
2025-07-31 20:32:38.294 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53920 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:33:38.260 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:33:38.264 [info] [command][2c7a84e1-3dee-40dc-b4c2-3871dcea4d3d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2c7a84e1-3dee-40dc-b4c2-3871dcea4d3d""}
2025-07-31 20:33:38.265 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ae02bcc9-64cc-4789-a061-a87623dd9a3b] received connection request
2025-07-31 20:33:38.266 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:33:38.312 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ae02bcc9-64cc-4789-a061-a87623dd9a3b] socks forwarding established
2025-07-31 20:33:38.361 [info] [command][2c7a84e1-3dee-40dc-b4c2-3871dcea4d3d] Process exited with code 0
2025-07-31 20:33:38.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ae02bcc9-64cc-4789-a061-a87623dd9a3b] socks connection closed
2025-07-31 20:33:38.362 [info] [command][2c7a84e1-3dee-40dc-b4c2-3871dcea4d3d] Socket close event received
2025-07-31 20:33:38.401 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53953 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:34:38.363 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:34:38.365 [info] [command][ded91ccf-a589-464f-a9d7-416421368bd5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ded91ccf-a589-464f-a9d7-416421368bd5""}
2025-07-31 20:34:38.365 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][945136ff-63f1-4587-999e-dae3751d4b8c] received connection request
2025-07-31 20:34:38.365 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:34:38.365 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:34:38.406 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][945136ff-63f1-4587-999e-dae3751d4b8c] socks forwarding established
2025-07-31 20:34:38.452 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][945136ff-63f1-4587-999e-dae3751d4b8c] socks connection closed
2025-07-31 20:34:38.452 [info] [command][ded91ccf-a589-464f-a9d7-416421368bd5] Process exited with code 0
2025-07-31 20:34:38.452 [info] [command][ded91ccf-a589-464f-a9d7-416421368bd5] Socket close event received
2025-07-31 20:34:38.492 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54002 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:35:38.455 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:35:38.458 [info] [command][908518d2-d9a4-4fbd-984f-edcdbb8226ce] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""908518d2-d9a4-4fbd-984f-edcdbb8226ce""}
2025-07-31 20:35:38.458 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][922e24a6-0ca5-4be3-99da-b6ed8055b3ef] received connection request
2025-07-31 20:35:38.459 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:35:38.504 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][922e24a6-0ca5-4be3-99da-b6ed8055b3ef] socks forwarding established
2025-07-31 20:35:38.549 [info] [command][908518d2-d9a4-4fbd-984f-edcdbb8226ce] Process exited with code 0
2025-07-31 20:35:38.549 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][922e24a6-0ca5-4be3-99da-b6ed8055b3ef] socks connection closed
2025-07-31 20:35:38.549 [info] [command][908518d2-d9a4-4fbd-984f-edcdbb8226ce] Socket close event received
2025-07-31 20:35:38.591 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54045 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:36:38.555 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:36:38.557 [info] [command][538891d3-3143-4598-b74b-8ddb8ae2ff81] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""538891d3-3143-4598-b74b-8ddb8ae2ff81""}
2025-07-31 20:36:38.558 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3f5456d-7181-4ce9-8842-7de458a70d67] received connection request
2025-07-31 20:36:38.559 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:36:38.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3f5456d-7181-4ce9-8842-7de458a70d67] socks forwarding established
2025-07-31 20:36:38.649 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3f5456d-7181-4ce9-8842-7de458a70d67] socks connection closed
2025-07-31 20:36:38.649 [info] [command][538891d3-3143-4598-b74b-8ddb8ae2ff81] Process exited with code 0
2025-07-31 20:36:38.649 [info] [command][538891d3-3143-4598-b74b-8ddb8ae2ff81] Socket close event received
2025-07-31 20:36:38.690 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54089 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:37:38.649 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:37:38.652 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fc04d632-5ca8-4bf8-a483-f11231538fef] received connection request
2025-07-31 20:37:38.652 [info] [command][a8a4a49f-2f30-4a30-8791-02c4af9d76a7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a8a4a49f-2f30-4a30-8791-02c4af9d76a7""}
2025-07-31 20:37:38.652 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:37:38.693 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fc04d632-5ca8-4bf8-a483-f11231538fef] socks forwarding established
2025-07-31 20:37:38.745 [info] [command][a8a4a49f-2f30-4a30-8791-02c4af9d76a7] Process exited with code 0
2025-07-31 20:37:38.745 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fc04d632-5ca8-4bf8-a483-f11231538fef] socks connection closed
2025-07-31 20:37:38.745 [info] [command][a8a4a49f-2f30-4a30-8791-02c4af9d76a7] Socket close event received
2025-07-31 20:37:38.785 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54153 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:38:38.746 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:38:38.747 [info] [command][03a43c6a-9a31-4ed8-ad2d-125b9abd7c8c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""03a43c6a-9a31-4ed8-ad2d-125b9abd7c8c""}
2025-07-31 20:38:38.748 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3197ed11-ee2d-4b4c-ba56-b87cfd6e9640] received connection request
2025-07-31 20:38:38.748 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:38:38.798 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3197ed11-ee2d-4b4c-ba56-b87cfd6e9640] socks forwarding established
2025-07-31 20:38:38.846 [info] [command][03a43c6a-9a31-4ed8-ad2d-125b9abd7c8c] Process exited with code 0
2025-07-31 20:38:38.846 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3197ed11-ee2d-4b4c-ba56-b87cfd6e9640] socks connection closed
2025-07-31 20:38:38.846 [info] [command][03a43c6a-9a31-4ed8-ad2d-125b9abd7c8c] Socket close event received
2025-07-31 20:38:38.886 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54179 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:39:38.851 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:39:38.852 [info] [command][699939eb-c496-4903-a59c-6b13205628ff] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""699939eb-c496-4903-a59c-6b13205628ff""}
2025-07-31 20:39:38.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][efc7dae1-5b4d-421a-bad0-02ab3c271413] received connection request
2025-07-31 20:39:38.853 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:39:38.966 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efc7dae1-5b4d-421a-bad0-02ab3c271413] socks forwarding established
2025-07-31 20:39:39.012 [info] [command][699939eb-c496-4903-a59c-6b13205628ff] Process exited with code 0
2025-07-31 20:39:39.013 [info] [command][699939eb-c496-4903-a59c-6b13205628ff] Socket close event received
2025-07-31 20:39:39.013 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efc7dae1-5b4d-421a-bad0-02ab3c271413] socks connection closed
2025-07-31 20:39:39.057 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54211 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:40:39.013 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:40:39.014 [info] [command][c2e65099-78b6-43c0-8cb9-f36d09b53583] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c2e65099-78b6-43c0-8cb9-f36d09b53583""}
2025-07-31 20:40:39.014 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ceb1b2f1-9e68-40fd-a930-aace7857540d] received connection request
2025-07-31 20:40:39.014 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:40:39.055 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ceb1b2f1-9e68-40fd-a930-aace7857540d] socks forwarding established
2025-07-31 20:40:39.099 [info] [command][c2e65099-78b6-43c0-8cb9-f36d09b53583] Process exited with code 0
2025-07-31 20:40:39.100 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ceb1b2f1-9e68-40fd-a930-aace7857540d] socks connection closed
2025-07-31 20:40:39.100 [info] [command][c2e65099-78b6-43c0-8cb9-f36d09b53583] Socket close event received
2025-07-31 20:40:39.140 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54243 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:41:39.102 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:41:39.104 [info] [command][15fe13ac-2f18-4ee7-808f-2c5564c29d06] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""15fe13ac-2f18-4ee7-808f-2c5564c29d06""}
2025-07-31 20:41:39.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][87d6d67d-5125-4d90-a047-d68932907c61] received connection request
2025-07-31 20:41:39.106 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:41:39.148 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87d6d67d-5125-4d90-a047-d68932907c61] socks forwarding established
2025-07-31 20:41:39.195 [info] [command][15fe13ac-2f18-4ee7-808f-2c5564c29d06] Process exited with code 0
2025-07-31 20:41:39.195 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87d6d67d-5125-4d90-a047-d68932907c61] socks connection closed
2025-07-31 20:41:39.195 [info] [command][15fe13ac-2f18-4ee7-808f-2c5564c29d06] Socket close event received
2025-07-31 20:41:39.235 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54288 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:42:39.199 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:42:39.201 [info] [command][83f5725a-e787-4b64-bd29-370ba309fbce] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""83f5725a-e787-4b64-bd29-370ba309fbce""}
2025-07-31 20:42:39.201 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0fa6641a-369a-451c-8ffd-ef885b4a6693] received connection request
2025-07-31 20:42:39.201 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:42:39.201 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:42:39.243 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0fa6641a-369a-451c-8ffd-ef885b4a6693] socks forwarding established
2025-07-31 20:42:39.288 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0fa6641a-369a-451c-8ffd-ef885b4a6693] socks connection closed
2025-07-31 20:42:39.288 [info] [command][83f5725a-e787-4b64-bd29-370ba309fbce] Process exited with code 0
2025-07-31 20:42:39.288 [info] [command][83f5725a-e787-4b64-bd29-370ba309fbce] Socket close event received
2025-07-31 20:42:39.327 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54339 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:43:39.291 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:43:39.293 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][47651b40-4b2d-43cf-b936-243c6190ad7d] received connection request
2025-07-31 20:43:39.294 [info] [command][5244b517-9c0b-4e6d-8139-98d27a22e8d3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5244b517-9c0b-4e6d-8139-98d27a22e8d3""}
2025-07-31 20:43:39.294 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:43:39.335 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47651b40-4b2d-43cf-b936-243c6190ad7d] socks forwarding established
2025-07-31 20:43:39.381 [info] [command][5244b517-9c0b-4e6d-8139-98d27a22e8d3] Process exited with code 0
2025-07-31 20:43:39.381 [info] [command][5244b517-9c0b-4e6d-8139-98d27a22e8d3] Socket close event received
2025-07-31 20:43:39.382 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47651b40-4b2d-43cf-b936-243c6190ad7d] socks connection closed
2025-07-31 20:43:39.427 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54354 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:44:39.385 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:44:39.388 [info] [command][f0546fc1-424b-4d73-90b4-9d5f00a6a5b4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f0546fc1-424b-4d73-90b4-9d5f00a6a5b4""}
2025-07-31 20:44:39.391 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][72a5f488-00ad-45e2-8878-8b0c0a64fcdd] received connection request
2025-07-31 20:44:39.392 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:44:39.435 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][72a5f488-00ad-45e2-8878-8b0c0a64fcdd] socks forwarding established
2025-07-31 20:44:39.479 [info] [command][f0546fc1-424b-4d73-90b4-9d5f00a6a5b4] Process exited with code 0
2025-07-31 20:44:39.479 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][72a5f488-00ad-45e2-8878-8b0c0a64fcdd] socks connection closed
2025-07-31 20:44:39.479 [info] [command][f0546fc1-424b-4d73-90b4-9d5f00a6a5b4] Socket close event received
2025-07-31 20:44:39.524 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54372 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:45:39.481 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:45:39.481 [info] [command][9d240ffc-a2d8-4ff7-8f27-a3a20356ec71] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9d240ffc-a2d8-4ff7-8f27-a3a20356ec71""}
2025-07-31 20:45:39.482 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1f80c70b-2ac2-41e4-81f2-68f3d3abb9ad] received connection request
2025-07-31 20:45:39.482 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:45:39.482 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:45:39.524 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f80c70b-2ac2-41e4-81f2-68f3d3abb9ad] socks forwarding established
2025-07-31 20:45:39.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f80c70b-2ac2-41e4-81f2-68f3d3abb9ad] socks connection closed
2025-07-31 20:45:39.570 [info] [command][9d240ffc-a2d8-4ff7-8f27-a3a20356ec71] Process exited with code 0
2025-07-31 20:45:39.570 [info] [command][9d240ffc-a2d8-4ff7-8f27-a3a20356ec71] Socket close event received
2025-07-31 20:45:39.611 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54394 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:46:39.575 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:46:39.577 [info] [command][b0699713-c3fb-45b0-bdb1-308e142e2158] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b0699713-c3fb-45b0-bdb1-308e142e2158""}
2025-07-31 20:46:39.577 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][99a85515-784a-45b5-8b0a-e3c409a505dc] received connection request
2025-07-31 20:46:39.577 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:46:39.707 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][99a85515-784a-45b5-8b0a-e3c409a505dc] socks forwarding established
2025-07-31 20:46:39.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][99a85515-784a-45b5-8b0a-e3c409a505dc] socks connection closed
2025-07-31 20:46:39.754 [info] [command][b0699713-c3fb-45b0-bdb1-308e142e2158] Process exited with code 0
2025-07-31 20:46:39.754 [info] [command][b0699713-c3fb-45b0-bdb1-308e142e2158] Socket close event received
2025-07-31 20:46:39.797 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54438 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:47:39.755 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:47:39.757 [info] [command][dfd816b1-223f-4c94-b624-46449fd60554] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dfd816b1-223f-4c94-b624-46449fd60554""}
2025-07-31 20:47:39.758 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a2f0fbea-adbc-46ea-8e15-150df0111168] received connection request
2025-07-31 20:47:39.758 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:47:39.807 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a2f0fbea-adbc-46ea-8e15-150df0111168] socks forwarding established
2025-07-31 20:47:39.857 [info] [command][dfd816b1-223f-4c94-b624-46449fd60554] Process exited with code 0
2025-07-31 20:47:39.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a2f0fbea-adbc-46ea-8e15-150df0111168] socks connection closed
2025-07-31 20:47:39.857 [info] [command][dfd816b1-223f-4c94-b624-46449fd60554] Socket close event received
2025-07-31 20:47:39.959 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54476 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:48:39.862 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:48:39.863 [info] [command][4ccd4f76-5ea2-4f3a-9d1c-0dfd12164ade] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4ccd4f76-5ea2-4f3a-9d1c-0dfd12164ade""}
2025-07-31 20:48:39.863 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][27c46ebd-a626-42f2-9fc9-1e81970856c0] received connection request
2025-07-31 20:48:39.864 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:48:39.904 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27c46ebd-a626-42f2-9fc9-1e81970856c0] socks forwarding established
2025-07-31 20:48:39.964 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27c46ebd-a626-42f2-9fc9-1e81970856c0] socks connection closed
2025-07-31 20:48:39.964 [info] [command][4ccd4f76-5ea2-4f3a-9d1c-0dfd12164ade] Process exited with code 0
2025-07-31 20:48:39.964 [info] [command][4ccd4f76-5ea2-4f3a-9d1c-0dfd12164ade] Socket close event received
2025-07-31 20:48:40.009 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54504 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:49:39.972 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:49:39.973 [info] [command][8f403af6-0461-4cfa-80b5-43a947394350] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8f403af6-0461-4cfa-80b5-43a947394350""}
2025-07-31 20:49:39.974 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][22a06b37-71b2-4e60-b8ec-3c91573ff73a] received connection request
2025-07-31 20:49:39.974 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:49:40.015 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22a06b37-71b2-4e60-b8ec-3c91573ff73a] socks forwarding established
2025-07-31 20:49:40.060 [info] [command][8f403af6-0461-4cfa-80b5-43a947394350] Process exited with code 0
2025-07-31 20:49:40.060 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22a06b37-71b2-4e60-b8ec-3c91573ff73a] socks connection closed
2025-07-31 20:49:40.060 [info] [command][8f403af6-0461-4cfa-80b5-43a947394350] Socket close event received
2025-07-31 20:49:40.099 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54529 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:50:40.065 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:50:40.067 [info] [command][2b41584c-29e8-4ca9-b3f4-13dedaf643c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2b41584c-29e8-4ca9-b3f4-13dedaf643c7""}
2025-07-31 20:50:40.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][13e40ed5-a5f6-4b43-b209-a985e7b139fa] received connection request
2025-07-31 20:50:40.067 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:50:40.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13e40ed5-a5f6-4b43-b209-a985e7b139fa] socks forwarding established
2025-07-31 20:50:40.219 [info] [command][2b41584c-29e8-4ca9-b3f4-13dedaf643c7] Process exited with code 0
2025-07-31 20:50:40.219 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13e40ed5-a5f6-4b43-b209-a985e7b139fa] socks connection closed
2025-07-31 20:50:40.219 [info] [command][2b41584c-29e8-4ca9-b3f4-13dedaf643c7] Socket close event received
2025-07-31 20:50:40.261 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54573 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:51:40.223 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:51:40.225 [info] [command][3faa21d6-c6eb-4440-98d1-dd03b773033b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3faa21d6-c6eb-4440-98d1-dd03b773033b""}
2025-07-31 20:51:40.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b5de86cb-214c-4f0d-9c43-2762f8760ca5] received connection request
2025-07-31 20:51:40.225 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:51:40.225 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:51:40.264 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b5de86cb-214c-4f0d-9c43-2762f8760ca5] socks forwarding established
2025-07-31 20:51:40.306 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b5de86cb-214c-4f0d-9c43-2762f8760ca5] socks connection closed
2025-07-31 20:51:40.306 [info] [command][3faa21d6-c6eb-4440-98d1-dd03b773033b] Process exited with code 0
2025-07-31 20:51:40.306 [info] [command][3faa21d6-c6eb-4440-98d1-dd03b773033b] Socket close event received
2025-07-31 20:51:40.349 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54619 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:52:40.306 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:52:40.308 [info] [command][3e63f341-199e-4bb7-aada-49c790cbee08] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3e63f341-199e-4bb7-aada-49c790cbee08""}
2025-07-31 20:52:40.308 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3fa45e68-49df-44bd-b002-488859acb3d8] received connection request
2025-07-31 20:52:40.309 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:52:40.309 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:52:40.349 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3fa45e68-49df-44bd-b002-488859acb3d8] socks forwarding established
2025-07-31 20:52:40.397 [info] [command][3e63f341-199e-4bb7-aada-49c790cbee08] Process exited with code 0
2025-07-31 20:52:40.398 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3fa45e68-49df-44bd-b002-488859acb3d8] socks connection closed
2025-07-31 20:52:40.398 [info] [command][3e63f341-199e-4bb7-aada-49c790cbee08] Socket close event received
2025-07-31 20:52:40.439 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54652 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:53:40.403 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:53:40.405 [info] [command][0c03c8dc-d022-41b1-99fd-0f11695c7f1c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0c03c8dc-d022-41b1-99fd-0f11695c7f1c""}
2025-07-31 20:53:40.406 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][607a565b-ecc7-4052-b68b-3c175d9ac9cb] received connection request
2025-07-31 20:53:40.406 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:53:40.454 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][607a565b-ecc7-4052-b68b-3c175d9ac9cb] socks forwarding established
2025-07-31 20:53:40.499 [info] [command][0c03c8dc-d022-41b1-99fd-0f11695c7f1c] Process exited with code 0
2025-07-31 20:53:40.500 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][607a565b-ecc7-4052-b68b-3c175d9ac9cb] socks connection closed
2025-07-31 20:53:40.500 [info] [command][0c03c8dc-d022-41b1-99fd-0f11695c7f1c] Socket close event received
2025-07-31 20:53:40.539 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54700 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:54:40.505 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:54:40.507 [info] [command][43b7ea51-074e-4992-ac1a-c3a4bd5fe0ea] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""43b7ea51-074e-4992-ac1a-c3a4bd5fe0ea""}
2025-07-31 20:54:40.507 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d0d817a2-b34a-4e63-9cb3-40813ba69baf] received connection request
2025-07-31 20:54:40.508 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:54:40.606 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0d817a2-b34a-4e63-9cb3-40813ba69baf] socks forwarding established
2025-07-31 20:54:40.651 [info] [command][43b7ea51-074e-4992-ac1a-c3a4bd5fe0ea] Process exited with code 0
2025-07-31 20:54:40.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0d817a2-b34a-4e63-9cb3-40813ba69baf] socks connection closed
2025-07-31 20:54:40.651 [info] [command][43b7ea51-074e-4992-ac1a-c3a4bd5fe0ea] Socket close event received
2025-07-31 20:54:40.695 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54739 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:55:40.656 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:55:40.658 [info] [command][ff9d6ef6-e9b2-4872-b4d5-ba1d33e9af9c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ff9d6ef6-e9b2-4872-b4d5-ba1d33e9af9c""}
2025-07-31 20:55:40.659 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][17bb57c1-6b2c-4b88-a1bd-1400184ee461] received connection request
2025-07-31 20:55:40.659 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:55:40.704 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][17bb57c1-6b2c-4b88-a1bd-1400184ee461] socks forwarding established
2025-07-31 20:55:40.754 [info] [command][ff9d6ef6-e9b2-4872-b4d5-ba1d33e9af9c] Process exited with code 0
2025-07-31 20:55:40.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][17bb57c1-6b2c-4b88-a1bd-1400184ee461] socks connection closed
2025-07-31 20:55:40.754 [info] [command][ff9d6ef6-e9b2-4872-b4d5-ba1d33e9af9c] Socket close event received
2025-07-31 20:55:40.795 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54760 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:56:40.755 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:56:40.758 [info] [command][6cab3118-4a3b-4e0b-abce-8e1b1c871c7f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6cab3118-4a3b-4e0b-abce-8e1b1c871c7f""}
2025-07-31 20:56:40.759 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][87c95112-f23f-4a1c-8432-a0f6701bd206] received connection request
2025-07-31 20:56:40.759 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:56:40.806 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87c95112-f23f-4a1c-8432-a0f6701bd206] socks forwarding established
2025-07-31 20:56:40.852 [info] [command][6cab3118-4a3b-4e0b-abce-8e1b1c871c7f] Process exited with code 0
2025-07-31 20:56:40.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87c95112-f23f-4a1c-8432-a0f6701bd206] socks connection closed
2025-07-31 20:56:40.853 [info] [command][6cab3118-4a3b-4e0b-abce-8e1b1c871c7f] Socket close event received
2025-07-31 20:56:40.893 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54813 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:57:40.858 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:57:40.859 [info] [command][8e37a501-a6b0-4978-85bd-c6c10e255de2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8e37a501-a6b0-4978-85bd-c6c10e255de2""}
2025-07-31 20:57:40.859 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e1c22a94-cfe0-4f29-bba5-d39710796bc2] received connection request
2025-07-31 20:57:40.859 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 20:57:40.860 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:57:40.905 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e1c22a94-cfe0-4f29-bba5-d39710796bc2] socks forwarding established
2025-07-31 20:57:40.951 [info] [command][8e37a501-a6b0-4978-85bd-c6c10e255de2] Process exited with code 0
2025-07-31 20:57:40.951 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e1c22a94-cfe0-4f29-bba5-d39710796bc2] socks connection closed
2025-07-31 20:57:40.951 [info] [command][8e37a501-a6b0-4978-85bd-c6c10e255de2] Socket close event received
2025-07-31 20:57:40.993 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54845 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:58:40.956 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:58:40.957 [info] [command][18249345-bb09-46fe-aaca-5c9575db890e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""18249345-bb09-46fe-aaca-5c9575db890e""}
2025-07-31 20:58:40.957 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0c1e23f5-9180-438f-87aa-8edf7e30337d] received connection request
2025-07-31 20:58:40.958 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:58:41.000 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c1e23f5-9180-438f-87aa-8edf7e30337d] socks forwarding established
2025-07-31 20:58:41.043 [info] [command][18249345-bb09-46fe-aaca-5c9575db890e] Process exited with code 0
2025-07-31 20:58:41.043 [info] [command][18249345-bb09-46fe-aaca-5c9575db890e] Socket close event received
2025-07-31 20:58:41.044 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c1e23f5-9180-438f-87aa-8edf7e30337d] socks connection closed
2025-07-31 20:58:41.084 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54864 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 20:59:41.047 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 20:59:41.049 [info] [command][0cd8ba55-7ef8-4353-80ed-4120d768c907] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0cd8ba55-7ef8-4353-80ed-4120d768c907""}
2025-07-31 20:59:41.050 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][446939c4-5763-4008-bb7d-4ce4e9aedc60] received connection request
2025-07-31 20:59:41.050 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 20:59:41.091 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][446939c4-5763-4008-bb7d-4ce4e9aedc60] socks forwarding established
2025-07-31 20:59:41.137 [info] [command][0cd8ba55-7ef8-4353-80ed-4120d768c907] Process exited with code 0
2025-07-31 20:59:41.137 [info] [command][0cd8ba55-7ef8-4353-80ed-4120d768c907] Socket close event received
2025-07-31 20:59:41.138 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][446939c4-5763-4008-bb7d-4ce4e9aedc60] socks connection closed
2025-07-31 20:59:41.183 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54893 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:00:41.141 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:00:41.143 [info] [command][3a09b4bb-e498-4eb8-9dbb-d533c34a714a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3a09b4bb-e498-4eb8-9dbb-d533c34a714a""}
2025-07-31 21:00:41.144 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a92e65bb-446b-4cc1-9243-183fc9b9985d] received connection request
2025-07-31 21:00:41.145 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 21:00:41.145 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:00:41.220 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a92e65bb-446b-4cc1-9243-183fc9b9985d] socks forwarding established
2025-07-31 21:00:41.268 [info] [command][3a09b4bb-e498-4eb8-9dbb-d533c34a714a] Process exited with code 0
2025-07-31 21:00:41.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a92e65bb-446b-4cc1-9243-183fc9b9985d] socks connection closed
2025-07-31 21:00:41.269 [info] [command][3a09b4bb-e498-4eb8-9dbb-d533c34a714a] Socket close event received
2025-07-31 21:00:41.312 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54911 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:01:41.268 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:01:41.269 [info] [command][07e1d27a-4126-470a-bd1a-c8358f512388] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""07e1d27a-4126-470a-bd1a-c8358f512388""}
2025-07-31 21:01:41.269 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f989a1f9-26b0-4b57-b7ad-b012df898eed] received connection request
2025-07-31 21:01:41.269 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:01:41.308 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f989a1f9-26b0-4b57-b7ad-b012df898eed] socks forwarding established
2025-07-31 21:01:41.360 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f989a1f9-26b0-4b57-b7ad-b012df898eed] socks connection closed
2025-07-31 21:01:41.360 [info] [command][07e1d27a-4126-470a-bd1a-c8358f512388] Process exited with code 0
2025-07-31 21:01:41.360 [info] [command][07e1d27a-4126-470a-bd1a-c8358f512388] Socket close event received
2025-07-31 21:01:41.399 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54943 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:02:41.363 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:02:41.365 [info] [command][8056517a-37a4-4c17-ad00-56300ce1e26d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8056517a-37a4-4c17-ad00-56300ce1e26d""}
2025-07-31 21:02:41.366 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][476f4379-4921-4256-8a9e-c7079ea3a036] received connection request
2025-07-31 21:02:41.367 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:02:41.409 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476f4379-4921-4256-8a9e-c7079ea3a036] socks forwarding established
2025-07-31 21:02:41.458 [info] [command][8056517a-37a4-4c17-ad00-56300ce1e26d] Process exited with code 0
2025-07-31 21:02:41.458 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476f4379-4921-4256-8a9e-c7079ea3a036] socks connection closed
2025-07-31 21:02:41.458 [info] [command][8056517a-37a4-4c17-ad00-56300ce1e26d] Socket close event received
2025-07-31 21:02:41.516 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54980 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:03:41.462 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:03:41.463 [info] [command][d8303900-099c-482c-8b75-948f0cf4cea7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d8303900-099c-482c-8b75-948f0cf4cea7""}
2025-07-31 21:03:41.464 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][76d5e0cc-3500-49ba-a1a2-62e8525ae97a] received connection request
2025-07-31 21:03:41.465 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:03:41.549 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76d5e0cc-3500-49ba-a1a2-62e8525ae97a] socks forwarding established
2025-07-31 21:03:41.597 [info] [command][d8303900-099c-482c-8b75-948f0cf4cea7] Process exited with code 0
2025-07-31 21:03:41.597 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76d5e0cc-3500-49ba-a1a2-62e8525ae97a] socks connection closed
2025-07-31 21:03:41.598 [info] [command][d8303900-099c-482c-8b75-948f0cf4cea7] Socket close event received
2025-07-31 21:03:41.641 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55012 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:04:41.603 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:04:41.605 [info] [command][1b90d05f-733f-4154-be7e-3927939478b4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1b90d05f-733f-4154-be7e-3927939478b4""}
2025-07-31 21:04:41.606 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5445e71e-57d9-42f7-924b-7e3c204fd8dc] received connection request
2025-07-31 21:04:41.606 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:04:41.648 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5445e71e-57d9-42f7-924b-7e3c204fd8dc] socks forwarding established
2025-07-31 21:04:41.692 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5445e71e-57d9-42f7-924b-7e3c204fd8dc] socks connection closed
2025-07-31 21:04:41.692 [info] [command][1b90d05f-733f-4154-be7e-3927939478b4] Process exited with code 0
2025-07-31 21:04:41.692 [info] [command][1b90d05f-733f-4154-be7e-3927939478b4] Socket close event received
2025-07-31 21:04:41.731 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55037 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:05:41.696 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:05:41.699 [info] [command][81cef1ce-d9f3-4ecf-997f-0b3c375e6b98] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""81cef1ce-d9f3-4ecf-997f-0b3c375e6b98""}
2025-07-31 21:05:41.700 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e311e198-ce50-4974-988c-ed7ba0d82d1a] received connection request
2025-07-31 21:05:41.701 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:05:41.745 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e311e198-ce50-4974-988c-ed7ba0d82d1a] socks forwarding established
2025-07-31 21:05:41.795 [info] [command][81cef1ce-d9f3-4ecf-997f-0b3c375e6b98] Process exited with code 0
2025-07-31 21:05:41.796 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e311e198-ce50-4974-988c-ed7ba0d82d1a] socks connection closed
2025-07-31 21:05:41.796 [info] [command][81cef1ce-d9f3-4ecf-997f-0b3c375e6b98] Socket close event received
2025-07-31 21:05:41.836 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55077 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:06:41.801 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:06:41.803 [info] [command][ccd3d6a3-45a7-4bc4-ac6e-999241253808] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ccd3d6a3-45a7-4bc4-ac6e-999241253808""}
2025-07-31 21:06:41.804 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][87041fe8-6a94-438e-a2a0-d6dd36bb0550] received connection request
2025-07-31 21:06:41.805 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:06:41.848 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87041fe8-6a94-438e-a2a0-d6dd36bb0550] socks forwarding established
2025-07-31 21:06:41.894 [info] [command][ccd3d6a3-45a7-4bc4-ac6e-999241253808] Process exited with code 0
2025-07-31 21:06:41.894 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87041fe8-6a94-438e-a2a0-d6dd36bb0550] socks connection closed
2025-07-31 21:06:41.895 [info] [command][ccd3d6a3-45a7-4bc4-ac6e-999241253808] Socket close event received
2025-07-31 21:06:41.934 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55120 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:07:41.898 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:07:41.899 [info] [command][92316b7d-5e80-4f3c-b239-a680a6b820fa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""92316b7d-5e80-4f3c-b239-a680a6b820fa""}
2025-07-31 21:07:41.899 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][03221749-9bf9-4c6b-a56c-c25165b3438a] received connection request
2025-07-31 21:07:41.899 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:07:41.943 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03221749-9bf9-4c6b-a56c-c25165b3438a] socks forwarding established
2025-07-31 21:07:41.988 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03221749-9bf9-4c6b-a56c-c25165b3438a] socks connection closed
2025-07-31 21:07:41.988 [info] [command][92316b7d-5e80-4f3c-b239-a680a6b820fa] Process exited with code 0
2025-07-31 21:07:41.988 [info] [command][92316b7d-5e80-4f3c-b239-a680a6b820fa] Socket close event received
2025-07-31 21:07:42.029 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55201 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:08:41.989 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:08:41.989 [info] [command][7fdc2f5a-4284-4215-82ba-5762b0c72151] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7fdc2f5a-4284-4215-82ba-5762b0c72151""}
2025-07-31 21:08:41.990 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][11044679-5344-4509-b742-548f0bdeca8d] received connection request
2025-07-31 21:08:41.990 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:08:42.031 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][11044679-5344-4509-b742-548f0bdeca8d] socks forwarding established
2025-07-31 21:08:42.074 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][11044679-5344-4509-b742-548f0bdeca8d] socks connection closed
2025-07-31 21:08:42.074 [info] [command][7fdc2f5a-4284-4215-82ba-5762b0c72151] Process exited with code 0
2025-07-31 21:08:42.074 [info] [command][7fdc2f5a-4284-4215-82ba-5762b0c72151] Socket close event received
2025-07-31 21:08:42.115 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55236 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:09:42.079 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:09:42.080 [info] [command][57953ef6-014f-499e-9367-8850975eab25] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""57953ef6-014f-499e-9367-8850975eab25""}
2025-07-31 21:09:42.080 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5bf0b9d8-7b50-4bfd-9cdb-66f2905d00eb] received connection request
2025-07-31 21:09:42.080 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:09:42.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5bf0b9d8-7b50-4bfd-9cdb-66f2905d00eb] socks forwarding established
2025-07-31 21:09:42.164 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5bf0b9d8-7b50-4bfd-9cdb-66f2905d00eb] socks connection closed
2025-07-31 21:09:42.164 [info] [command][57953ef6-014f-499e-9367-8850975eab25] Process exited with code 0
2025-07-31 21:09:42.164 [info] [command][57953ef6-014f-499e-9367-8850975eab25] Socket close event received
2025-07-31 21:09:42.205 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55288 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:10:42.168 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:10:42.169 [info] [command][a1b8bcc7-d50e-4374-8549-9eb34f424f39] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a1b8bcc7-d50e-4374-8549-9eb34f424f39""}
2025-07-31 21:10:42.169 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c4b4881c-3491-4a00-92b9-2cba329a45de] received connection request
2025-07-31 21:10:42.169 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:10:42.276 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4b4881c-3491-4a00-92b9-2cba329a45de] socks forwarding established
2025-07-31 21:10:42.329 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4b4881c-3491-4a00-92b9-2cba329a45de] socks connection closed
2025-07-31 21:10:42.330 [info] [command][a1b8bcc7-d50e-4374-8549-9eb34f424f39] Process exited with code 0
2025-07-31 21:10:42.330 [info] [command][a1b8bcc7-d50e-4374-8549-9eb34f424f39] Socket close event received
2025-07-31 21:10:42.369 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55343 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:11:42.330 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:11:42.332 [info] [command][fbb8ff7a-e715-4ac0-9c2d-b778bbbc8c01] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fbb8ff7a-e715-4ac0-9c2d-b778bbbc8c01""}
2025-07-31 21:11:42.333 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][86447f71-8e37-4693-b1fb-215df9de5223] received connection request
2025-07-31 21:11:42.333 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:11:42.376 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][86447f71-8e37-4693-b1fb-215df9de5223] socks forwarding established
2025-07-31 21:11:42.422 [info] [command][fbb8ff7a-e715-4ac0-9c2d-b778bbbc8c01] Process exited with code 0
2025-07-31 21:11:42.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][86447f71-8e37-4693-b1fb-215df9de5223] socks connection closed
2025-07-31 21:11:42.422 [info] [command][fbb8ff7a-e715-4ac0-9c2d-b778bbbc8c01] Socket close event received
2025-07-31 21:11:42.462 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55388 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:12:42.426 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:12:42.428 [info] [command][e3d6251c-323d-4f89-ba2d-2468aec56084] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e3d6251c-323d-4f89-ba2d-2468aec56084""}
2025-07-31 21:12:42.428 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cd166840-1afd-4036-a086-c57018d65700] received connection request
2025-07-31 21:12:42.430 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:12:42.584 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd166840-1afd-4036-a086-c57018d65700] socks forwarding established
2025-07-31 21:12:42.630 [info] [command][e3d6251c-323d-4f89-ba2d-2468aec56084] Process exited with code 0
2025-07-31 21:12:42.630 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd166840-1afd-4036-a086-c57018d65700] socks connection closed
2025-07-31 21:12:42.630 [info] [command][e3d6251c-323d-4f89-ba2d-2468aec56084] Socket close event received
2025-07-31 21:12:42.669 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55420 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:13:42.630 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:13:42.631 [info] [command][53b8d45d-8c38-4dbd-a884-287bb7fac3c1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""53b8d45d-8c38-4dbd-a884-287bb7fac3c1""}
2025-07-31 21:13:42.631 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][736f0402-f5ea-4354-a6b4-4ea4a7a8353a] received connection request
2025-07-31 21:13:42.631 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:13:42.673 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][736f0402-f5ea-4354-a6b4-4ea4a7a8353a] socks forwarding established
2025-07-31 21:13:42.718 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][736f0402-f5ea-4354-a6b4-4ea4a7a8353a] socks connection closed
2025-07-31 21:13:42.718 [info] [command][53b8d45d-8c38-4dbd-a884-287bb7fac3c1] Process exited with code 0
2025-07-31 21:13:42.718 [info] [command][53b8d45d-8c38-4dbd-a884-287bb7fac3c1] Socket close event received
2025-07-31 21:13:42.757 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55443 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:14:42.720 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:14:42.722 [info] [command][ffe58a38-3bb0-4259-9ebe-c1977664d61b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ffe58a38-3bb0-4259-9ebe-c1977664d61b""}
2025-07-31 21:14:42.722 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][90e1256b-738c-47c0-9bea-7d04c2a63bd6] received connection request
2025-07-31 21:14:42.723 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:14:42.779 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][90e1256b-738c-47c0-9bea-7d04c2a63bd6] socks forwarding established
2025-07-31 21:14:42.826 [info] [command][ffe58a38-3bb0-4259-9ebe-c1977664d61b] Process exited with code 0
2025-07-31 21:14:42.827 [info] [command][ffe58a38-3bb0-4259-9ebe-c1977664d61b] Socket close event received
2025-07-31 21:14:42.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][90e1256b-738c-47c0-9bea-7d04c2a63bd6] socks connection closed
2025-07-31 21:14:42.875 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55464 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:15:42.829 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:15:42.831 [info] [command][f85d0228-85e3-4cf7-b24d-212298bf91cd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f85d0228-85e3-4cf7-b24d-212298bf91cd""}
2025-07-31 21:15:42.832 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][471fb581-c9a5-4c07-a772-e67fa204fa65] received connection request
2025-07-31 21:15:42.833 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:15:42.876 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][471fb581-c9a5-4c07-a772-e67fa204fa65] socks forwarding established
2025-07-31 21:15:42.922 [info] [command][f85d0228-85e3-4cf7-b24d-212298bf91cd] Process exited with code 0
2025-07-31 21:15:42.922 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][471fb581-c9a5-4c07-a772-e67fa204fa65] socks connection closed
2025-07-31 21:15:42.922 [info] [command][f85d0228-85e3-4cf7-b24d-212298bf91cd] Socket close event received
2025-07-31 21:15:42.962 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55487 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:16:42.925 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:16:42.927 [info] [command][2d16bc43-00ff-4f6c-b01b-2820fdb9d610] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2d16bc43-00ff-4f6c-b01b-2820fdb9d610""}
2025-07-31 21:16:42.928 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9f5aa7d6-13ff-4b33-beb8-a5f04c9366aa] received connection request
2025-07-31 21:16:42.928 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:16:43.021 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9f5aa7d6-13ff-4b33-beb8-a5f04c9366aa] socks forwarding established
2025-07-31 21:16:43.065 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9f5aa7d6-13ff-4b33-beb8-a5f04c9366aa] socks connection closed
2025-07-31 21:16:43.065 [info] [command][2d16bc43-00ff-4f6c-b01b-2820fdb9d610] Process exited with code 0
2025-07-31 21:16:43.065 [info] [command][2d16bc43-00ff-4f6c-b01b-2820fdb9d610] Socket close event received
2025-07-31 21:16:43.185 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55528 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:17:43.068 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:17:43.070 [info] [command][78501f8d-0f1b-4c00-8190-0528d24fbdab] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""78501f8d-0f1b-4c00-8190-0528d24fbdab""}
2025-07-31 21:17:43.071 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e2ae7d14-dd89-47b7-ac44-7067831176b0] received connection request
2025-07-31 21:17:43.071 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:17:43.120 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e2ae7d14-dd89-47b7-ac44-7067831176b0] socks forwarding established
2025-07-31 21:17:43.170 [info] [command][78501f8d-0f1b-4c00-8190-0528d24fbdab] Process exited with code 0
2025-07-31 21:17:43.170 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e2ae7d14-dd89-47b7-ac44-7067831176b0] socks connection closed
2025-07-31 21:17:43.170 [info] [command][78501f8d-0f1b-4c00-8190-0528d24fbdab] Socket close event received
2025-07-31 21:17:43.210 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55563 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:18:43.174 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:18:43.176 [info] [command][23dc2dc4-b8b0-46aa-87ac-16b11a759bf9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""23dc2dc4-b8b0-46aa-87ac-16b11a759bf9""}
2025-07-31 21:18:43.177 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d5c7acd5-186d-47f8-bd5c-0326826bf709] received connection request
2025-07-31 21:18:43.178 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:18:43.222 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d5c7acd5-186d-47f8-bd5c-0326826bf709] socks forwarding established
2025-07-31 21:18:43.271 [info] [command][23dc2dc4-b8b0-46aa-87ac-16b11a759bf9] Process exited with code 0
2025-07-31 21:18:43.271 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d5c7acd5-186d-47f8-bd5c-0326826bf709] socks connection closed
2025-07-31 21:18:43.271 [info] [command][23dc2dc4-b8b0-46aa-87ac-16b11a759bf9] Socket close event received
2025-07-31 21:18:43.312 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55588 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:19:43.273 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:19:43.274 [info] [command][89a488ad-e558-483c-a839-f2b23790ce9e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""89a488ad-e558-483c-a839-f2b23790ce9e""}
2025-07-31 21:19:43.274 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][85682208-c863-445f-909b-621fc63a4192] received connection request
2025-07-31 21:19:43.275 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 21:19:43.275 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:19:43.317 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85682208-c863-445f-909b-621fc63a4192] socks forwarding established
2025-07-31 21:19:43.453 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85682208-c863-445f-909b-621fc63a4192] socks connection closed
2025-07-31 21:19:43.453 [info] [command][89a488ad-e558-483c-a839-f2b23790ce9e] Process exited with code 0
2025-07-31 21:19:43.453 [info] [command][89a488ad-e558-483c-a839-f2b23790ce9e] Socket close event received
2025-07-31 21:19:43.506 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55615 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:20:43.459 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:20:43.461 [info] [command][9bd79f58-afe4-4a37-ab7e-d2abe52041d5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9bd79f58-afe4-4a37-ab7e-d2abe52041d5""}
2025-07-31 21:20:43.461 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8838b956-fae6-4e64-a776-89435e8435f0] received connection request
2025-07-31 21:20:43.462 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:20:43.504 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8838b956-fae6-4e64-a776-89435e8435f0] socks forwarding established
2025-07-31 21:20:43.601 [info] [command][9bd79f58-afe4-4a37-ab7e-d2abe52041d5] Process exited with code 0
2025-07-31 21:20:43.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8838b956-fae6-4e64-a776-89435e8435f0] socks connection closed
2025-07-31 21:20:43.601 [info] [command][9bd79f58-afe4-4a37-ab7e-d2abe52041d5] Socket close event received
2025-07-31 21:20:43.641 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55655 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:21:43.602 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:21:43.603 [info] [command][e11da44f-690f-46af-95c9-c594872169c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e11da44f-690f-46af-95c9-c594872169c7""}
2025-07-31 21:21:43.603 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][71e54f64-d2fa-4511-b923-24ac10d0727d] received connection request
2025-07-31 21:21:43.603 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 21:21:43.603 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:21:43.643 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][71e54f64-d2fa-4511-b923-24ac10d0727d] socks forwarding established
2025-07-31 21:21:43.687 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][71e54f64-d2fa-4511-b923-24ac10d0727d] socks connection closed
2025-07-31 21:21:43.687 [info] [command][e11da44f-690f-46af-95c9-c594872169c7] Process exited with code 0
2025-07-31 21:21:43.687 [info] [command][e11da44f-690f-46af-95c9-c594872169c7] Socket close event received
2025-07-31 21:21:43.733 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55696 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:22:43.691 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:22:43.692 [info] [command][98692d42-13dd-42b8-bb61-aba78d71d618] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""98692d42-13dd-42b8-bb61-aba78d71d618""}
2025-07-31 21:22:43.693 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f5e1e952-8124-46ff-93fe-2f2a6385f439] received connection request
2025-07-31 21:22:43.693 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:22:43.736 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f5e1e952-8124-46ff-93fe-2f2a6385f439] socks forwarding established
2025-07-31 21:22:43.782 [info] [command][98692d42-13dd-42b8-bb61-aba78d71d618] Process exited with code 0
2025-07-31 21:22:43.782 [info] [command][98692d42-13dd-42b8-bb61-aba78d71d618] Socket close event received
2025-07-31 21:22:43.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f5e1e952-8124-46ff-93fe-2f2a6385f439] socks connection closed
2025-07-31 21:22:43.823 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55725 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:23:43.783 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:23:43.784 [info] [command][33f60d83-4c97-472e-8720-0b3c05123ecd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""33f60d83-4c97-472e-8720-0b3c05123ecd""}
2025-07-31 21:23:43.785 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fe983168-8cd8-4799-8861-ca18ddc6afe4] received connection request
2025-07-31 21:23:43.785 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:23:43.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fe983168-8cd8-4799-8861-ca18ddc6afe4] socks forwarding established
2025-07-31 21:23:43.874 [info] [command][33f60d83-4c97-472e-8720-0b3c05123ecd] Process exited with code 0
2025-07-31 21:23:43.874 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fe983168-8cd8-4799-8861-ca18ddc6afe4] socks connection closed
2025-07-31 21:23:43.874 [info] [command][33f60d83-4c97-472e-8720-0b3c05123ecd] Socket close event received
2025-07-31 21:23:43.915 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55742 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:24:43.879 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:24:43.885 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c7b72d67-7f35-4d56-8461-9ae36558fac7] received connection request
2025-07-31 21:24:43.886 [info] [command][3dccb25c-408d-4776-99e4-9a7526e12298] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3dccb25c-408d-4776-99e4-9a7526e12298""}
2025-07-31 21:24:43.886 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:24:43.928 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c7b72d67-7f35-4d56-8461-9ae36558fac7] socks forwarding established
2025-07-31 21:24:43.975 [info] [command][3dccb25c-408d-4776-99e4-9a7526e12298] Process exited with code 0
2025-07-31 21:24:43.975 [info] [command][3dccb25c-408d-4776-99e4-9a7526e12298] Socket close event received
2025-07-31 21:24:43.976 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c7b72d67-7f35-4d56-8461-9ae36558fac7] socks connection closed
2025-07-31 21:24:44.015 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55759 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:25:43.976 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:25:43.977 [info] [command][04b5db9d-95dc-4b3a-b48d-cd105c2201f4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""04b5db9d-95dc-4b3a-b48d-cd105c2201f4""}
2025-07-31 21:25:43.977 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][17077f7d-2deb-4677-981c-1ea8d22b04b1] received connection request
2025-07-31 21:25:43.977 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 21:25:43.977 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:25:44.020 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][17077f7d-2deb-4677-981c-1ea8d22b04b1] socks forwarding established
2025-07-31 21:25:44.096 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][17077f7d-2deb-4677-981c-1ea8d22b04b1] socks connection closed
2025-07-31 21:25:44.097 [info] [command][04b5db9d-95dc-4b3a-b48d-cd105c2201f4] Process exited with code 0
2025-07-31 21:25:44.097 [info] [command][04b5db9d-95dc-4b3a-b48d-cd105c2201f4] Socket close event received
2025-07-31 21:25:44.137 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55785 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:26:44.098 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:26:44.100 [info] [command][2913a252-c254-45c9-8e46-cbfdba0a2355] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2913a252-c254-45c9-8e46-cbfdba0a2355""}
2025-07-31 21:26:44.100 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][706d2d22-f1d3-4422-8dbe-dd3cb4a56f43] received connection request
2025-07-31 21:26:44.101 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:26:44.141 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][706d2d22-f1d3-4422-8dbe-dd3cb4a56f43] socks forwarding established
2025-07-31 21:26:44.186 [info] [command][2913a252-c254-45c9-8e46-cbfdba0a2355] Process exited with code 0
2025-07-31 21:26:44.186 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][706d2d22-f1d3-4422-8dbe-dd3cb4a56f43] socks connection closed
2025-07-31 21:26:44.187 [info] [command][2913a252-c254-45c9-8e46-cbfdba0a2355] Socket close event received
2025-07-31 21:26:44.225 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55830 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:27:44.190 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:27:44.191 [info] [command][b3af2c61-c6d4-469b-9f73-fd4f96419acc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b3af2c61-c6d4-469b-9f73-fd4f96419acc""}
2025-07-31 21:27:44.192 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][956d3246-96a2-4a36-8448-18d34884ca39] received connection request
2025-07-31 21:27:44.192 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:27:44.321 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][956d3246-96a2-4a36-8448-18d34884ca39] socks forwarding established
2025-07-31 21:27:44.373 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][956d3246-96a2-4a36-8448-18d34884ca39] socks connection closed
2025-07-31 21:27:44.374 [info] [command][b3af2c61-c6d4-469b-9f73-fd4f96419acc] Process exited with code 0
2025-07-31 21:27:44.374 [info] [command][b3af2c61-c6d4-469b-9f73-fd4f96419acc] Socket close event received
2025-07-31 21:27:44.414 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55868 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:28:44.376 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:28:44.378 [info] [command][9fe6fb44-8963-4f74-9c85-1b03b6e133b4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9fe6fb44-8963-4f74-9c85-1b03b6e133b4""}
2025-07-31 21:28:44.378 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][52739d3b-a412-43e7-86ba-872329d14ae3] received connection request
2025-07-31 21:28:44.379 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:28:44.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52739d3b-a412-43e7-86ba-872329d14ae3] socks forwarding established
2025-07-31 21:28:44.466 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52739d3b-a412-43e7-86ba-872329d14ae3] socks connection closed
2025-07-31 21:28:44.466 [info] [command][9fe6fb44-8963-4f74-9c85-1b03b6e133b4] Process exited with code 0
2025-07-31 21:28:44.466 [info] [command][9fe6fb44-8963-4f74-9c85-1b03b6e133b4] Socket close event received
2025-07-31 21:28:44.508 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55886 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:29:44.468 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:29:44.471 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a47e1ce8-9095-447b-8d7b-9077165522bc] received connection request
2025-07-31 21:29:44.471 [info] [command][91825896-f0ee-40f5-85cb-82d9216a7555] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""91825896-f0ee-40f5-85cb-82d9216a7555""}
2025-07-31 21:29:44.471 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:29:44.511 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a47e1ce8-9095-447b-8d7b-9077165522bc] socks forwarding established
2025-07-31 21:29:44.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a47e1ce8-9095-447b-8d7b-9077165522bc] socks connection closed
2025-07-31 21:29:44.570 [info] [command][91825896-f0ee-40f5-85cb-82d9216a7555] Process exited with code 0
2025-07-31 21:29:44.570 [info] [command][91825896-f0ee-40f5-85cb-82d9216a7555] Socket close event received
2025-07-31 21:29:44.613 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55923 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:30:44.575 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:30:44.577 [info] [command][3c07bc57-2c59-44a5-bfc6-3d518aa63f95] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3c07bc57-2c59-44a5-bfc6-3d518aa63f95""}
2025-07-31 21:30:44.577 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][909c5261-f7df-46e7-89a0-98696e2d834c] received connection request
2025-07-31 21:30:44.578 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:30:44.618 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][909c5261-f7df-46e7-89a0-98696e2d834c] socks forwarding established
2025-07-31 21:30:44.664 [info] [command][3c07bc57-2c59-44a5-bfc6-3d518aa63f95] Process exited with code 0
2025-07-31 21:30:44.664 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][909c5261-f7df-46e7-89a0-98696e2d834c] socks connection closed
2025-07-31 21:30:44.664 [info] [command][3c07bc57-2c59-44a5-bfc6-3d518aa63f95] Socket close event received
2025-07-31 21:30:44.705 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55950 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:31:44.669 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:31:44.671 [info] [command][645c2422-1cc3-45ef-b403-1fa392bfb8fa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""645c2422-1cc3-45ef-b403-1fa392bfb8fa""}
2025-07-31 21:31:44.671 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0106162d-2a7d-46e9-bd1e-115b161f9c3e] received connection request
2025-07-31 21:31:44.671 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:31:44.711 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0106162d-2a7d-46e9-bd1e-115b161f9c3e] socks forwarding established
2025-07-31 21:31:44.765 [info] [command][645c2422-1cc3-45ef-b403-1fa392bfb8fa] Process exited with code 0
2025-07-31 21:31:44.765 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0106162d-2a7d-46e9-bd1e-115b161f9c3e] socks connection closed
2025-07-31 21:31:44.765 [info] [command][645c2422-1cc3-45ef-b403-1fa392bfb8fa] Socket close event received
2025-07-31 21:31:44.829 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55985 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:32:44.767 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:32:44.768 [info] [command][b2e2f1d1-161a-43db-b61b-2afbbda65f09] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b2e2f1d1-161a-43db-b61b-2afbbda65f09""}
2025-07-31 21:32:44.769 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][02d27c9a-2742-4f76-87ae-b1eaff0316b6] received connection request
2025-07-31 21:32:44.769 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:32:44.809 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][02d27c9a-2742-4f76-87ae-b1eaff0316b6] socks forwarding established
2025-07-31 21:32:44.851 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][02d27c9a-2742-4f76-87ae-b1eaff0316b6] socks connection closed
2025-07-31 21:32:44.851 [info] [command][b2e2f1d1-161a-43db-b61b-2afbbda65f09] Process exited with code 0
2025-07-31 21:32:44.851 [info] [command][b2e2f1d1-161a-43db-b61b-2afbbda65f09] Socket close event received
2025-07-31 21:32:44.890 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56016 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:33:44.852 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:33:44.854 [info] [command][e5b2b0e8-f8a6-4e4f-9752-5eba0ca9d1e4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e5b2b0e8-f8a6-4e4f-9752-5eba0ca9d1e4""}
2025-07-31 21:33:44.854 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8afd10a7-2d28-4d6b-9d38-e84be769b8c6] received connection request
2025-07-31 21:33:44.854 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:33:44.975 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8afd10a7-2d28-4d6b-9d38-e84be769b8c6] socks forwarding established
2025-07-31 21:33:45.021 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8afd10a7-2d28-4d6b-9d38-e84be769b8c6] socks connection closed
2025-07-31 21:33:45.021 [info] [command][e5b2b0e8-f8a6-4e4f-9752-5eba0ca9d1e4] Process exited with code 0
2025-07-31 21:33:45.021 [info] [command][e5b2b0e8-f8a6-4e4f-9752-5eba0ca9d1e4] Socket close event received
2025-07-31 21:33:45.062 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56050 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:34:45.025 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:34:45.026 [info] [command][e8850a52-6b7b-4766-bd97-53e98b124c45] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e8850a52-6b7b-4766-bd97-53e98b124c45""}
2025-07-31 21:34:45.027 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][918c79a8-35ad-485a-96f2-22ac33e8a57b] received connection request
2025-07-31 21:34:45.027 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:34:45.110 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][918c79a8-35ad-485a-96f2-22ac33e8a57b] socks forwarding established
2025-07-31 21:34:45.160 [info] [command][e8850a52-6b7b-4766-bd97-53e98b124c45] Process exited with code 0
2025-07-31 21:34:45.161 [info] [command][e8850a52-6b7b-4766-bd97-53e98b124c45] Socket close event received
2025-07-31 21:34:45.161 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][918c79a8-35ad-485a-96f2-22ac33e8a57b] socks connection closed
2025-07-31 21:34:45.200 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56072 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:35:45.165 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:35:45.166 [info] [command][37c4d4b2-a8ac-477c-8dba-d93440f09a92] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""37c4d4b2-a8ac-477c-8dba-d93440f09a92""}
2025-07-31 21:35:45.167 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][187b5def-ebe7-430d-bf56-bc217304137c] received connection request
2025-07-31 21:35:45.168 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:35:45.228 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][187b5def-ebe7-430d-bf56-bc217304137c] socks forwarding established
2025-07-31 21:35:45.276 [info] [command][37c4d4b2-a8ac-477c-8dba-d93440f09a92] Process exited with code 0
2025-07-31 21:35:45.276 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][187b5def-ebe7-430d-bf56-bc217304137c] socks connection closed
2025-07-31 21:35:45.276 [info] [command][37c4d4b2-a8ac-477c-8dba-d93440f09a92] Socket close event received
2025-07-31 21:35:45.375 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56091 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:36:45.282 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:36:45.283 [info] [command][947b801d-41ee-4aac-95a5-6fd6c0bb6f58] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""947b801d-41ee-4aac-95a5-6fd6c0bb6f58""}
2025-07-31 21:36:45.284 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0d73d167-0a42-4702-8de3-61f30e807946] received connection request
2025-07-31 21:36:45.284 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:36:45.326 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0d73d167-0a42-4702-8de3-61f30e807946] socks forwarding established
2025-07-31 21:36:45.369 [info] [command][947b801d-41ee-4aac-95a5-6fd6c0bb6f58] Process exited with code 0
2025-07-31 21:36:45.370 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0d73d167-0a42-4702-8de3-61f30e807946] socks connection closed
2025-07-31 21:36:45.370 [info] [command][947b801d-41ee-4aac-95a5-6fd6c0bb6f58] Socket close event received
2025-07-31 21:36:45.408 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56147 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:37:45.372 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:37:45.374 [info] [command][c00c03f5-3c59-45c3-a27a-9147ba06218e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c00c03f5-3c59-45c3-a27a-9147ba06218e""}
2025-07-31 21:37:45.375 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ab18ea24-d00c-4618-b218-e1258a3de78d] received connection request
2025-07-31 21:37:45.375 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:37:45.417 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab18ea24-d00c-4618-b218-e1258a3de78d] socks forwarding established
2025-07-31 21:37:45.464 [info] [command][c00c03f5-3c59-45c3-a27a-9147ba06218e] Process exited with code 0
2025-07-31 21:37:45.464 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab18ea24-d00c-4618-b218-e1258a3de78d] socks connection closed
2025-07-31 21:37:45.464 [info] [command][c00c03f5-3c59-45c3-a27a-9147ba06218e] Socket close event received
2025-07-31 21:37:45.513 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56184 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:38:45.469 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:38:45.472 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cd018442-804c-42e3-91cc-53f34e7342cc] received connection request
2025-07-31 21:38:45.474 [info] [command][e4f4259e-28f1-4321-ae95-a0fe5a661dde] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e4f4259e-28f1-4321-ae95-a0fe5a661dde""}
2025-07-31 21:38:45.474 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:38:45.517 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd018442-804c-42e3-91cc-53f34e7342cc] socks forwarding established
2025-07-31 21:38:45.563 [info] [command][e4f4259e-28f1-4321-ae95-a0fe5a661dde] Process exited with code 0
2025-07-31 21:38:45.563 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd018442-804c-42e3-91cc-53f34e7342cc] socks connection closed
2025-07-31 21:38:45.564 [info] [command][e4f4259e-28f1-4321-ae95-a0fe5a661dde] Socket close event received
2025-07-31 21:38:45.605 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56205 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:39:45.569 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:39:45.571 [info] [command][f7f1d777-5902-4fa9-a823-6a193e3ac2c3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f7f1d777-5902-4fa9-a823-6a193e3ac2c3""}
2025-07-31 21:39:45.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e82f6a45-5bff-422d-a395-577da4084e9c] received connection request
2025-07-31 21:39:45.573 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:39:45.617 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e82f6a45-5bff-422d-a395-577da4084e9c] socks forwarding established
2025-07-31 21:39:45.661 [info] [command][f7f1d777-5902-4fa9-a823-6a193e3ac2c3] Process exited with code 0
2025-07-31 21:39:45.661 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e82f6a45-5bff-422d-a395-577da4084e9c] socks connection closed
2025-07-31 21:39:45.662 [info] [command][f7f1d777-5902-4fa9-a823-6a193e3ac2c3] Socket close event received
2025-07-31 21:39:45.704 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56224 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:40:45.666 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:40:45.666 [info] [command][9f10c38b-720a-4ab1-bcd4-99d1822005e9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9f10c38b-720a-4ab1-bcd4-99d1822005e9""}
2025-07-31 21:40:45.666 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c7d99e69-87c8-44d9-af6a-a38d7874ce61] received connection request
2025-07-31 21:40:45.667 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:40:45.755 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c7d99e69-87c8-44d9-af6a-a38d7874ce61] socks forwarding established
2025-07-31 21:40:45.800 [info] [command][9f10c38b-720a-4ab1-bcd4-99d1822005e9] Process exited with code 0
2025-07-31 21:40:45.801 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c7d99e69-87c8-44d9-af6a-a38d7874ce61] socks connection closed
2025-07-31 21:40:45.804 [info] [command][9f10c38b-720a-4ab1-bcd4-99d1822005e9] Socket close event received
2025-07-31 21:40:45.847 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56244 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:41:45.804 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:41:45.806 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0ec3d144-a93a-43c0-9db0-f2b918c39fa0] received connection request
2025-07-31 21:41:45.807 [info] [command][13b21b5a-1567-4e65-93ec-1847b98d6669] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""13b21b5a-1567-4e65-93ec-1847b98d6669""}
2025-07-31 21:41:45.807 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:41:45.849 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ec3d144-a93a-43c0-9db0-f2b918c39fa0] socks forwarding established
2025-07-31 21:41:45.896 [info] [command][13b21b5a-1567-4e65-93ec-1847b98d6669] Process exited with code 0
2025-07-31 21:41:45.897 [info] [command][13b21b5a-1567-4e65-93ec-1847b98d6669] Socket close event received
2025-07-31 21:41:45.897 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ec3d144-a93a-43c0-9db0-f2b918c39fa0] socks connection closed
2025-07-31 21:41:45.939 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56279 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:42:45.903 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:42:45.904 [info] [command][d56484bf-3c6c-4899-b7b7-4e3b3a02db75] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d56484bf-3c6c-4899-b7b7-4e3b3a02db75""}
2025-07-31 21:42:45.905 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4ab43804-3adf-4ad1-bda6-0dc4fab71365] received connection request
2025-07-31 21:42:45.905 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:42:45.947 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4ab43804-3adf-4ad1-bda6-0dc4fab71365] socks forwarding established
2025-07-31 21:42:45.993 [info] [command][d56484bf-3c6c-4899-b7b7-4e3b3a02db75] Process exited with code 0
2025-07-31 21:42:45.994 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4ab43804-3adf-4ad1-bda6-0dc4fab71365] socks connection closed
2025-07-31 21:42:45.994 [info] [command][d56484bf-3c6c-4899-b7b7-4e3b3a02db75] Socket close event received
2025-07-31 21:42:46.039 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56319 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:43:45.996 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:43:46.001 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3df64f51-3ead-4628-b553-15478a678bb3] received connection request
2025-07-31 21:43:46.002 [info] [command][1b52bd93-d6b5-4646-9526-522cdd5bf909] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1b52bd93-d6b5-4646-9526-522cdd5bf909""}
2025-07-31 21:43:46.003 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:43:46.122 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3df64f51-3ead-4628-b553-15478a678bb3] socks forwarding established
2025-07-31 21:43:46.169 [info] [command][1b52bd93-d6b5-4646-9526-522cdd5bf909] Process exited with code 0
2025-07-31 21:43:46.169 [info] [command][1b52bd93-d6b5-4646-9526-522cdd5bf909] Socket close event received
2025-07-31 21:43:46.169 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3df64f51-3ead-4628-b553-15478a678bb3] socks connection closed
2025-07-31 21:43:46.213 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56345 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:44:46.170 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:44:46.173 [info] [command][b111830e-6753-4aec-a8ff-e51ebf20e158] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b111830e-6753-4aec-a8ff-e51ebf20e158""}
2025-07-31 21:44:46.173 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][79aa69b4-1ec7-46b9-8b00-f67884dc1743] received connection request
2025-07-31 21:44:46.174 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:44:46.219 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][79aa69b4-1ec7-46b9-8b00-f67884dc1743] socks forwarding established
2025-07-31 21:44:46.267 [info] [command][b111830e-6753-4aec-a8ff-e51ebf20e158] Process exited with code 0
2025-07-31 21:44:46.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][79aa69b4-1ec7-46b9-8b00-f67884dc1743] socks connection closed
2025-07-31 21:44:46.268 [info] [command][b111830e-6753-4aec-a8ff-e51ebf20e158] Socket close event received
2025-07-31 21:44:46.311 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56367 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:45:46.272 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:45:46.273 [info] [command][730ac23d-90b2-4ae3-b967-2a0f76dcd2fc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""730ac23d-90b2-4ae3-b967-2a0f76dcd2fc""}
2025-07-31 21:45:46.274 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][37b0355f-66de-44e2-b3c4-5f4db5fb35fb] received connection request
2025-07-31 21:45:46.274 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:45:46.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][37b0355f-66de-44e2-b3c4-5f4db5fb35fb] socks forwarding established
2025-07-31 21:45:46.366 [info] [command][730ac23d-90b2-4ae3-b967-2a0f76dcd2fc] Process exited with code 0
2025-07-31 21:45:46.366 [info] [command][730ac23d-90b2-4ae3-b967-2a0f76dcd2fc] Socket close event received
2025-07-31 21:45:46.367 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][37b0355f-66de-44e2-b3c4-5f4db5fb35fb] socks connection closed
2025-07-31 21:45:46.407 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56379 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:46:46.371 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:46:46.374 [info] [command][82a39293-1447-4944-a770-33e334e20204] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""82a39293-1447-4944-a770-33e334e20204""}
2025-07-31 21:46:46.375 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d0abe9a4-5f7b-48eb-8ec7-3f504d41587b] received connection request
2025-07-31 21:46:46.376 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:46:46.418 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0abe9a4-5f7b-48eb-8ec7-3f504d41587b] socks forwarding established
2025-07-31 21:46:46.468 [info] [command][82a39293-1447-4944-a770-33e334e20204] Process exited with code 0
2025-07-31 21:46:46.469 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0abe9a4-5f7b-48eb-8ec7-3f504d41587b] socks connection closed
2025-07-31 21:46:46.469 [info] [command][82a39293-1447-4944-a770-33e334e20204] Socket close event received
2025-07-31 21:46:46.577 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56422 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:47:46.472 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:47:46.474 [info] [command][55c8796d-c60d-46cb-870a-2c9647ee9830] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""55c8796d-c60d-46cb-870a-2c9647ee9830""}
2025-07-31 21:47:46.474 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][db0dd408-4110-469e-aa35-57bb853069cc] received connection request
2025-07-31 21:47:46.474 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 21:47:46.474 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:47:46.514 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db0dd408-4110-469e-aa35-57bb853069cc] socks forwarding established
2025-07-31 21:47:46.557 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db0dd408-4110-469e-aa35-57bb853069cc] socks connection closed
2025-07-31 21:47:46.557 [info] [command][55c8796d-c60d-46cb-870a-2c9647ee9830] Process exited with code 0
2025-07-31 21:47:46.557 [info] [command][55c8796d-c60d-46cb-870a-2c9647ee9830] Socket close event received
2025-07-31 21:47:46.599 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56480 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:48:46.558 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:48:46.560 [info] [command][6826cf90-3abb-48ff-bb59-a06462efe4a8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6826cf90-3abb-48ff-bb59-a06462efe4a8""}
2025-07-31 21:48:46.560 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][25d4d374-01c1-467d-bc86-f66cfa9d22f9] received connection request
2025-07-31 21:48:46.560 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:48:46.600 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][25d4d374-01c1-467d-bc86-f66cfa9d22f9] socks forwarding established
2025-07-31 21:48:46.644 [info] [command][6826cf90-3abb-48ff-bb59-a06462efe4a8] Process exited with code 0
2025-07-31 21:48:46.644 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][25d4d374-01c1-467d-bc86-f66cfa9d22f9] socks connection closed
2025-07-31 21:48:46.644 [info] [command][6826cf90-3abb-48ff-bb59-a06462efe4a8] Socket close event received
2025-07-31 21:48:46.684 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56507 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:49:46.647 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:49:46.649 [info] [command][694189f4-f4ac-42a6-8178-854289106a7d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""694189f4-f4ac-42a6-8178-854289106a7d""}
2025-07-31 21:49:46.649 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][50f1eab4-4653-4a0b-ada4-b0fbaca6aff6] received connection request
2025-07-31 21:49:46.649 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:49:46.716 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50f1eab4-4653-4a0b-ada4-b0fbaca6aff6] socks forwarding established
2025-07-31 21:49:46.760 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50f1eab4-4653-4a0b-ada4-b0fbaca6aff6] socks connection closed
2025-07-31 21:49:46.760 [info] [command][694189f4-f4ac-42a6-8178-854289106a7d] Process exited with code 0
2025-07-31 21:49:46.760 [info] [command][694189f4-f4ac-42a6-8178-854289106a7d] Socket close event received
2025-07-31 21:49:46.799 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56532 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:50:46.762 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:50:46.763 [info] [command][599cf1ba-5028-4742-99eb-e90212bdcf95] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""599cf1ba-5028-4742-99eb-e90212bdcf95""}
2025-07-31 21:50:46.763 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][69f40c53-56be-4abc-b0f6-ba0ff8afff2a] received connection request
2025-07-31 21:50:46.763 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:50:46.802 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69f40c53-56be-4abc-b0f6-ba0ff8afff2a] socks forwarding established
2025-07-31 21:50:46.851 [info] [command][599cf1ba-5028-4742-99eb-e90212bdcf95] Process exited with code 0
2025-07-31 21:50:46.851 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69f40c53-56be-4abc-b0f6-ba0ff8afff2a] socks connection closed
2025-07-31 21:50:46.851 [info] [command][599cf1ba-5028-4742-99eb-e90212bdcf95] Socket close event received
2025-07-31 21:50:46.890 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56551 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:51:46.854 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:51:46.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7d0832cc-180f-4400-86e0-9a4c81ca8b08] received connection request
2025-07-31 21:51:46.857 [info] [command][f7916df1-6f0e-4d89-b0c3-938632e31826] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f7916df1-6f0e-4d89-b0c3-938632e31826""}
2025-07-31 21:51:46.858 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:51:46.917 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7d0832cc-180f-4400-86e0-9a4c81ca8b08] socks forwarding established
2025-07-31 21:51:46.966 [info] [command][f7916df1-6f0e-4d89-b0c3-938632e31826] Process exited with code 0
2025-07-31 21:51:46.966 [info] [command][f7916df1-6f0e-4d89-b0c3-938632e31826] Socket close event received
2025-07-31 21:51:46.966 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7d0832cc-180f-4400-86e0-9a4c81ca8b08] socks connection closed
2025-07-31 21:51:47.065 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56605 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:52:46.971 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:52:46.974 [info] [command][f15e6134-b2bd-4d61-bce6-605a149aea89] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f15e6134-b2bd-4d61-bce6-605a149aea89""}
2025-07-31 21:52:46.975 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][03d6c946-1d7c-4330-804c-357e7b9440dc] received connection request
2025-07-31 21:52:46.975 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:52:47.077 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03d6c946-1d7c-4330-804c-357e7b9440dc] socks forwarding established
2025-07-31 21:52:47.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03d6c946-1d7c-4330-804c-357e7b9440dc] socks connection closed
2025-07-31 21:52:47.121 [info] [command][f15e6134-b2bd-4d61-bce6-605a149aea89] Process exited with code 0
2025-07-31 21:52:47.121 [info] [command][f15e6134-b2bd-4d61-bce6-605a149aea89] Socket close event received
2025-07-31 21:52:47.161 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56646 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:53:47.127 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:53:47.130 [info] [command][4c5928d5-885e-401a-be67-4a97d1f86bcf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4c5928d5-885e-401a-be67-4a97d1f86bcf""}
2025-07-31 21:53:47.130 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][83665c52-08fc-4bea-aea0-25338d019d77] received connection request
2025-07-31 21:53:47.130 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:53:47.173 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][83665c52-08fc-4bea-aea0-25338d019d77] socks forwarding established
2025-07-31 21:53:47.220 [info] [command][4c5928d5-885e-401a-be67-4a97d1f86bcf] Process exited with code 0
2025-07-31 21:53:47.221 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][83665c52-08fc-4bea-aea0-25338d019d77] socks connection closed
2025-07-31 21:53:47.221 [info] [command][4c5928d5-885e-401a-be67-4a97d1f86bcf] Socket close event received
2025-07-31 21:53:47.336 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56675 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:54:47.226 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:54:47.229 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a05830f5-fbae-45e2-979d-a4224294a8c4] received connection request
2025-07-31 21:54:47.229 [info] [command][ffe63278-845a-4d28-a20e-d0c6d9e6dc0e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ffe63278-845a-4d28-a20e-d0c6d9e6dc0e""}
2025-07-31 21:54:47.229 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:54:47.270 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a05830f5-fbae-45e2-979d-a4224294a8c4] socks forwarding established
2025-07-31 21:54:47.318 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a05830f5-fbae-45e2-979d-a4224294a8c4] socks connection closed
2025-07-31 21:54:47.318 [info] [command][ffe63278-845a-4d28-a20e-d0c6d9e6dc0e] Process exited with code 0
2025-07-31 21:54:47.318 [info] [command][ffe63278-845a-4d28-a20e-d0c6d9e6dc0e] Socket close event received
2025-07-31 21:54:47.362 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56695 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:55:47.319 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:55:47.323 [info] [command][3ec0ca0b-7059-4a7d-9704-d7143c2f50e7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3ec0ca0b-7059-4a7d-9704-d7143c2f50e7""}
2025-07-31 21:55:47.328 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b4f4f4fc-5c70-4bb4-afa5-8f267012c409] received connection request
2025-07-31 21:55:47.329 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:55:47.369 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b4f4f4fc-5c70-4bb4-afa5-8f267012c409] socks forwarding established
2025-07-31 21:55:47.413 [info] [command][3ec0ca0b-7059-4a7d-9704-d7143c2f50e7] Process exited with code 0
2025-07-31 21:55:47.413 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b4f4f4fc-5c70-4bb4-afa5-8f267012c409] socks connection closed
2025-07-31 21:55:47.413 [info] [command][3ec0ca0b-7059-4a7d-9704-d7143c2f50e7] Socket close event received
2025-07-31 21:55:47.454 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56717 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:56:47.414 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:56:47.416 [info] [command][4fc2f677-b511-4fd3-936a-08e2c224075e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4fc2f677-b511-4fd3-936a-08e2c224075e""}
2025-07-31 21:56:47.416 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fbccb8f4-34c7-4d29-bbcb-e8cfa165724a] received connection request
2025-07-31 21:56:47.417 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:56:47.462 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fbccb8f4-34c7-4d29-bbcb-e8cfa165724a] socks forwarding established
2025-07-31 21:56:47.509 [info] [command][4fc2f677-b511-4fd3-936a-08e2c224075e] Process exited with code 0
2025-07-31 21:56:47.509 [info] [command][4fc2f677-b511-4fd3-936a-08e2c224075e] Socket close event received
2025-07-31 21:56:47.511 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fbccb8f4-34c7-4d29-bbcb-e8cfa165724a] socks connection closed
2025-07-31 21:56:47.551 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56749 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:57:47.514 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:57:47.516 [info] [command][39cd611b-32e5-4ca9-b436-bb79022ec3ab] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""39cd611b-32e5-4ca9-b436-bb79022ec3ab""}
2025-07-31 21:57:47.516 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][31df3e1d-7d5f-4484-a90f-efbcb16640d3] received connection request
2025-07-31 21:57:47.517 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:57:47.562 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][31df3e1d-7d5f-4484-a90f-efbcb16640d3] socks forwarding established
2025-07-31 21:57:47.610 [info] [command][39cd611b-32e5-4ca9-b436-bb79022ec3ab] Process exited with code 0
2025-07-31 21:57:47.610 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][31df3e1d-7d5f-4484-a90f-efbcb16640d3] socks connection closed
2025-07-31 21:57:47.610 [info] [command][39cd611b-32e5-4ca9-b436-bb79022ec3ab] Socket close event received
2025-07-31 21:57:47.652 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56776 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:58:47.613 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:58:47.615 [info] [command][40a567fd-3e8a-4cd6-92d8-57937c2c9a93] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""40a567fd-3e8a-4cd6-92d8-57937c2c9a93""}
2025-07-31 21:58:47.616 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f3538ee0-b4fe-4468-80a7-a3a5a88bcdad] received connection request
2025-07-31 21:58:47.616 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:58:47.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f3538ee0-b4fe-4468-80a7-a3a5a88bcdad] socks forwarding established
2025-07-31 21:58:47.704 [info] [command][40a567fd-3e8a-4cd6-92d8-57937c2c9a93] Process exited with code 0
2025-07-31 21:58:47.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f3538ee0-b4fe-4468-80a7-a3a5a88bcdad] socks connection closed
2025-07-31 21:58:47.705 [info] [command][40a567fd-3e8a-4cd6-92d8-57937c2c9a93] Socket close event received
2025-07-31 21:58:47.745 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56797 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 21:59:47.706 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 21:59:47.709 [info] [command][a9de077b-c714-47fc-870d-08cc455d67a4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a9de077b-c714-47fc-870d-08cc455d67a4""}
2025-07-31 21:59:47.709 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5fad51a8-cf09-4e55-aa26-4497fd60dabf] received connection request
2025-07-31 21:59:47.710 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 21:59:47.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5fad51a8-cf09-4e55-aa26-4497fd60dabf] socks forwarding established
2025-07-31 21:59:47.809 [info] [command][a9de077b-c714-47fc-870d-08cc455d67a4] Process exited with code 0
2025-07-31 21:59:47.809 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5fad51a8-cf09-4e55-aa26-4497fd60dabf] socks connection closed
2025-07-31 21:59:47.809 [info] [command][a9de077b-c714-47fc-870d-08cc455d67a4] Socket close event received
2025-07-31 21:59:47.848 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56813 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:00:47.811 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:00:47.814 [info] [command][9cb9514d-9738-4625-9fa8-111bf4bb17c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9cb9514d-9738-4625-9fa8-111bf4bb17c7""}
2025-07-31 22:00:47.815 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][88dae9c0-102a-4129-baf8-9a380fef19b3] received connection request
2025-07-31 22:00:47.816 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:00:47.860 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][88dae9c0-102a-4129-baf8-9a380fef19b3] socks forwarding established
2025-07-31 22:00:47.908 [info] [command][9cb9514d-9738-4625-9fa8-111bf4bb17c7] Process exited with code 0
2025-07-31 22:00:47.908 [info] [command][9cb9514d-9738-4625-9fa8-111bf4bb17c7] Socket close event received
2025-07-31 22:00:47.909 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][88dae9c0-102a-4129-baf8-9a380fef19b3] socks connection closed
2025-07-31 22:00:47.951 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56847 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:01:47.910 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:01:47.912 [info] [command][e3927fb5-62b8-4410-9f57-e6de077baef4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e3927fb5-62b8-4410-9f57-e6de077baef4""}
2025-07-31 22:01:47.912 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][06425027-44de-4f4a-a2da-52fec5559f56] received connection request
2025-07-31 22:01:47.912 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:01:47.957 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06425027-44de-4f4a-a2da-52fec5559f56] socks forwarding established
2025-07-31 22:01:48.002 [info] [command][e3927fb5-62b8-4410-9f57-e6de077baef4] Process exited with code 0
2025-07-31 22:01:48.005 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06425027-44de-4f4a-a2da-52fec5559f56] socks connection closed
2025-07-31 22:01:48.005 [info] [command][e3927fb5-62b8-4410-9f57-e6de077baef4] Socket close event received
2025-07-31 22:01:48.046 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56879 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:02:48.005 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:02:48.007 [info] [command][b7037857-b1cd-4bf5-a267-3355ae80acf2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7037857-b1cd-4bf5-a267-3355ae80acf2""}
2025-07-31 22:02:48.008 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2c30194c-6d12-4ef2-8de5-47291f69bdbf] received connection request
2025-07-31 22:02:48.008 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:02:48.047 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c30194c-6d12-4ef2-8de5-47291f69bdbf] socks forwarding established
2025-07-31 22:02:48.092 [info] [command][b7037857-b1cd-4bf5-a267-3355ae80acf2] Process exited with code 0
2025-07-31 22:02:48.092 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c30194c-6d12-4ef2-8de5-47291f69bdbf] socks connection closed
2025-07-31 22:02:48.092 [info] [command][b7037857-b1cd-4bf5-a267-3355ae80acf2] Socket close event received
2025-07-31 22:02:48.133 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56909 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:03:48.097 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:03:48.100 [info] [command][b56c3bdc-53d8-4bc4-af5a-495609f150fa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b56c3bdc-53d8-4bc4-af5a-495609f150fa""}
2025-07-31 22:03:48.100 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0b9874aa-ce5a-468d-be18-1471f2ae4e6e] received connection request
2025-07-31 22:03:48.101 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:03:48.146 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0b9874aa-ce5a-468d-be18-1471f2ae4e6e] socks forwarding established
2025-07-31 22:03:48.191 [info] [command][b56c3bdc-53d8-4bc4-af5a-495609f150fa] Process exited with code 0
2025-07-31 22:03:48.191 [info] [command][b56c3bdc-53d8-4bc4-af5a-495609f150fa] Socket close event received
2025-07-31 22:03:48.192 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0b9874aa-ce5a-468d-be18-1471f2ae4e6e] socks connection closed
2025-07-31 22:03:48.230 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56937 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:04:48.196 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:04:48.199 [info] [command][c22cd563-8004-48f6-8861-fe4a7b03e02e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c22cd563-8004-48f6-8861-fe4a7b03e02e""}
2025-07-31 22:04:48.200 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4bd8cc04-50c0-44f2-9ed3-e8db66a6d259] received connection request
2025-07-31 22:04:48.201 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:04:48.243 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bd8cc04-50c0-44f2-9ed3-e8db66a6d259] socks forwarding established
2025-07-31 22:04:48.290 [info] [command][c22cd563-8004-48f6-8861-fe4a7b03e02e] Process exited with code 0
2025-07-31 22:04:48.290 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bd8cc04-50c0-44f2-9ed3-e8db66a6d259] socks connection closed
2025-07-31 22:04:48.290 [info] [command][c22cd563-8004-48f6-8861-fe4a7b03e02e] Socket close event received
2025-07-31 22:04:48.329 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56963 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:05:48.295 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:05:48.297 [info] [command][a813fc14-4dc2-4e39-9ab1-16a8fa3343e3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a813fc14-4dc2-4e39-9ab1-16a8fa3343e3""}
2025-07-31 22:05:48.298 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3dd2bdc7-97f5-4ba1-9153-341c40c5b4bb] received connection request
2025-07-31 22:05:48.298 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:05:48.340 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3dd2bdc7-97f5-4ba1-9153-341c40c5b4bb] socks forwarding established
2025-07-31 22:05:48.386 [info] [command][a813fc14-4dc2-4e39-9ab1-16a8fa3343e3] Process exited with code 0
2025-07-31 22:05:48.386 [info] [command][a813fc14-4dc2-4e39-9ab1-16a8fa3343e3] Socket close event received
2025-07-31 22:05:48.386 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3dd2bdc7-97f5-4ba1-9153-341c40c5b4bb] socks connection closed
2025-07-31 22:05:48.425 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57018 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:06:48.387 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:06:48.388 [info] [command][8a43eb79-17cb-4e12-9d50-e14fa3c181b3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8a43eb79-17cb-4e12-9d50-e14fa3c181b3""}
2025-07-31 22:06:48.388 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9167a406-c8c3-4a27-99cb-8e9d232c82dd] received connection request
2025-07-31 22:06:48.388 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 22:06:48.388 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:06:48.477 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9167a406-c8c3-4a27-99cb-8e9d232c82dd] socks forwarding established
2025-07-31 22:06:48.541 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9167a406-c8c3-4a27-99cb-8e9d232c82dd] socks connection closed
2025-07-31 22:06:48.541 [info] [command][8a43eb79-17cb-4e12-9d50-e14fa3c181b3] Process exited with code 0
2025-07-31 22:06:48.541 [info] [command][8a43eb79-17cb-4e12-9d50-e14fa3c181b3] Socket close event received
2025-07-31 22:06:48.581 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57070 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:07:48.545 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:07:48.547 [info] [command][060ff13c-4bec-47a8-bd3e-090039d898c9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""060ff13c-4bec-47a8-bd3e-090039d898c9""}
2025-07-31 22:07:48.547 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][06941e0e-36e4-452e-80f8-b18d06c59484] received connection request
2025-07-31 22:07:48.547 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:07:48.610 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06941e0e-36e4-452e-80f8-b18d06c59484] socks forwarding established
2025-07-31 22:07:48.654 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06941e0e-36e4-452e-80f8-b18d06c59484] socks connection closed
2025-07-31 22:07:48.654 [info] [command][060ff13c-4bec-47a8-bd3e-090039d898c9] Process exited with code 0
2025-07-31 22:07:48.654 [info] [command][060ff13c-4bec-47a8-bd3e-090039d898c9] Socket close event received
2025-07-31 22:07:48.694 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57139 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:08:48.658 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:08:48.659 [info] [command][f2b98354-eef8-4390-a207-cc8d6c734e52] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f2b98354-eef8-4390-a207-cc8d6c734e52""}
2025-07-31 22:08:48.659 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ab1b5b32-5ad1-4121-ad70-749956d95b27] received connection request
2025-07-31 22:08:48.659 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:08:48.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1b5b32-5ad1-4121-ad70-749956d95b27] socks forwarding established
2025-07-31 22:08:48.745 [info] [command][f2b98354-eef8-4390-a207-cc8d6c734e52] Process exited with code 0
2025-07-31 22:08:48.746 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1b5b32-5ad1-4121-ad70-749956d95b27] socks connection closed
2025-07-31 22:08:48.746 [info] [command][f2b98354-eef8-4390-a207-cc8d6c734e52] Socket close event received
2025-07-31 22:08:48.787 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57166 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:09:48.749 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:09:48.751 [info] [command][25d4b3dd-a928-4c1e-9d99-e986b9b24b10] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""25d4b3dd-a928-4c1e-9d99-e986b9b24b10""}
2025-07-31 22:09:48.751 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][778d06a4-8fa2-42a5-b737-43d68c2c2e0d] received connection request
2025-07-31 22:09:48.751 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:09:48.791 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][778d06a4-8fa2-42a5-b737-43d68c2c2e0d] socks forwarding established
2025-07-31 22:09:48.837 [info] [command][25d4b3dd-a928-4c1e-9d99-e986b9b24b10] Process exited with code 0
2025-07-31 22:09:48.837 [info] [command][25d4b3dd-a928-4c1e-9d99-e986b9b24b10] Socket close event received
2025-07-31 22:09:48.837 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][778d06a4-8fa2-42a5-b737-43d68c2c2e0d] socks connection closed
2025-07-31 22:09:48.879 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57207 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:10:48.842 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:10:48.845 [info] [command][9965b818-8e7d-477a-8b79-8a0b7935c98b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9965b818-8e7d-477a-8b79-8a0b7935c98b""}
2025-07-31 22:10:48.846 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7d482aa0-0cc6-4070-88a4-68fd7b727981] received connection request
2025-07-31 22:10:48.846 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:10:48.890 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7d482aa0-0cc6-4070-88a4-68fd7b727981] socks forwarding established
2025-07-31 22:10:48.937 [info] [command][9965b818-8e7d-477a-8b79-8a0b7935c98b] Process exited with code 0
2025-07-31 22:10:48.937 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7d482aa0-0cc6-4070-88a4-68fd7b727981] socks connection closed
2025-07-31 22:10:48.937 [info] [command][9965b818-8e7d-477a-8b79-8a0b7935c98b] Socket close event received
2025-07-31 22:10:48.978 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57238 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:11:48.941 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:11:48.944 [info] [command][8e85e36e-16df-4aa1-9300-8d92e802a27f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8e85e36e-16df-4aa1-9300-8d92e802a27f""}
2025-07-31 22:11:48.944 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0ce4efc1-0dee-4bbd-84b5-61ad04cb6127] received connection request
2025-07-31 22:11:48.945 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:11:49.047 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ce4efc1-0dee-4bbd-84b5-61ad04cb6127] socks forwarding established
2025-07-31 22:11:49.096 [info] [command][8e85e36e-16df-4aa1-9300-8d92e802a27f] Process exited with code 0
2025-07-31 22:11:49.096 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ce4efc1-0dee-4bbd-84b5-61ad04cb6127] socks connection closed
2025-07-31 22:11:49.096 [info] [command][8e85e36e-16df-4aa1-9300-8d92e802a27f] Socket close event received
2025-07-31 22:11:49.135 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57340 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:12:49.098 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:12:49.101 [info] [command][0585afef-03b8-48b5-b3db-5039f8414cd2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0585afef-03b8-48b5-b3db-5039f8414cd2""}
2025-07-31 22:12:49.102 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ac47981d-d7c5-4227-bf38-033bd6edd525] received connection request
2025-07-31 22:12:49.103 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:12:49.143 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac47981d-d7c5-4227-bf38-033bd6edd525] socks forwarding established
2025-07-31 22:12:49.190 [info] [command][0585afef-03b8-48b5-b3db-5039f8414cd2] Process exited with code 0
2025-07-31 22:12:49.190 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac47981d-d7c5-4227-bf38-033bd6edd525] socks connection closed
2025-07-31 22:12:49.190 [info] [command][0585afef-03b8-48b5-b3db-5039f8414cd2] Socket close event received
2025-07-31 22:12:49.236 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57398 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:13:49.195 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:13:49.201 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][baa042b0-8ad1-40eb-84cd-8129f9c25b38] received connection request
2025-07-31 22:13:49.202 [info] [command][ac3063c5-c116-4081-9220-a0b5cb0765af] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ac3063c5-c116-4081-9220-a0b5cb0765af""}
2025-07-31 22:13:49.202 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:13:49.248 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][baa042b0-8ad1-40eb-84cd-8129f9c25b38] socks forwarding established
2025-07-31 22:13:49.300 [info] [command][ac3063c5-c116-4081-9220-a0b5cb0765af] Process exited with code 0
2025-07-31 22:13:49.301 [info] [command][ac3063c5-c116-4081-9220-a0b5cb0765af] Socket close event received
2025-07-31 22:13:49.301 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][baa042b0-8ad1-40eb-84cd-8129f9c25b38] socks connection closed
2025-07-31 22:13:49.340 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57412 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:14:49.306 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:14:49.309 [info] [command][2c7ed9de-f943-40b7-bcf3-a5db3ed8692e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2c7ed9de-f943-40b7-bcf3-a5db3ed8692e""}
2025-07-31 22:14:49.310 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2e92e4d5-822f-42cd-8052-7157cde584f5] received connection request
2025-07-31 22:14:49.311 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:14:49.355 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2e92e4d5-822f-42cd-8052-7157cde584f5] socks forwarding established
2025-07-31 22:14:49.401 [info] [command][2c7ed9de-f943-40b7-bcf3-a5db3ed8692e] Process exited with code 0
2025-07-31 22:14:49.402 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2e92e4d5-822f-42cd-8052-7157cde584f5] socks connection closed
2025-07-31 22:14:49.402 [info] [command][2c7ed9de-f943-40b7-bcf3-a5db3ed8692e] Socket close event received
2025-07-31 22:14:49.444 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57439 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:15:49.407 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:15:49.410 [info] [command][f38588f0-1f1c-4363-b53f-1ef31975a708] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f38588f0-1f1c-4363-b53f-1ef31975a708""}
2025-07-31 22:15:49.411 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][714dd2b5-65dd-452d-8116-1638a84a81f2] received connection request
2025-07-31 22:15:49.411 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:15:49.452 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][714dd2b5-65dd-452d-8116-1638a84a81f2] socks forwarding established
2025-07-31 22:15:49.501 [info] [command][f38588f0-1f1c-4363-b53f-1ef31975a708] Process exited with code 0
2025-07-31 22:15:49.501 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][714dd2b5-65dd-452d-8116-1638a84a81f2] socks connection closed
2025-07-31 22:15:49.501 [info] [command][f38588f0-1f1c-4363-b53f-1ef31975a708] Socket close event received
2025-07-31 22:15:49.541 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57455 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:16:49.507 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:16:49.510 [info] [command][7bbcdf49-90f5-4b88-89e7-b24ae6d470d1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7bbcdf49-90f5-4b88-89e7-b24ae6d470d1""}
2025-07-31 22:16:49.513 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][93743f6f-b22f-4142-8ed6-0e52d54f3cd9] received connection request
2025-07-31 22:16:49.514 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:16:49.565 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93743f6f-b22f-4142-8ed6-0e52d54f3cd9] socks forwarding established
2025-07-31 22:16:49.614 [info] [command][7bbcdf49-90f5-4b88-89e7-b24ae6d470d1] Process exited with code 0
2025-07-31 22:16:49.615 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93743f6f-b22f-4142-8ed6-0e52d54f3cd9] socks connection closed
2025-07-31 22:16:49.615 [info] [command][7bbcdf49-90f5-4b88-89e7-b24ae6d470d1] Socket close event received
2025-07-31 22:16:49.657 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57496 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:17:49.616 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:17:49.618 [info] [command][8f70e09a-c233-4d07-a02a-ce12a54bff9c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8f70e09a-c233-4d07-a02a-ce12a54bff9c""}
2025-07-31 22:17:49.619 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fce88b0c-76f9-4a34-8c26-4b15b4e541e4] received connection request
2025-07-31 22:17:49.620 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:17:49.665 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fce88b0c-76f9-4a34-8c26-4b15b4e541e4] socks forwarding established
2025-07-31 22:17:49.715 [info] [command][8f70e09a-c233-4d07-a02a-ce12a54bff9c] Process exited with code 0
2025-07-31 22:17:49.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fce88b0c-76f9-4a34-8c26-4b15b4e541e4] socks connection closed
2025-07-31 22:17:49.715 [info] [command][8f70e09a-c233-4d07-a02a-ce12a54bff9c] Socket close event received
2025-07-31 22:17:49.756 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57538 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:18:49.716 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:18:49.718 [info] [command][ef8f2bb0-6a31-4d4c-b4d7-3a2dd3a499e6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ef8f2bb0-6a31-4d4c-b4d7-3a2dd3a499e6""}
2025-07-31 22:18:49.719 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fb6affa6-86da-4d09-847b-8dafaa351965] received connection request
2025-07-31 22:18:49.719 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:18:49.761 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fb6affa6-86da-4d09-847b-8dafaa351965] socks forwarding established
2025-07-31 22:18:49.805 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fb6affa6-86da-4d09-847b-8dafaa351965] socks connection closed
2025-07-31 22:18:49.805 [info] [command][ef8f2bb0-6a31-4d4c-b4d7-3a2dd3a499e6] Process exited with code 0
2025-07-31 22:18:49.805 [info] [command][ef8f2bb0-6a31-4d4c-b4d7-3a2dd3a499e6] Socket close event received
2025-07-31 22:18:49.850 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57567 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:19:49.809 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:19:49.811 [info] [command][b3a9622d-f301-42a1-9b3d-a904392a0524] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b3a9622d-f301-42a1-9b3d-a904392a0524""}
2025-07-31 22:19:49.812 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9c2db44d-b08d-459e-80c3-710aa89b30fb] received connection request
2025-07-31 22:19:49.812 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:19:49.854 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9c2db44d-b08d-459e-80c3-710aa89b30fb] socks forwarding established
2025-07-31 22:19:49.902 [info] [command][b3a9622d-f301-42a1-9b3d-a904392a0524] Process exited with code 0
2025-07-31 22:19:49.902 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9c2db44d-b08d-459e-80c3-710aa89b30fb] socks connection closed
2025-07-31 22:19:49.902 [info] [command][b3a9622d-f301-42a1-9b3d-a904392a0524] Socket close event received
2025-07-31 22:19:49.946 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57598 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:20:49.906 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:20:49.911 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c0b593b4-1d14-468b-a033-2cf05b4d58d6] received connection request
2025-07-31 22:20:49.912 [info] [command][affbf9c9-1631-4f43-9cf0-dc552a797d19] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""affbf9c9-1631-4f43-9cf0-dc552a797d19""}
2025-07-31 22:20:49.913 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:20:49.956 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c0b593b4-1d14-468b-a033-2cf05b4d58d6] socks forwarding established
2025-07-31 22:20:50.003 [info] [command][affbf9c9-1631-4f43-9cf0-dc552a797d19] Process exited with code 0
2025-07-31 22:20:50.003 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c0b593b4-1d14-468b-a033-2cf05b4d58d6] socks connection closed
2025-07-31 22:20:50.003 [info] [command][affbf9c9-1631-4f43-9cf0-dc552a797d19] Socket close event received
2025-07-31 22:20:50.157 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57620 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:21:50.009 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:21:50.012 [info] [command][a94bfdf0-b1d1-4e94-bf0a-638d3bdbdfdd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a94bfdf0-b1d1-4e94-bf0a-638d3bdbdfdd""}
2025-07-31 22:21:50.013 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][89886e4d-4a99-4607-b6d0-baebc2f2e45b] received connection request
2025-07-31 22:21:50.014 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:21:50.054 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89886e4d-4a99-4607-b6d0-baebc2f2e45b] socks forwarding established
2025-07-31 22:21:50.099 [info] [command][a94bfdf0-b1d1-4e94-bf0a-638d3bdbdfdd] Process exited with code 0
2025-07-31 22:21:50.100 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89886e4d-4a99-4607-b6d0-baebc2f2e45b] socks connection closed
2025-07-31 22:21:50.100 [info] [command][a94bfdf0-b1d1-4e94-bf0a-638d3bdbdfdd] Socket close event received
2025-07-31 22:21:50.139 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57665 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:22:50.103 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:22:50.105 [info] [command][a43d2fa6-db3e-4c12-8817-f8726d51b3e9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a43d2fa6-db3e-4c12-8817-f8726d51b3e9""}
2025-07-31 22:22:50.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8a9f7e95-443e-4847-b4c9-8a815be9643d] received connection request
2025-07-31 22:22:50.106 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:22:50.150 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8a9f7e95-443e-4847-b4c9-8a815be9643d] socks forwarding established
2025-07-31 22:22:50.195 [info] [command][a43d2fa6-db3e-4c12-8817-f8726d51b3e9] Process exited with code 0
2025-07-31 22:22:50.195 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8a9f7e95-443e-4847-b4c9-8a815be9643d] socks connection closed
2025-07-31 22:22:50.195 [info] [command][a43d2fa6-db3e-4c12-8817-f8726d51b3e9] Socket close event received
2025-07-31 22:22:50.241 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57723 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:23:50.196 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:23:50.198 [info] [command][ed8cdbfc-9a6b-4632-94c0-15fc8d5af420] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ed8cdbfc-9a6b-4632-94c0-15fc8d5af420""}
2025-07-31 22:23:50.199 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][03c54f6a-ff24-4847-9f4b-94be0d44c4a3] received connection request
2025-07-31 22:23:50.200 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:23:50.243 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03c54f6a-ff24-4847-9f4b-94be0d44c4a3] socks forwarding established
2025-07-31 22:23:50.293 [info] [command][ed8cdbfc-9a6b-4632-94c0-15fc8d5af420] Process exited with code 0
2025-07-31 22:23:50.293 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][03c54f6a-ff24-4847-9f4b-94be0d44c4a3] socks connection closed
2025-07-31 22:23:50.293 [info] [command][ed8cdbfc-9a6b-4632-94c0-15fc8d5af420] Socket close event received
2025-07-31 22:23:50.418 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57773 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:24:50.298 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:24:50.300 [info] [command][8fed6e1c-3851-46fd-b962-bafa22d8519e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8fed6e1c-3851-46fd-b962-bafa22d8519e""}
2025-07-31 22:24:50.301 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ee3cf125-6bf8-4c68-ae8f-fa1515fc4f20] received connection request
2025-07-31 22:24:50.302 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:24:50.426 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee3cf125-6bf8-4c68-ae8f-fa1515fc4f20] socks forwarding established
2025-07-31 22:24:50.474 [info] [command][8fed6e1c-3851-46fd-b962-bafa22d8519e] Process exited with code 0
2025-07-31 22:24:50.476 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee3cf125-6bf8-4c68-ae8f-fa1515fc4f20] socks connection closed
2025-07-31 22:24:50.476 [info] [command][8fed6e1c-3851-46fd-b962-bafa22d8519e] Socket close event received
2025-07-31 22:24:50.676 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57820 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:25:50.479 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:25:50.481 [info] [command][b01af7dd-e3da-4bdd-be13-3fee1ca3eddd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b01af7dd-e3da-4bdd-be13-3fee1ca3eddd""}
2025-07-31 22:25:50.481 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d3043a7a-cfb4-4cea-9aa6-c93aa4ca4180] received connection request
2025-07-31 22:25:50.482 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:25:50.529 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3043a7a-cfb4-4cea-9aa6-c93aa4ca4180] socks forwarding established
2025-07-31 22:25:50.575 [info] [command][b01af7dd-e3da-4bdd-be13-3fee1ca3eddd] Process exited with code 0
2025-07-31 22:25:50.576 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3043a7a-cfb4-4cea-9aa6-c93aa4ca4180] socks connection closed
2025-07-31 22:25:50.576 [info] [command][b01af7dd-e3da-4bdd-be13-3fee1ca3eddd] Socket close event received
2025-07-31 22:25:50.615 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57848 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:26:50.577 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:26:50.579 [info] [command][f81f9d6b-41cd-4cbc-80fc-f316dc0945b3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f81f9d6b-41cd-4cbc-80fc-f316dc0945b3""}
2025-07-31 22:26:50.579 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][832b427b-9f12-48f0-aa8d-11444df2f2b2] received connection request
2025-07-31 22:26:50.579 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:26:50.623 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][832b427b-9f12-48f0-aa8d-11444df2f2b2] socks forwarding established
2025-07-31 22:26:50.669 [info] [command][f81f9d6b-41cd-4cbc-80fc-f316dc0945b3] Process exited with code 0
2025-07-31 22:26:50.670 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][832b427b-9f12-48f0-aa8d-11444df2f2b2] socks connection closed
2025-07-31 22:26:50.670 [info] [command][f81f9d6b-41cd-4cbc-80fc-f316dc0945b3] Socket close event received
2025-07-31 22:26:50.710 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57902 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:27:50.672 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:27:50.675 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][10d7401e-82ca-4008-b2a7-f864f74d76b8] received connection request
2025-07-31 22:27:50.675 [info] [command][1eb98efb-e95f-40ab-b18d-a3bf50b8471e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1eb98efb-e95f-40ab-b18d-a3bf50b8471e""}
2025-07-31 22:27:50.675 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:27:50.717 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][10d7401e-82ca-4008-b2a7-f864f74d76b8] socks forwarding established
2025-07-31 22:27:50.764 [info] [command][1eb98efb-e95f-40ab-b18d-a3bf50b8471e] Process exited with code 0
2025-07-31 22:27:50.764 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][10d7401e-82ca-4008-b2a7-f864f74d76b8] socks connection closed
2025-07-31 22:27:50.764 [info] [command][1eb98efb-e95f-40ab-b18d-a3bf50b8471e] Socket close event received
2025-07-31 22:27:50.802 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57943 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:28:50.772 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:28:50.772 [info] [command][42fee4af-2819-4705-9c49-fa1555a7ec8c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""42fee4af-2819-4705-9c49-fa1555a7ec8c""}
2025-07-31 22:28:50.773 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eba41c08-72f7-4faa-a505-c14a7bc9f050] received connection request
2025-07-31 22:28:50.773 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:28:50.816 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eba41c08-72f7-4faa-a505-c14a7bc9f050] socks forwarding established
2025-07-31 22:28:50.859 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eba41c08-72f7-4faa-a505-c14a7bc9f050] socks connection closed
2025-07-31 22:28:50.859 [info] [command][42fee4af-2819-4705-9c49-fa1555a7ec8c] Process exited with code 0
2025-07-31 22:28:50.859 [info] [command][42fee4af-2819-4705-9c49-fa1555a7ec8c] Socket close event received
2025-07-31 22:28:50.899 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57988 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:29:50.866 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:29:50.868 [info] [command][a090df82-4f55-4740-803f-5d944b449107] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a090df82-4f55-4740-803f-5d944b449107""}
2025-07-31 22:29:50.872 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a696f3ca-f5dd-4ce4-8270-ff919d1dce4e] received connection request
2025-07-31 22:29:50.873 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:29:50.992 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a696f3ca-f5dd-4ce4-8270-ff919d1dce4e] socks forwarding established
2025-07-31 22:29:51.047 [info] [command][a090df82-4f55-4740-803f-5d944b449107] Process exited with code 0
2025-07-31 22:29:51.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a696f3ca-f5dd-4ce4-8270-ff919d1dce4e] socks connection closed
2025-07-31 22:29:51.048 [info] [command][a090df82-4f55-4740-803f-5d944b449107] Socket close event received
2025-07-31 22:29:51.092 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58015 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:30:51.053 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:30:51.054 [info] [command][d5af4c74-04d9-492a-9346-0d120f3c1768] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d5af4c74-04d9-492a-9346-0d120f3c1768""}
2025-07-31 22:30:51.054 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9db03041-292b-4ea3-b70c-7f163b26c926] received connection request
2025-07-31 22:30:51.055 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:30:51.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9db03041-292b-4ea3-b70c-7f163b26c926] socks forwarding established
2025-07-31 22:30:51.158 [info] [command][d5af4c74-04d9-492a-9346-0d120f3c1768] Process exited with code 0
2025-07-31 22:30:51.158 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9db03041-292b-4ea3-b70c-7f163b26c926] socks connection closed
2025-07-31 22:30:51.158 [info] [command][d5af4c74-04d9-492a-9346-0d120f3c1768] Socket close event received
2025-07-31 22:30:51.200 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58039 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:31:51.159 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:31:51.160 [info] [command][4c1af9af-0c12-420e-946f-3828ac9c45c6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4c1af9af-0c12-420e-946f-3828ac9c45c6""}
2025-07-31 22:31:51.160 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f1039641-ef50-45d7-a43b-6ab6dd836c04] received connection request
2025-07-31 22:31:51.160 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:31:51.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f1039641-ef50-45d7-a43b-6ab6dd836c04] socks forwarding established
2025-07-31 22:31:51.323 [info] [command][4c1af9af-0c12-420e-946f-3828ac9c45c6] Process exited with code 0
2025-07-31 22:31:51.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f1039641-ef50-45d7-a43b-6ab6dd836c04] socks connection closed
2025-07-31 22:31:51.324 [info] [command][4c1af9af-0c12-420e-946f-3828ac9c45c6] Socket close event received
2025-07-31 22:31:51.364 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58085 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:32:51.328 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:32:51.331 [info] [command][68d272ce-9985-412e-a670-33e737cc248c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""68d272ce-9985-412e-a670-33e737cc248c""}
2025-07-31 22:32:51.331 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4d2d373f-9b73-4cc4-829c-37d37769ca9e] received connection request
2025-07-31 22:32:51.332 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:32:51.373 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d2d373f-9b73-4cc4-829c-37d37769ca9e] socks forwarding established
2025-07-31 22:32:51.420 [info] [command][68d272ce-9985-412e-a670-33e737cc248c] Process exited with code 0
2025-07-31 22:32:51.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d2d373f-9b73-4cc4-829c-37d37769ca9e] socks connection closed
2025-07-31 22:32:51.422 [info] [command][68d272ce-9985-412e-a670-33e737cc248c] Socket close event received
2025-07-31 22:32:51.463 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58127 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:33:51.425 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:33:51.427 [info] [command][edd0e7da-8d28-4e6b-b71d-010f449aed26] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""edd0e7da-8d28-4e6b-b71d-010f449aed26""}
2025-07-31 22:33:51.427 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0a7f8f2a-bfa4-45d9-8550-67b91f1df0de] received connection request
2025-07-31 22:33:51.428 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 22:33:51.428 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:33:51.468 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0a7f8f2a-bfa4-45d9-8550-67b91f1df0de] socks forwarding established
2025-07-31 22:33:51.514 [info] [command][edd0e7da-8d28-4e6b-b71d-010f449aed26] Process exited with code 0
2025-07-31 22:33:51.515 [info] [command][edd0e7da-8d28-4e6b-b71d-010f449aed26] Socket close event received
2025-07-31 22:33:51.515 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0a7f8f2a-bfa4-45d9-8550-67b91f1df0de] socks connection closed
2025-07-31 22:33:51.555 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58153 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:34:51.519 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:34:51.522 [info] [command][0cf9d227-1264-42ef-99cb-09d393a4754e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0cf9d227-1264-42ef-99cb-09d393a4754e""}
2025-07-31 22:34:51.523 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eb3cf546-f54c-4953-951b-51c5eb69054a] received connection request
2025-07-31 22:34:51.523 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:34:51.678 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb3cf546-f54c-4953-951b-51c5eb69054a] socks forwarding established
2025-07-31 22:34:51.726 [info] [command][0cf9d227-1264-42ef-99cb-09d393a4754e] Process exited with code 0
2025-07-31 22:34:51.727 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb3cf546-f54c-4953-951b-51c5eb69054a] socks connection closed
2025-07-31 22:34:51.727 [info] [command][0cf9d227-1264-42ef-99cb-09d393a4754e] Socket close event received
2025-07-31 22:34:51.769 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58173 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:35:51.733 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:35:51.736 [info] [command][48c48048-f132-4c94-a780-7d20a57fa2fd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""48c48048-f132-4c94-a780-7d20a57fa2fd""}
2025-07-31 22:35:51.737 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ad83171c-9e02-42b1-a692-e2ef9648b02e] received connection request
2025-07-31 22:35:51.737 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:35:51.780 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad83171c-9e02-42b1-a692-e2ef9648b02e] socks forwarding established
2025-07-31 22:35:51.828 [info] [command][48c48048-f132-4c94-a780-7d20a57fa2fd] Process exited with code 0
2025-07-31 22:35:51.829 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad83171c-9e02-42b1-a692-e2ef9648b02e] socks connection closed
2025-07-31 22:35:51.829 [info] [command][48c48048-f132-4c94-a780-7d20a57fa2fd] Socket close event received
2025-07-31 22:35:51.870 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58201 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:36:51.833 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:36:51.836 [info] [command][96eccc3d-79c8-453e-b494-adeb51df3c1c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""96eccc3d-79c8-453e-b494-adeb51df3c1c""}
2025-07-31 22:36:51.837 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ee1e7ed0-7a5d-4e42-bafe-435028e33d73] received connection request
2025-07-31 22:36:51.837 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:36:51.902 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee1e7ed0-7a5d-4e42-bafe-435028e33d73] socks forwarding established
2025-07-31 22:36:51.950 [info] [command][96eccc3d-79c8-453e-b494-adeb51df3c1c] Process exited with code 0
2025-07-31 22:36:51.950 [info] [command][96eccc3d-79c8-453e-b494-adeb51df3c1c] Socket close event received
2025-07-31 22:36:51.952 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee1e7ed0-7a5d-4e42-bafe-435028e33d73] socks connection closed
2025-07-31 22:36:51.993 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58259 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:37:51.953 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:37:51.955 [info] [command][2e2683ae-aa84-4842-a6b9-4a40e477ce8f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2e2683ae-aa84-4842-a6b9-4a40e477ce8f""}
2025-07-31 22:37:51.956 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fa02871c-a9a5-48b1-8cf1-a7f8aef34c53] received connection request
2025-07-31 22:37:51.957 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:37:52.002 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fa02871c-a9a5-48b1-8cf1-a7f8aef34c53] socks forwarding established
2025-07-31 22:37:52.049 [info] [command][2e2683ae-aa84-4842-a6b9-4a40e477ce8f] Process exited with code 0
2025-07-31 22:37:52.051 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fa02871c-a9a5-48b1-8cf1-a7f8aef34c53] socks connection closed
2025-07-31 22:37:52.051 [info] [command][2e2683ae-aa84-4842-a6b9-4a40e477ce8f] Socket close event received
2025-07-31 22:37:52.092 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58297 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:38:52.052 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:38:52.054 [info] [command][d210c55f-13a2-4892-bd7f-c5a1055dd426] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d210c55f-13a2-4892-bd7f-c5a1055dd426""}
2025-07-31 22:38:52.055 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e573cced-dfbe-4d12-a2bb-a19e56e308d8] received connection request
2025-07-31 22:38:52.056 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:38:52.172 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e573cced-dfbe-4d12-a2bb-a19e56e308d8] socks forwarding established
2025-07-31 22:38:52.217 [info] [command][d210c55f-13a2-4892-bd7f-c5a1055dd426] Process exited with code 0
2025-07-31 22:38:52.217 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e573cced-dfbe-4d12-a2bb-a19e56e308d8] socks connection closed
2025-07-31 22:38:52.217 [info] [command][d210c55f-13a2-4892-bd7f-c5a1055dd426] Socket close event received
2025-07-31 22:38:52.257 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58317 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:39:52.218 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:39:52.220 [info] [command][39f21389-d64c-4670-baea-7968167e15d8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""39f21389-d64c-4670-baea-7968167e15d8""}
2025-07-31 22:39:52.220 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][743c3c4a-8156-4ae9-9d6c-51de1747cfa5] received connection request
2025-07-31 22:39:52.221 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:39:52.269 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][743c3c4a-8156-4ae9-9d6c-51de1747cfa5] socks forwarding established
2025-07-31 22:39:52.318 [info] [command][39f21389-d64c-4670-baea-7968167e15d8] Process exited with code 0
2025-07-31 22:39:52.318 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][743c3c4a-8156-4ae9-9d6c-51de1747cfa5] socks connection closed
2025-07-31 22:39:52.318 [info] [command][39f21389-d64c-4670-baea-7968167e15d8] Socket close event received
2025-07-31 22:39:52.359 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58338 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:40:52.321 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:40:52.325 [info] [command][eef44896-0cdd-43e7-9569-715d80927691] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eef44896-0cdd-43e7-9569-715d80927691""}
2025-07-31 22:40:52.326 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2027b82a-6dbe-4acb-a377-a573be29fb34] received connection request
2025-07-31 22:40:52.327 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:40:52.372 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2027b82a-6dbe-4acb-a377-a573be29fb34] socks forwarding established
2025-07-31 22:40:52.417 [info] [command][eef44896-0cdd-43e7-9569-715d80927691] Process exited with code 0
2025-07-31 22:40:52.417 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2027b82a-6dbe-4acb-a377-a573be29fb34] socks connection closed
2025-07-31 22:40:52.417 [info] [command][eef44896-0cdd-43e7-9569-715d80927691] Socket close event received
2025-07-31 22:40:52.457 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58363 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:41:52.419 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:41:52.421 [info] [command][5e839e62-933d-4239-867a-66e498b08f27] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5e839e62-933d-4239-867a-66e498b08f27""}
2025-07-31 22:41:52.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f18454d1-6400-4e8a-b6cb-06fd3ffe4558] received connection request
2025-07-31 22:41:52.422 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:41:52.468 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f18454d1-6400-4e8a-b6cb-06fd3ffe4558] socks forwarding established
2025-07-31 22:41:52.514 [info] [command][5e839e62-933d-4239-867a-66e498b08f27] Process exited with code 0
2025-07-31 22:41:52.515 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f18454d1-6400-4e8a-b6cb-06fd3ffe4558] socks connection closed
2025-07-31 22:41:52.515 [info] [command][5e839e62-933d-4239-867a-66e498b08f27] Socket close event received
2025-07-31 22:41:52.555 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58424 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:42:52.517 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:42:52.518 [info] [command][df9191e8-62b2-4858-a9cf-d414bfff35e3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""df9191e8-62b2-4858-a9cf-d414bfff35e3""}
2025-07-31 22:42:52.518 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f951a34b-6d69-47d5-ba56-665aceb5dc2d] received connection request
2025-07-31 22:42:52.519 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:42:52.559 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f951a34b-6d69-47d5-ba56-665aceb5dc2d] socks forwarding established
2025-07-31 22:42:52.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f951a34b-6d69-47d5-ba56-665aceb5dc2d] socks connection closed
2025-07-31 22:42:52.602 [info] [command][df9191e8-62b2-4858-a9cf-d414bfff35e3] Process exited with code 0
2025-07-31 22:42:52.602 [info] [command][df9191e8-62b2-4858-a9cf-d414bfff35e3] Socket close event received
2025-07-31 22:42:52.641 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58478 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:43:52.606 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:43:52.608 [info] [command][a3962e26-5a4d-40d6-ba65-0d4b775f5a25] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a3962e26-5a4d-40d6-ba65-0d4b775f5a25""}
2025-07-31 22:43:52.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cc0012ab-37db-4a51-a1dc-bdb0448ebcbe] received connection request
2025-07-31 22:43:52.609 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:43:52.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cc0012ab-37db-4a51-a1dc-bdb0448ebcbe] socks forwarding established
2025-07-31 22:43:52.696 [info] [command][a3962e26-5a4d-40d6-ba65-0d4b775f5a25] Process exited with code 0
2025-07-31 22:43:52.696 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cc0012ab-37db-4a51-a1dc-bdb0448ebcbe] socks connection closed
2025-07-31 22:43:52.696 [info] [command][a3962e26-5a4d-40d6-ba65-0d4b775f5a25] Socket close event received
2025-07-31 22:43:52.736 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58539 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:44:52.699 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:44:52.701 [info] [command][33493d30-4c7a-49db-9ce6-d9ea4684ad85] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""33493d30-4c7a-49db-9ce6-d9ea4684ad85""}
2025-07-31 22:44:52.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][476aedae-0ad8-4cc1-bfca-cc3b06f4af31] received connection request
2025-07-31 22:44:52.702 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:44:52.746 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476aedae-0ad8-4cc1-bfca-cc3b06f4af31] socks forwarding established
2025-07-31 22:44:52.865 [info] [command][33493d30-4c7a-49db-9ce6-d9ea4684ad85] Process exited with code 0
2025-07-31 22:44:52.865 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476aedae-0ad8-4cc1-bfca-cc3b06f4af31] socks connection closed
2025-07-31 22:44:52.865 [info] [command][33493d30-4c7a-49db-9ce6-d9ea4684ad85] Socket close event received
2025-07-31 22:44:52.906 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58566 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:45:52.870 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:45:52.873 [info] [command][2daca876-c5db-4ec8-ab79-e26666d4caa9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2daca876-c5db-4ec8-ab79-e26666d4caa9""}
2025-07-31 22:45:52.874 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b21ffca7-154e-48fe-b6b9-048342cbb7ba] received connection request
2025-07-31 22:45:52.874 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:45:52.918 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b21ffca7-154e-48fe-b6b9-048342cbb7ba] socks forwarding established
2025-07-31 22:45:52.965 [info] [command][2daca876-c5db-4ec8-ab79-e26666d4caa9] Process exited with code 0
2025-07-31 22:45:52.965 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b21ffca7-154e-48fe-b6b9-048342cbb7ba] socks connection closed
2025-07-31 22:45:52.965 [info] [command][2daca876-c5db-4ec8-ab79-e26666d4caa9] Socket close event received
2025-07-31 22:45:53.007 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58591 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:46:52.965 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:46:52.966 [info] [command][e877386f-4505-4df0-aa66-47fe35d5a2fb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e877386f-4505-4df0-aa66-47fe35d5a2fb""}
2025-07-31 22:46:52.967 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e9b5d67f-7266-4ba3-8450-9794bce8b093] received connection request
2025-07-31 22:46:52.967 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:46:53.014 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9b5d67f-7266-4ba3-8450-9794bce8b093] socks forwarding established
2025-07-31 22:46:53.062 [info] [command][e877386f-4505-4df0-aa66-47fe35d5a2fb] Process exited with code 0
2025-07-31 22:46:53.062 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9b5d67f-7266-4ba3-8450-9794bce8b093] socks connection closed
2025-07-31 22:46:53.063 [info] [command][e877386f-4505-4df0-aa66-47fe35d5a2fb] Socket close event received
2025-07-31 22:46:53.109 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58622 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:47:53.065 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:47:53.067 [info] [command][0cf2f93c-ec3a-4f76-a291-aa487066dd63] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0cf2f93c-ec3a-4f76-a291-aa487066dd63""}
2025-07-31 22:47:53.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][41ca1f75-4453-4756-8b49-b90dbc374890] received connection request
2025-07-31 22:47:53.067 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 22:47:53.067 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:47:53.108 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][41ca1f75-4453-4756-8b49-b90dbc374890] socks forwarding established
2025-07-31 22:47:53.152 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][41ca1f75-4453-4756-8b49-b90dbc374890] socks connection closed
2025-07-31 22:47:53.153 [info] [command][0cf2f93c-ec3a-4f76-a291-aa487066dd63] Process exited with code 0
2025-07-31 22:47:53.153 [info] [command][0cf2f93c-ec3a-4f76-a291-aa487066dd63] Socket close event received
2025-07-31 22:47:53.192 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58679 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:48:53.155 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:48:53.161 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6b4a93a7-0d6f-419d-8aea-ff3e283a6d5b] received connection request
2025-07-31 22:48:53.161 [info] [command][f32d2c58-a0e2-4a69-a549-ce220e880c1a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f32d2c58-a0e2-4a69-a549-ce220e880c1a""}
2025-07-31 22:48:53.161 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:48:53.232 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b4a93a7-0d6f-419d-8aea-ff3e283a6d5b] socks forwarding established
2025-07-31 22:48:53.279 [info] [command][f32d2c58-a0e2-4a69-a549-ce220e880c1a] Process exited with code 0
2025-07-31 22:48:53.279 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b4a93a7-0d6f-419d-8aea-ff3e283a6d5b] socks connection closed
2025-07-31 22:48:53.279 [info] [command][f32d2c58-a0e2-4a69-a549-ce220e880c1a] Socket close event received
2025-07-31 22:48:53.318 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58710 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:49:53.280 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:49:53.282 [info] [command][8a9e01a9-1a04-4b6e-9f8c-69d7d9e5b13e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8a9e01a9-1a04-4b6e-9f8c-69d7d9e5b13e""}
2025-07-31 22:49:53.282 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fae6c213-5a6e-4e3d-885b-2023889382ea] received connection request
2025-07-31 22:49:53.282 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:49:53.321 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fae6c213-5a6e-4e3d-885b-2023889382ea] socks forwarding established
2025-07-31 22:49:53.367 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fae6c213-5a6e-4e3d-885b-2023889382ea] socks connection closed
2025-07-31 22:49:53.368 [info] [command][8a9e01a9-1a04-4b6e-9f8c-69d7d9e5b13e] Process exited with code 0
2025-07-31 22:49:53.368 [info] [command][8a9e01a9-1a04-4b6e-9f8c-69d7d9e5b13e] Socket close event received
2025-07-31 22:49:53.410 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58736 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:50:53.372 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:50:53.374 [info] [command][d234b9af-1a67-45c3-86a6-23cc13fe048a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d234b9af-1a67-45c3-86a6-23cc13fe048a""}
2025-07-31 22:50:53.374 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][af64e245-7168-461c-9d36-6e99edf71669] received connection request
2025-07-31 22:50:53.375 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:50:53.420 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][af64e245-7168-461c-9d36-6e99edf71669] socks forwarding established
2025-07-31 22:50:53.465 [info] [command][d234b9af-1a67-45c3-86a6-23cc13fe048a] Process exited with code 0
2025-07-31 22:50:53.466 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][af64e245-7168-461c-9d36-6e99edf71669] socks connection closed
2025-07-31 22:50:53.466 [info] [command][d234b9af-1a67-45c3-86a6-23cc13fe048a] Socket close event received
2025-07-31 22:50:53.509 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58769 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:51:53.471 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:51:53.473 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b19139d6-5f09-4d71-97cb-bc2ee54e85a7] received connection request
2025-07-31 22:51:53.473 [info] [command][da805eca-6795-4f5f-9aa9-6be4173bc9f6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""da805eca-6795-4f5f-9aa9-6be4173bc9f6""}
2025-07-31 22:51:53.474 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:51:53.514 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b19139d6-5f09-4d71-97cb-bc2ee54e85a7] socks forwarding established
2025-07-31 22:51:53.559 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b19139d6-5f09-4d71-97cb-bc2ee54e85a7] socks connection closed
2025-07-31 22:51:53.559 [info] [command][da805eca-6795-4f5f-9aa9-6be4173bc9f6] Process exited with code 0
2025-07-31 22:51:53.559 [info] [command][da805eca-6795-4f5f-9aa9-6be4173bc9f6] Socket close event received
2025-07-31 22:51:53.599 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58807 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:52:53.560 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:52:53.562 [info] [command][9c4d82f5-1a0f-4406-a1d6-3b6eff7b7346] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9c4d82f5-1a0f-4406-a1d6-3b6eff7b7346""}
2025-07-31 22:52:53.563 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ba83c5e8-ca33-499a-af11-8f57294e3860] received connection request
2025-07-31 22:52:53.564 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:52:53.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba83c5e8-ca33-499a-af11-8f57294e3860] socks forwarding established
2025-07-31 22:52:53.658 [info] [command][9c4d82f5-1a0f-4406-a1d6-3b6eff7b7346] Process exited with code 0
2025-07-31 22:52:53.658 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba83c5e8-ca33-499a-af11-8f57294e3860] socks connection closed
2025-07-31 22:52:53.658 [info] [command][9c4d82f5-1a0f-4406-a1d6-3b6eff7b7346] Socket close event received
2025-07-31 22:52:53.698 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58839 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:53:53.662 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:53:53.665 [info] [command][65c1058e-0f76-4ee3-9448-486eb05ccfcc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""65c1058e-0f76-4ee3-9448-486eb05ccfcc""}
2025-07-31 22:53:53.666 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1133d974-dbc7-49f2-8af5-7dfc283225af] received connection request
2025-07-31 22:53:53.667 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:53:53.730 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1133d974-dbc7-49f2-8af5-7dfc283225af] socks forwarding established
2025-07-31 22:53:53.777 [info] [command][65c1058e-0f76-4ee3-9448-486eb05ccfcc] Process exited with code 0
2025-07-31 22:53:53.777 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1133d974-dbc7-49f2-8af5-7dfc283225af] socks connection closed
2025-07-31 22:53:53.778 [info] [command][65c1058e-0f76-4ee3-9448-486eb05ccfcc] Socket close event received
2025-07-31 22:53:53.889 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58861 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:54:53.782 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:54:53.784 [info] [command][fc2487d4-3bb5-45e4-8c74-fa284020500d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fc2487d4-3bb5-45e4-8c74-fa284020500d""}
2025-07-31 22:54:53.784 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bb0c0e54-d544-45b5-8a1a-be0e2d60ace3] received connection request
2025-07-31 22:54:53.785 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 22:54:53.786 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:54:53.830 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bb0c0e54-d544-45b5-8a1a-be0e2d60ace3] socks forwarding established
2025-07-31 22:54:53.876 [info] [command][fc2487d4-3bb5-45e4-8c74-fa284020500d] Process exited with code 0
2025-07-31 22:54:53.877 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bb0c0e54-d544-45b5-8a1a-be0e2d60ace3] socks connection closed
2025-07-31 22:54:53.877 [info] [command][fc2487d4-3bb5-45e4-8c74-fa284020500d] Socket close event received
2025-07-31 22:54:53.918 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58893 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:55:53.879 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:55:53.882 [info] [command][948f1d63-8bce-4210-be39-d2b11b482a30] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""948f1d63-8bce-4210-be39-d2b11b482a30""}
2025-07-31 22:55:53.883 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][33437e6a-f254-43a3-9de7-6df3f9f7bee6] received connection request
2025-07-31 22:55:53.884 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:55:53.935 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33437e6a-f254-43a3-9de7-6df3f9f7bee6] socks forwarding established
2025-07-31 22:55:53.987 [info] [command][948f1d63-8bce-4210-be39-d2b11b482a30] Process exited with code 0
2025-07-31 22:55:53.987 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33437e6a-f254-43a3-9de7-6df3f9f7bee6] socks connection closed
2025-07-31 22:55:53.988 [info] [command][948f1d63-8bce-4210-be39-d2b11b482a30] Socket close event received
2025-07-31 22:55:54.030 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58917 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:56:53.988 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:56:53.991 [info] [command][73760288-e7ed-49de-a3d7-d162d6c8cc1b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""73760288-e7ed-49de-a3d7-d162d6c8cc1b""}
2025-07-31 22:56:53.991 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][347f89c0-a407-4636-9c84-aff74e1cb0fd] received connection request
2025-07-31 22:56:53.992 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:56:54.038 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][347f89c0-a407-4636-9c84-aff74e1cb0fd] socks forwarding established
2025-07-31 22:56:54.084 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][347f89c0-a407-4636-9c84-aff74e1cb0fd] socks connection closed
2025-07-31 22:56:54.085 [info] [command][73760288-e7ed-49de-a3d7-d162d6c8cc1b] Process exited with code 0
2025-07-31 22:56:54.085 [info] [command][73760288-e7ed-49de-a3d7-d162d6c8cc1b] Socket close event received
2025-07-31 22:56:54.196 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58962 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:57:54.088 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:57:54.089 [info] [command][d47be8e2-9ce4-4dd3-a8b2-1b8e5124ff03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d47be8e2-9ce4-4dd3-a8b2-1b8e5124ff03""}
2025-07-31 22:57:54.090 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][51696133-a3ff-471b-b035-74302d8140be] received connection request
2025-07-31 22:57:54.090 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 22:57:54.090 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:57:54.132 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][51696133-a3ff-471b-b035-74302d8140be] socks forwarding established
2025-07-31 22:57:54.178 [info] [command][d47be8e2-9ce4-4dd3-a8b2-1b8e5124ff03] Process exited with code 0
2025-07-31 22:57:54.178 [info] [command][d47be8e2-9ce4-4dd3-a8b2-1b8e5124ff03] Socket close event received
2025-07-31 22:57:54.179 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][51696133-a3ff-471b-b035-74302d8140be] socks connection closed
2025-07-31 22:57:54.218 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59035 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:58:54.181 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:58:54.182 [info] [command][f6ffdf28-c47b-4cd8-8a3b-dfb8a0d9f1ef] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f6ffdf28-c47b-4cd8-8a3b-dfb8a0d9f1ef""}
2025-07-31 22:58:54.182 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][85ad59cc-16e8-4790-a097-4512f588097a] received connection request
2025-07-31 22:58:54.182 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:58:54.304 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85ad59cc-16e8-4790-a097-4512f588097a] socks forwarding established
2025-07-31 22:58:54.349 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85ad59cc-16e8-4790-a097-4512f588097a] socks connection closed
2025-07-31 22:58:54.349 [info] [command][f6ffdf28-c47b-4cd8-8a3b-dfb8a0d9f1ef] Process exited with code 0
2025-07-31 22:58:54.349 [info] [command][f6ffdf28-c47b-4cd8-8a3b-dfb8a0d9f1ef] Socket close event received
2025-07-31 22:58:54.393 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59093 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 22:59:54.351 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 22:59:54.353 [info] [command][3479e50c-0c45-4b48-be87-d07056e87314] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3479e50c-0c45-4b48-be87-d07056e87314""}
2025-07-31 22:59:54.354 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cb4efca7-1bc6-4063-b8d0-572cd3607cc1] received connection request
2025-07-31 22:59:54.355 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 22:59:54.397 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb4efca7-1bc6-4063-b8d0-572cd3607cc1] socks forwarding established
2025-07-31 22:59:54.447 [info] [command][3479e50c-0c45-4b48-be87-d07056e87314] Process exited with code 0
2025-07-31 22:59:54.447 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb4efca7-1bc6-4063-b8d0-572cd3607cc1] socks connection closed
2025-07-31 22:59:54.447 [info] [command][3479e50c-0c45-4b48-be87-d07056e87314] Socket close event received
2025-07-31 22:59:54.487 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59120 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:00:54.450 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:00:54.453 [info] [command][743e3203-0eb9-40af-a8d2-0666647ac6c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""743e3203-0eb9-40af-a8d2-0666647ac6c7""}
2025-07-31 23:00:54.453 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2765f8a9-c1bb-46d4-9dff-ad180bc561eb] received connection request
2025-07-31 23:00:54.454 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:00:54.496 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2765f8a9-c1bb-46d4-9dff-ad180bc561eb] socks forwarding established
2025-07-31 23:00:54.542 [info] [command][743e3203-0eb9-40af-a8d2-0666647ac6c7] Process exited with code 0
2025-07-31 23:00:54.543 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2765f8a9-c1bb-46d4-9dff-ad180bc561eb] socks connection closed
2025-07-31 23:00:54.543 [info] [command][743e3203-0eb9-40af-a8d2-0666647ac6c7] Socket close event received
2025-07-31 23:00:54.583 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59147 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:01:54.548 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:01:54.549 [info] [command][aa0d46ec-a9fb-425d-805e-0e48f228924e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""aa0d46ec-a9fb-425d-805e-0e48f228924e""}
2025-07-31 23:01:54.549 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][37e341b3-fd7f-4a0d-8705-79d1c0975876] received connection request
2025-07-31 23:01:54.550 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:01:54.594 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][37e341b3-fd7f-4a0d-8705-79d1c0975876] socks forwarding established
2025-07-31 23:01:54.642 [info] [command][aa0d46ec-a9fb-425d-805e-0e48f228924e] Process exited with code 0
2025-07-31 23:01:54.642 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][37e341b3-fd7f-4a0d-8705-79d1c0975876] socks connection closed
2025-07-31 23:01:54.642 [info] [command][aa0d46ec-a9fb-425d-805e-0e48f228924e] Socket close event received
2025-07-31 23:01:54.683 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59190 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:02:54.646 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:02:54.648 [info] [command][268fff02-89c2-41fa-8e82-560fdd787f9e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""268fff02-89c2-41fa-8e82-560fdd787f9e""}
2025-07-31 23:02:54.648 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][12664046-3cfc-4f3a-85e6-e65e0b70f08d] received connection request
2025-07-31 23:02:54.648 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:02:54.691 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][12664046-3cfc-4f3a-85e6-e65e0b70f08d] socks forwarding established
2025-07-31 23:02:54.734 [info] [command][268fff02-89c2-41fa-8e82-560fdd787f9e] Process exited with code 0
2025-07-31 23:02:54.735 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][12664046-3cfc-4f3a-85e6-e65e0b70f08d] socks connection closed
2025-07-31 23:02:54.735 [info] [command][268fff02-89c2-41fa-8e82-560fdd787f9e] Socket close event received
2025-07-31 23:02:54.774 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59225 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:03:54.737 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:03:54.739 [info] [command][2b312d31-4bb2-48fa-bf03-7599e1c05635] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2b312d31-4bb2-48fa-bf03-7599e1c05635""}
2025-07-31 23:03:54.739 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d0c7a381-db33-4361-90bf-265ab9442962] received connection request
2025-07-31 23:03:54.740 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:03:54.794 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0c7a381-db33-4361-90bf-265ab9442962] socks forwarding established
2025-07-31 23:03:54.861 [info] [command][2b312d31-4bb2-48fa-bf03-7599e1c05635] Process exited with code 0
2025-07-31 23:03:54.862 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d0c7a381-db33-4361-90bf-265ab9442962] socks connection closed
2025-07-31 23:03:54.862 [info] [command][2b312d31-4bb2-48fa-bf03-7599e1c05635] Socket close event received
2025-07-31 23:03:54.904 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59240 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:04:54.864 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:04:54.866 [info] [command][f351c12a-e58c-4341-84f5-72a37516ea3b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f351c12a-e58c-4341-84f5-72a37516ea3b""}
2025-07-31 23:04:54.867 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9827bcb6-0320-494e-8ddc-7d4441198096] received connection request
2025-07-31 23:04:54.867 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:04:54.913 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9827bcb6-0320-494e-8ddc-7d4441198096] socks forwarding established
2025-07-31 23:04:54.960 [info] [command][f351c12a-e58c-4341-84f5-72a37516ea3b] Process exited with code 0
2025-07-31 23:04:54.960 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9827bcb6-0320-494e-8ddc-7d4441198096] socks connection closed
2025-07-31 23:04:54.960 [info] [command][f351c12a-e58c-4341-84f5-72a37516ea3b] Socket close event received
2025-07-31 23:04:55.002 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59287 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:05:54.962 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:05:54.965 [info] [command][9ccfa922-bd61-4293-9491-ffbc9084fe01] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9ccfa922-bd61-4293-9491-ffbc9084fe01""}
2025-07-31 23:05:54.966 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][91d81498-0310-4ea0-943a-841b76459475] received connection request
2025-07-31 23:05:54.966 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:05:55.013 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][91d81498-0310-4ea0-943a-841b76459475] socks forwarding established
2025-07-31 23:05:55.059 [info] [command][9ccfa922-bd61-4293-9491-ffbc9084fe01] Process exited with code 0
2025-07-31 23:05:55.059 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][91d81498-0310-4ea0-943a-841b76459475] socks connection closed
2025-07-31 23:05:55.060 [info] [command][9ccfa922-bd61-4293-9491-ffbc9084fe01] Socket close event received
2025-07-31 23:05:55.099 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59311 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:06:55.064 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:06:55.066 [info] [command][a0503963-3d2e-429e-8ae4-e245ec5104a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a0503963-3d2e-429e-8ae4-e245ec5104a2""}
2025-07-31 23:06:55.066 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9bb942c1-c5a6-4f64-8782-478b2007afef] received connection request
2025-07-31 23:06:55.066 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:06:55.109 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9bb942c1-c5a6-4f64-8782-478b2007afef] socks forwarding established
2025-07-31 23:06:55.155 [info] [command][a0503963-3d2e-429e-8ae4-e245ec5104a2] Process exited with code 0
2025-07-31 23:06:55.155 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9bb942c1-c5a6-4f64-8782-478b2007afef] socks connection closed
2025-07-31 23:06:55.155 [info] [command][a0503963-3d2e-429e-8ae4-e245ec5104a2] Socket close event received
2025-07-31 23:06:55.195 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59373 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:07:55.161 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:07:55.163 [info] [command][df7ecdfe-61ce-45c6-bad6-04a3431b6e9d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""df7ecdfe-61ce-45c6-bad6-04a3431b6e9d""}
2025-07-31 23:07:55.163 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bf936513-4138-4ffb-90df-2f9e8c2f8af6] received connection request
2025-07-31 23:07:55.164 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:07:55.272 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bf936513-4138-4ffb-90df-2f9e8c2f8af6] socks forwarding established
2025-07-31 23:07:55.316 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bf936513-4138-4ffb-90df-2f9e8c2f8af6] socks connection closed
2025-07-31 23:07:55.316 [info] [command][df7ecdfe-61ce-45c6-bad6-04a3431b6e9d] Process exited with code 0
2025-07-31 23:07:55.316 [info] [command][df7ecdfe-61ce-45c6-bad6-04a3431b6e9d] Socket close event received
2025-07-31 23:07:55.359 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59452 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:08:55.318 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:08:55.319 [info] [command][cc2ba3ad-8275-47d6-973c-235d85eca68c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cc2ba3ad-8275-47d6-973c-235d85eca68c""}
2025-07-31 23:08:55.319 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5617346e-bcfc-42da-8cd9-c72d72ed1032] received connection request
2025-07-31 23:08:55.319 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:08:55.359 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5617346e-bcfc-42da-8cd9-c72d72ed1032] socks forwarding established
2025-07-31 23:08:55.404 [info] [command][cc2ba3ad-8275-47d6-973c-235d85eca68c] Process exited with code 0
2025-07-31 23:08:55.404 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5617346e-bcfc-42da-8cd9-c72d72ed1032] socks connection closed
2025-07-31 23:08:55.404 [info] [command][cc2ba3ad-8275-47d6-973c-235d85eca68c] Socket close event received
2025-07-31 23:08:55.444 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59514 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:09:55.406 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:09:55.408 [info] [command][f3f74d69-0410-441e-8ab7-b465e25bb8a7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f3f74d69-0410-441e-8ab7-b465e25bb8a7""}
2025-07-31 23:09:55.408 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][38187e0d-84c1-4649-b4f8-1c3259d41d4c] received connection request
2025-07-31 23:09:55.409 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 23:09:55.409 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:09:55.452 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][38187e0d-84c1-4649-b4f8-1c3259d41d4c] socks forwarding established
2025-07-31 23:09:55.499 [info] [command][f3f74d69-0410-441e-8ab7-b465e25bb8a7] Process exited with code 0
2025-07-31 23:09:55.500 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][38187e0d-84c1-4649-b4f8-1c3259d41d4c] socks connection closed
2025-07-31 23:09:55.500 [info] [command][f3f74d69-0410-441e-8ab7-b465e25bb8a7] Socket close event received
2025-07-31 23:09:55.541 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59573 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:10:55.499 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:10:55.502 [info] [command][0e0feff4-0237-4a58-9fdb-b30585783309] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0e0feff4-0237-4a58-9fdb-b30585783309""}
2025-07-31 23:10:55.503 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][edb20163-6015-4289-bce8-3f2568dfec59] received connection request
2025-07-31 23:10:55.504 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:10:55.545 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][edb20163-6015-4289-bce8-3f2568dfec59] socks forwarding established
2025-07-31 23:10:55.592 [info] [command][0e0feff4-0237-4a58-9fdb-b30585783309] Process exited with code 0
2025-07-31 23:10:55.592 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][edb20163-6015-4289-bce8-3f2568dfec59] socks connection closed
2025-07-31 23:10:55.593 [info] [command][0e0feff4-0237-4a58-9fdb-b30585783309] Socket close event received
2025-07-31 23:10:55.636 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59625 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:11:55.594 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:11:55.596 [info] [command][39f5fba6-68ac-4457-9819-17e85437a914] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""39f5fba6-68ac-4457-9819-17e85437a914""}
2025-07-31 23:11:55.596 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][36b5ce4f-d9cb-4e14-be14-76e3a86f6423] received connection request
2025-07-31 23:11:55.597 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:11:55.639 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][36b5ce4f-d9cb-4e14-be14-76e3a86f6423] socks forwarding established
2025-07-31 23:11:55.688 [info] [command][39f5fba6-68ac-4457-9819-17e85437a914] Process exited with code 0
2025-07-31 23:11:55.689 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][36b5ce4f-d9cb-4e14-be14-76e3a86f6423] socks connection closed
2025-07-31 23:11:55.689 [info] [command][39f5fba6-68ac-4457-9819-17e85437a914] Socket close event received
2025-07-31 23:11:55.729 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59659 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:12:55.693 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:12:55.695 [info] [command][4bfc8a55-eb75-44b6-bd3a-726caf36c2e9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4bfc8a55-eb75-44b6-bd3a-726caf36c2e9""}
2025-07-31 23:12:55.696 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ab1afcfe-ae61-4d2f-bd9b-7c51fd47479f] received connection request
2025-07-31 23:12:55.696 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:12:55.738 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1afcfe-ae61-4d2f-bd9b-7c51fd47479f] socks forwarding established
2025-07-31 23:12:55.784 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1afcfe-ae61-4d2f-bd9b-7c51fd47479f] socks connection closed
2025-07-31 23:12:55.784 [info] [command][4bfc8a55-eb75-44b6-bd3a-726caf36c2e9] Process exited with code 0
2025-07-31 23:12:55.784 [info] [command][4bfc8a55-eb75-44b6-bd3a-726caf36c2e9] Socket close event received
2025-07-31 23:12:55.825 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59689 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:13:55.790 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:13:55.791 [info] [command][06a1678b-fbcf-4eac-ae63-41b5548486ce] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""06a1678b-fbcf-4eac-ae63-41b5548486ce""}
2025-07-31 23:13:55.792 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d78271cb-3891-40ee-9bbd-11bb09a1eb7e] received connection request
2025-07-31 23:13:55.793 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:13:55.837 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d78271cb-3891-40ee-9bbd-11bb09a1eb7e] socks forwarding established
2025-07-31 23:13:55.970 [info] [command][06a1678b-fbcf-4eac-ae63-41b5548486ce] Process exited with code 0
2025-07-31 23:13:55.971 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d78271cb-3891-40ee-9bbd-11bb09a1eb7e] socks connection closed
2025-07-31 23:13:55.971 [info] [command][06a1678b-fbcf-4eac-ae63-41b5548486ce] Socket close event received
2025-07-31 23:13:56.012 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59707 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:14:55.973 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:14:55.976 [info] [command][e3e81473-a34e-47fe-b423-fab16204f3b2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e3e81473-a34e-47fe-b423-fab16204f3b2""}
2025-07-31 23:14:55.977 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][458ce02a-31f5-45c5-96cd-a6d3b3c3c145] received connection request
2025-07-31 23:14:55.977 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:14:56.022 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][458ce02a-31f5-45c5-96cd-a6d3b3c3c145] socks forwarding established
2025-07-31 23:14:56.068 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][458ce02a-31f5-45c5-96cd-a6d3b3c3c145] socks connection closed
2025-07-31 23:14:56.069 [info] [command][e3e81473-a34e-47fe-b423-fab16204f3b2] Process exited with code 0
2025-07-31 23:14:56.069 [info] [command][e3e81473-a34e-47fe-b423-fab16204f3b2] Socket close event received
2025-07-31 23:14:56.109 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59722 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:15:56.070 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:15:56.076 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][53f87891-f820-4fb5-b610-02b6762e45b2] received connection request
2025-07-31 23:15:56.076 [info] [command][a90204a3-f948-45a0-8125-59a5438e887a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a90204a3-f948-45a0-8125-59a5438e887a""}
2025-07-31 23:15:56.077 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:15:56.120 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][53f87891-f820-4fb5-b610-02b6762e45b2] socks forwarding established
2025-07-31 23:15:56.165 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][53f87891-f820-4fb5-b610-02b6762e45b2] socks connection closed
2025-07-31 23:15:56.166 [info] [command][a90204a3-f948-45a0-8125-59a5438e887a] Process exited with code 0
2025-07-31 23:15:56.166 [info] [command][a90204a3-f948-45a0-8125-59a5438e887a] Socket close event received
2025-07-31 23:15:56.206 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59749 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:16:56.170 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:16:56.173 [info] [command][9a0e98ee-7bde-4a25-995a-fadb0519f667] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9a0e98ee-7bde-4a25-995a-fadb0519f667""}
2025-07-31 23:16:56.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][91e1de2f-3cf1-48aa-a206-baf78e026dcd] received connection request
2025-07-31 23:16:56.174 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 23:16:56.175 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:16:56.216 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][91e1de2f-3cf1-48aa-a206-baf78e026dcd] socks forwarding established
2025-07-31 23:16:56.261 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][91e1de2f-3cf1-48aa-a206-baf78e026dcd] socks connection closed
2025-07-31 23:16:56.261 [info] [command][9a0e98ee-7bde-4a25-995a-fadb0519f667] Process exited with code 0
2025-07-31 23:16:56.261 [info] [command][9a0e98ee-7bde-4a25-995a-fadb0519f667] Socket close event received
2025-07-31 23:16:56.371 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59842 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:17:56.263 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:17:56.265 [info] [command][4cb7f625-c939-4a11-8540-c4bce03bef03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4cb7f625-c939-4a11-8540-c4bce03bef03""}
2025-07-31 23:17:56.266 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ab1dd664-083b-4179-b008-4e71306a43e3] received connection request
2025-07-31 23:17:56.266 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:17:56.309 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1dd664-083b-4179-b008-4e71306a43e3] socks forwarding established
2025-07-31 23:17:56.353 [info] [command][4cb7f625-c939-4a11-8540-c4bce03bef03] Process exited with code 0
2025-07-31 23:17:56.354 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab1dd664-083b-4179-b008-4e71306a43e3] socks connection closed
2025-07-31 23:17:56.354 [info] [command][4cb7f625-c939-4a11-8540-c4bce03bef03] Socket close event received
2025-07-31 23:17:56.394 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59906 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:18:56.359 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:18:56.361 [info] [command][771f2e51-8124-46c3-a694-0e5b0e3ebcaa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""771f2e51-8124-46c3-a694-0e5b0e3ebcaa""}
2025-07-31 23:18:56.361 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0ebcbcf3-760c-4535-84ad-fd1ae19f8339] received connection request
2025-07-31 23:18:56.362 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:18:56.406 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ebcbcf3-760c-4535-84ad-fd1ae19f8339] socks forwarding established
2025-07-31 23:18:56.450 [info] [command][771f2e51-8124-46c3-a694-0e5b0e3ebcaa] Process exited with code 0
2025-07-31 23:18:56.451 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ebcbcf3-760c-4535-84ad-fd1ae19f8339] socks connection closed
2025-07-31 23:18:56.451 [info] [command][771f2e51-8124-46c3-a694-0e5b0e3ebcaa] Socket close event received
2025-07-31 23:18:56.492 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59931 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:19:56.454 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:19:56.456 [info] [command][41fc778b-2528-47bc-a583-279539bb0da3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""41fc778b-2528-47bc-a583-279539bb0da3""}
2025-07-31 23:19:56.457 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ff7e106c-694c-4730-b080-8572c2e925b7] received connection request
2025-07-31 23:19:56.457 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:19:56.504 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff7e106c-694c-4730-b080-8572c2e925b7] socks forwarding established
2025-07-31 23:19:56.552 [info] [command][41fc778b-2528-47bc-a583-279539bb0da3] Process exited with code 0
2025-07-31 23:19:56.553 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff7e106c-694c-4730-b080-8572c2e925b7] socks connection closed
2025-07-31 23:19:56.553 [info] [command][41fc778b-2528-47bc-a583-279539bb0da3] Socket close event received
2025-07-31 23:19:56.593 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59955 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:20:56.557 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:20:56.559 [info] [command][d593a2d6-f3e5-494d-b9ea-3c24de633a58] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d593a2d6-f3e5-494d-b9ea-3c24de633a58""}
2025-07-31 23:20:56.559 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][27e54b60-6cac-44cc-9cad-8eb3fb242ae3] received connection request
2025-07-31 23:20:56.560 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:20:56.609 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27e54b60-6cac-44cc-9cad-8eb3fb242ae3] socks forwarding established
2025-07-31 23:20:56.655 [info] [command][d593a2d6-f3e5-494d-b9ea-3c24de633a58] Process exited with code 0
2025-07-31 23:20:56.655 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27e54b60-6cac-44cc-9cad-8eb3fb242ae3] socks connection closed
2025-07-31 23:20:56.655 [info] [command][d593a2d6-f3e5-494d-b9ea-3c24de633a58] Socket close event received
2025-07-31 23:20:56.696 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 59970 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:21:56.656 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:21:56.658 [info] [command][fee3cec4-5c2f-4819-a565-59650ea2b816] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fee3cec4-5c2f-4819-a565-59650ea2b816""}
2025-07-31 23:21:56.659 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5b770488-9623-4fbd-9f30-307ee8055b40] received connection request
2025-07-31 23:21:56.659 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:21:56.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b770488-9623-4fbd-9f30-307ee8055b40] socks forwarding established
2025-07-31 23:21:56.745 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b770488-9623-4fbd-9f30-307ee8055b40] socks connection closed
2025-07-31 23:21:56.745 [info] [command][fee3cec4-5c2f-4819-a565-59650ea2b816] Process exited with code 0
2025-07-31 23:21:56.745 [info] [command][fee3cec4-5c2f-4819-a565-59650ea2b816] Socket close event received
2025-07-31 23:21:56.785 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60002 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:22:56.750 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:22:56.751 [info] [command][1ec29552-608e-44c1-81f3-043daebd1c95] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1ec29552-608e-44c1-81f3-043daebd1c95""}
2025-07-31 23:22:56.753 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cfbeb140-2f06-4e67-8b7e-5946013ebd82] received connection request
2025-07-31 23:22:56.754 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:22:56.795 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfbeb140-2f06-4e67-8b7e-5946013ebd82] socks forwarding established
2025-07-31 23:22:56.838 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfbeb140-2f06-4e67-8b7e-5946013ebd82] socks connection closed
2025-07-31 23:22:56.839 [info] [command][1ec29552-608e-44c1-81f3-043daebd1c95] Process exited with code 0
2025-07-31 23:22:56.839 [info] [command][1ec29552-608e-44c1-81f3-043daebd1c95] Socket close event received
2025-07-31 23:22:56.878 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60032 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:23:56.845 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:23:56.847 [info] [command][4862299f-4f79-4cd6-ac58-7e832262189b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4862299f-4f79-4cd6-ac58-7e832262189b""}
2025-07-31 23:23:56.848 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][14a30c5d-794e-4b0e-a952-21f92df28f7c] received connection request
2025-07-31 23:23:56.849 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:23:56.896 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14a30c5d-794e-4b0e-a952-21f92df28f7c] socks forwarding established
2025-07-31 23:23:56.941 [info] [command][4862299f-4f79-4cd6-ac58-7e832262189b] Process exited with code 0
2025-07-31 23:23:56.942 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14a30c5d-794e-4b0e-a952-21f92df28f7c] socks connection closed
2025-07-31 23:23:56.942 [info] [command][4862299f-4f79-4cd6-ac58-7e832262189b] Socket close event received
2025-07-31 23:23:56.982 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60054 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:24:56.942 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:24:56.943 [info] [command][254800aa-01b2-47e2-8bf9-0eed44ad9169] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""254800aa-01b2-47e2-8bf9-0eed44ad9169""}
2025-07-31 23:24:56.944 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0b423f55-c9c8-4acd-bc9c-4826c3e7eed8] received connection request
2025-07-31 23:24:56.944 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:24:57.076 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0b423f55-c9c8-4acd-bc9c-4826c3e7eed8] socks forwarding established
2025-07-31 23:24:57.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0b423f55-c9c8-4acd-bc9c-4826c3e7eed8] socks connection closed
2025-07-31 23:24:57.121 [info] [command][254800aa-01b2-47e2-8bf9-0eed44ad9169] Process exited with code 0
2025-07-31 23:24:57.121 [info] [command][254800aa-01b2-47e2-8bf9-0eed44ad9169] Socket close event received
2025-07-31 23:24:57.161 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60083 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:25:57.122 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:25:57.123 [info] [command][4db67598-954f-4bd0-a8ed-558f0905572d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4db67598-954f-4bd0-a8ed-558f0905572d""}
2025-07-31 23:25:57.123 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3175671a-7089-49e2-9f8f-2d30f96764a6] received connection request
2025-07-31 23:25:57.123 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:25:57.163 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3175671a-7089-49e2-9f8f-2d30f96764a6] socks forwarding established
2025-07-31 23:25:57.208 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3175671a-7089-49e2-9f8f-2d30f96764a6] socks connection closed
2025-07-31 23:25:57.208 [info] [command][4db67598-954f-4bd0-a8ed-558f0905572d] Process exited with code 0
2025-07-31 23:25:57.208 [info] [command][4db67598-954f-4bd0-a8ed-558f0905572d] Socket close event received
2025-07-31 23:25:57.247 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60136 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:26:57.209 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:26:57.212 [info] [command][746ccfe1-e867-4db1-acbb-64b97fdfb22a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""746ccfe1-e867-4db1-acbb-64b97fdfb22a""}
2025-07-31 23:26:57.212 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][67a4dc58-9d4e-422c-9ebe-fbb4a957901b] received connection request
2025-07-31 23:26:57.213 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:26:57.255 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67a4dc58-9d4e-422c-9ebe-fbb4a957901b] socks forwarding established
2025-07-31 23:26:57.298 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67a4dc58-9d4e-422c-9ebe-fbb4a957901b] socks connection closed
2025-07-31 23:26:57.298 [info] [command][746ccfe1-e867-4db1-acbb-64b97fdfb22a] Process exited with code 0
2025-07-31 23:26:57.298 [info] [command][746ccfe1-e867-4db1-acbb-64b97fdfb22a] Socket close event received
2025-07-31 23:26:57.337 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60190 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:27:57.298 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:27:57.300 [info] [command][b53427ca-1a86-4f88-8c55-dbdab38b6f4d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b53427ca-1a86-4f88-8c55-dbdab38b6f4d""}
2025-07-31 23:27:57.300 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a1c3e65b-6bd1-46a0-9267-2eb73e8dd05d] received connection request
2025-07-31 23:27:57.300 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:27:57.342 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1c3e65b-6bd1-46a0-9267-2eb73e8dd05d] socks forwarding established
2025-07-31 23:27:57.387 [info] [command][b53427ca-1a86-4f88-8c55-dbdab38b6f4d] Process exited with code 0
2025-07-31 23:27:57.388 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1c3e65b-6bd1-46a0-9267-2eb73e8dd05d] socks connection closed
2025-07-31 23:27:57.388 [info] [command][b53427ca-1a86-4f88-8c55-dbdab38b6f4d] Socket close event received
2025-07-31 23:27:57.432 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60229 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:28:57.392 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:28:57.393 [info] [command][d2379e8e-2293-496b-b15f-dd1c827ef4ca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d2379e8e-2293-496b-b15f-dd1c827ef4ca""}
2025-07-31 23:28:57.394 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5ff98186-992e-43e3-9fe5-a11a99eca5d4] received connection request
2025-07-31 23:28:57.395 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:28:57.435 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ff98186-992e-43e3-9fe5-a11a99eca5d4] socks forwarding established
2025-07-31 23:28:57.482 [info] [command][d2379e8e-2293-496b-b15f-dd1c827ef4ca] Process exited with code 0
2025-07-31 23:28:57.483 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ff98186-992e-43e3-9fe5-a11a99eca5d4] socks connection closed
2025-07-31 23:28:57.484 [info] [command][d2379e8e-2293-496b-b15f-dd1c827ef4ca] Socket close event received
2025-07-31 23:28:57.587 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60248 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:29:57.487 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:29:57.490 [info] [command][8e2fa72f-0283-4e51-a27e-d4c894998d6f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8e2fa72f-0283-4e51-a27e-d4c894998d6f""}
2025-07-31 23:29:57.491 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][596d2da3-d6a6-450f-bc8a-0207be9638e4] received connection request
2025-07-31 23:29:57.492 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:29:57.537 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][596d2da3-d6a6-450f-bc8a-0207be9638e4] socks forwarding established
2025-07-31 23:29:57.584 [info] [command][8e2fa72f-0283-4e51-a27e-d4c894998d6f] Process exited with code 0
2025-07-31 23:29:57.584 [info] [command][8e2fa72f-0283-4e51-a27e-d4c894998d6f] Socket close event received
2025-07-31 23:29:57.584 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][596d2da3-d6a6-450f-bc8a-0207be9638e4] socks connection closed
2025-07-31 23:29:57.626 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60270 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:30:57.583 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:30:57.586 [info] [command][235140cc-0967-45a3-bda8-528d07bab0ec] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""235140cc-0967-45a3-bda8-528d07bab0ec""}
2025-07-31 23:30:57.586 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ad069dd6-d9bb-4b92-bb75-4269553ddb54] received connection request
2025-07-31 23:30:57.587 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:30:57.629 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad069dd6-d9bb-4b92-bb75-4269553ddb54] socks forwarding established
2025-07-31 23:30:57.676 [info] [command][235140cc-0967-45a3-bda8-528d07bab0ec] Process exited with code 0
2025-07-31 23:30:57.676 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad069dd6-d9bb-4b92-bb75-4269553ddb54] socks connection closed
2025-07-31 23:30:57.676 [info] [command][235140cc-0967-45a3-bda8-528d07bab0ec] Socket close event received
2025-07-31 23:30:57.775 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60298 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:31:57.677 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:31:57.678 [info] [command][1ee7dc4a-161e-4404-9eb3-9ec0efb978b1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1ee7dc4a-161e-4404-9eb3-9ec0efb978b1""}
2025-07-31 23:31:57.678 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ff8ee684-3f7e-4be9-867c-ce4badc5b6cd] received connection request
2025-07-31 23:31:57.678 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:31:57.717 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff8ee684-3f7e-4be9-867c-ce4badc5b6cd] socks forwarding established
2025-07-31 23:31:57.762 [info] [command][1ee7dc4a-161e-4404-9eb3-9ec0efb978b1] Process exited with code 0
2025-07-31 23:31:57.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff8ee684-3f7e-4be9-867c-ce4badc5b6cd] socks connection closed
2025-07-31 23:31:57.762 [info] [command][1ee7dc4a-161e-4404-9eb3-9ec0efb978b1] Socket close event received
2025-07-31 23:31:57.802 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60336 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:32:57.764 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:32:57.765 [info] [command][2ba53ebc-1980-4406-b9d6-c134efc277ad] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2ba53ebc-1980-4406-b9d6-c134efc277ad""}
2025-07-31 23:32:57.765 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a16deff3-a7c6-4fb4-8e63-be827bb803b2] received connection request
2025-07-31 23:32:57.766 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:32:57.808 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a16deff3-a7c6-4fb4-8e63-be827bb803b2] socks forwarding established
2025-07-31 23:32:57.856 [info] [command][2ba53ebc-1980-4406-b9d6-c134efc277ad] Process exited with code 0
2025-07-31 23:32:57.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a16deff3-a7c6-4fb4-8e63-be827bb803b2] socks connection closed
2025-07-31 23:32:57.857 [info] [command][2ba53ebc-1980-4406-b9d6-c134efc277ad] Socket close event received
2025-07-31 23:32:57.898 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60369 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:33:57.859 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:33:57.861 [info] [command][95a66d93-47c4-4c61-bde6-529ce729d22c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""95a66d93-47c4-4c61-bde6-529ce729d22c""}
2025-07-31 23:33:57.863 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][36afb873-41cd-433f-9890-e629e88cc228] received connection request
2025-07-31 23:33:57.864 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:33:57.908 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][36afb873-41cd-433f-9890-e629e88cc228] socks forwarding established
2025-07-31 23:33:57.954 [info] [command][95a66d93-47c4-4c61-bde6-529ce729d22c] Process exited with code 0
2025-07-31 23:33:57.955 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][36afb873-41cd-433f-9890-e629e88cc228] socks connection closed
2025-07-31 23:33:57.955 [info] [command][95a66d93-47c4-4c61-bde6-529ce729d22c] Socket close event received
2025-07-31 23:33:57.995 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60393 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:34:57.957 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:34:57.958 [info] [command][cc513b71-4de0-4101-b3fa-391f89e4a35d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cc513b71-4de0-4101-b3fa-391f89e4a35d""}
2025-07-31 23:34:57.958 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5b38b251-9b21-4ecc-8f91-5315e2e4b010] received connection request
2025-07-31 23:34:57.958 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:34:57.999 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b38b251-9b21-4ecc-8f91-5315e2e4b010] socks forwarding established
2025-07-31 23:34:58.106 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b38b251-9b21-4ecc-8f91-5315e2e4b010] socks connection closed
2025-07-31 23:34:58.106 [info] [command][cc513b71-4de0-4101-b3fa-391f89e4a35d] Process exited with code 0
2025-07-31 23:34:58.106 [info] [command][cc513b71-4de0-4101-b3fa-391f89e4a35d] Socket close event received
2025-07-31 23:34:58.147 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60447 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:35:58.112 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:35:58.115 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e7fcbc22-0e18-4936-aaa1-75166a62997c] received connection request
2025-07-31 23:35:58.115 [info] [command][a619493a-5b18-485b-82a4-22222377d78b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a619493a-5b18-485b-82a4-22222377d78b""}
2025-07-31 23:35:58.115 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:35:58.157 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e7fcbc22-0e18-4936-aaa1-75166a62997c] socks forwarding established
2025-07-31 23:35:58.202 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e7fcbc22-0e18-4936-aaa1-75166a62997c] socks connection closed
2025-07-31 23:35:58.202 [info] [command][a619493a-5b18-485b-82a4-22222377d78b] Process exited with code 0
2025-07-31 23:35:58.202 [info] [command][a619493a-5b18-485b-82a4-22222377d78b] Socket close event received
2025-07-31 23:35:58.240 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60476 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:36:58.204 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:36:58.206 [info] [command][6dc07d81-7a01-4c8c-9b6a-62341a1697ba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6dc07d81-7a01-4c8c-9b6a-62341a1697ba""}
2025-07-31 23:36:58.207 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9d909de6-6986-4323-bad9-fb5c3ab86cfc] received connection request
2025-07-31 23:36:58.207 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:36:58.248 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9d909de6-6986-4323-bad9-fb5c3ab86cfc] socks forwarding established
2025-07-31 23:36:58.295 [info] [command][6dc07d81-7a01-4c8c-9b6a-62341a1697ba] Process exited with code 0
2025-07-31 23:36:58.295 [info] [command][6dc07d81-7a01-4c8c-9b6a-62341a1697ba] Socket close event received
2025-07-31 23:36:58.295 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9d909de6-6986-4323-bad9-fb5c3ab86cfc] socks connection closed
2025-07-31 23:36:58.335 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60528 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:37:58.299 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:37:58.301 [info] [command][f5971d41-3d0b-40e9-a840-d39d98d8a073] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f5971d41-3d0b-40e9-a840-d39d98d8a073""}
2025-07-31 23:37:58.302 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1264aa1a-0627-4a92-a74c-a9691b9b995a] received connection request
2025-07-31 23:37:58.303 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:37:58.348 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1264aa1a-0627-4a92-a74c-a9691b9b995a] socks forwarding established
2025-07-31 23:37:58.392 [info] [command][f5971d41-3d0b-40e9-a840-d39d98d8a073] Process exited with code 0
2025-07-31 23:37:58.393 [info] [command][f5971d41-3d0b-40e9-a840-d39d98d8a073] Socket close event received
2025-07-31 23:37:58.394 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1264aa1a-0627-4a92-a74c-a9691b9b995a] socks connection closed
2025-07-31 23:37:58.436 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60564 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:38:58.395 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:38:58.396 [info] [command][bd74df63-9e83-40be-a596-db9bf7f8d074] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bd74df63-9e83-40be-a596-db9bf7f8d074""}
2025-07-31 23:38:58.396 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][425c390e-5851-46c1-96b1-fe86ac4ca6ba] received connection request
2025-07-31 23:38:58.396 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:38:58.435 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][425c390e-5851-46c1-96b1-fe86ac4ca6ba] socks forwarding established
2025-07-31 23:38:58.478 [info] [command][bd74df63-9e83-40be-a596-db9bf7f8d074] Process exited with code 0
2025-07-31 23:38:58.479 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][425c390e-5851-46c1-96b1-fe86ac4ca6ba] socks connection closed
2025-07-31 23:38:58.479 [info] [command][bd74df63-9e83-40be-a596-db9bf7f8d074] Socket close event received
2025-07-31 23:38:58.518 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60607 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:39:58.482 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:39:58.484 [info] [command][c5e500f1-9c54-4b92-a868-1cba8dac4172] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c5e500f1-9c54-4b92-a868-1cba8dac4172""}
2025-07-31 23:39:58.485 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][08d27d39-c9c3-406e-98ef-fbbf47cfed38] received connection request
2025-07-31 23:39:58.485 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:39:58.530 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][08d27d39-c9c3-406e-98ef-fbbf47cfed38] socks forwarding established
2025-07-31 23:39:58.665 [info] [command][c5e500f1-9c54-4b92-a868-1cba8dac4172] Process exited with code 0
2025-07-31 23:39:58.665 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][08d27d39-c9c3-406e-98ef-fbbf47cfed38] socks connection closed
2025-07-31 23:39:58.666 [info] [command][c5e500f1-9c54-4b92-a868-1cba8dac4172] Socket close event received
2025-07-31 23:39:58.705 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60638 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:40:58.670 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:40:58.671 [info] [command][d8403d98-13a4-451f-be91-b6d6710bc692] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d8403d98-13a4-451f-be91-b6d6710bc692""}
2025-07-31 23:40:58.671 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b507a008-c156-4260-b135-eed005f976b5] received connection request
2025-07-31 23:40:58.671 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 23:40:58.672 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:40:58.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b507a008-c156-4260-b135-eed005f976b5] socks forwarding established
2025-07-31 23:40:58.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b507a008-c156-4260-b135-eed005f976b5] socks connection closed
2025-07-31 23:40:58.762 [info] [command][d8403d98-13a4-451f-be91-b6d6710bc692] Process exited with code 0
2025-07-31 23:40:58.762 [info] [command][d8403d98-13a4-451f-be91-b6d6710bc692] Socket close event received
2025-07-31 23:40:58.801 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60654 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:41:58.764 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:41:58.765 [info] [command][b0d8453b-e676-4c19-a4ca-030c097f6855] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b0d8453b-e676-4c19-a4ca-030c097f6855""}
2025-07-31 23:41:58.766 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][29a46b9d-28d8-4907-a669-5c9ae8954e18] received connection request
2025-07-31 23:41:58.766 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:41:58.808 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29a46b9d-28d8-4907-a669-5c9ae8954e18] socks forwarding established
2025-07-31 23:41:58.852 [info] [command][b0d8453b-e676-4c19-a4ca-030c097f6855] Process exited with code 0
2025-07-31 23:41:58.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29a46b9d-28d8-4907-a669-5c9ae8954e18] socks connection closed
2025-07-31 23:41:58.853 [info] [command][b0d8453b-e676-4c19-a4ca-030c097f6855] Socket close event received
2025-07-31 23:41:58.892 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60695 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:42:58.854 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:42:58.855 [info] [command][04542b42-482a-459e-aa7f-8088321d0bf2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""04542b42-482a-459e-aa7f-8088321d0bf2""}
2025-07-31 23:42:58.856 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][777e6821-e66d-44fa-9e11-d14cfd184ed5] received connection request
2025-07-31 23:42:58.857 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:42:58.897 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][777e6821-e66d-44fa-9e11-d14cfd184ed5] socks forwarding established
2025-07-31 23:42:59.025 [info] [command][04542b42-482a-459e-aa7f-8088321d0bf2] Process exited with code 0
2025-07-31 23:42:59.025 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][777e6821-e66d-44fa-9e11-d14cfd184ed5] socks connection closed
2025-07-31 23:42:59.025 [info] [command][04542b42-482a-459e-aa7f-8088321d0bf2] Socket close event received
2025-07-31 23:42:59.067 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60728 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:43:59.030 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:43:59.032 [info] [command][b9aeeef7-51d5-4cba-ad0b-b4ddd80c4ab4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b9aeeef7-51d5-4cba-ad0b-b4ddd80c4ab4""}
2025-07-31 23:43:59.032 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6289e7d3-667e-4bdf-ac00-393df2586656] received connection request
2025-07-31 23:43:59.033 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:43:59.074 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6289e7d3-667e-4bdf-ac00-393df2586656] socks forwarding established
2025-07-31 23:43:59.118 [info] [command][b9aeeef7-51d5-4cba-ad0b-b4ddd80c4ab4] Process exited with code 0
2025-07-31 23:43:59.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6289e7d3-667e-4bdf-ac00-393df2586656] socks connection closed
2025-07-31 23:43:59.118 [info] [command][b9aeeef7-51d5-4cba-ad0b-b4ddd80c4ab4] Socket close event received
2025-07-31 23:43:59.158 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60760 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:44:59.122 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:44:59.123 [info] [command][343806ae-e835-49c0-b5b8-da4680e224b1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""343806ae-e835-49c0-b5b8-da4680e224b1""}
2025-07-31 23:44:59.124 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1f40d02d-2553-4a10-9e3f-dd6c59877840] received connection request
2025-07-31 23:44:59.125 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:44:59.167 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f40d02d-2553-4a10-9e3f-dd6c59877840] socks forwarding established
2025-07-31 23:44:59.215 [info] [command][343806ae-e835-49c0-b5b8-da4680e224b1] Process exited with code 0
2025-07-31 23:44:59.216 [info] [command][343806ae-e835-49c0-b5b8-da4680e224b1] Socket close event received
2025-07-31 23:44:59.216 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f40d02d-2553-4a10-9e3f-dd6c59877840] socks connection closed
2025-07-31 23:44:59.257 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60787 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:45:59.215 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:45:59.217 [info] [command][f18c99fb-9608-40ff-a76a-a7199f395a85] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f18c99fb-9608-40ff-a76a-a7199f395a85""}
2025-07-31 23:45:59.217 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a811f293-d83e-4015-9ffc-2f0c58f8ca73] received connection request
2025-07-31 23:45:59.217 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:45:59.282 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a811f293-d83e-4015-9ffc-2f0c58f8ca73] socks forwarding established
2025-07-31 23:45:59.329 [info] [command][f18c99fb-9608-40ff-a76a-a7199f395a85] Process exited with code 0
2025-07-31 23:45:59.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a811f293-d83e-4015-9ffc-2f0c58f8ca73] socks connection closed
2025-07-31 23:45:59.330 [info] [command][f18c99fb-9608-40ff-a76a-a7199f395a85] Socket close event received
2025-07-31 23:45:59.372 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60814 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:46:59.335 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:46:59.337 [info] [command][5ecd2a65-0374-4ee6-afb2-120e4cfbf6d2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5ecd2a65-0374-4ee6-afb2-120e4cfbf6d2""}
2025-07-31 23:46:59.338 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][97de11ad-d8b8-4b21-b16d-38f2b9f42d6b] received connection request
2025-07-31 23:46:59.338 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:46:59.383 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97de11ad-d8b8-4b21-b16d-38f2b9f42d6b] socks forwarding established
2025-07-31 23:46:59.427 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97de11ad-d8b8-4b21-b16d-38f2b9f42d6b] socks connection closed
2025-07-31 23:46:59.427 [info] [command][5ecd2a65-0374-4ee6-afb2-120e4cfbf6d2] Process exited with code 0
2025-07-31 23:46:59.427 [info] [command][5ecd2a65-0374-4ee6-afb2-120e4cfbf6d2] Socket close event received
2025-07-31 23:46:59.465 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60847 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:47:59.431 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:47:59.433 [info] [command][727dad9c-4861-4d43-9245-5f98ba1ac2ef] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""727dad9c-4861-4d43-9245-5f98ba1ac2ef""}
2025-07-31 23:47:59.434 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][abe18633-805b-4423-8002-3582cbbcf679] received connection request
2025-07-31 23:47:59.436 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:47:59.478 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][abe18633-805b-4423-8002-3582cbbcf679] socks forwarding established
2025-07-31 23:47:59.524 [info] [command][727dad9c-4861-4d43-9245-5f98ba1ac2ef] Process exited with code 0
2025-07-31 23:47:59.525 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][abe18633-805b-4423-8002-3582cbbcf679] socks connection closed
2025-07-31 23:47:59.525 [info] [command][727dad9c-4861-4d43-9245-5f98ba1ac2ef] Socket close event received
2025-07-31 23:47:59.565 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60904 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:48:59.529 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:48:59.531 [info] [command][77f0aea3-2697-4aa3-8505-211f42d0df23] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""77f0aea3-2697-4aa3-8505-211f42d0df23""}
2025-07-31 23:48:59.532 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][00a40f47-9977-47b4-8edf-c32cfa51eae7] received connection request
2025-07-31 23:48:59.532 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:48:59.575 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][00a40f47-9977-47b4-8edf-c32cfa51eae7] socks forwarding established
2025-07-31 23:48:59.620 [info] [command][77f0aea3-2697-4aa3-8505-211f42d0df23] Process exited with code 0
2025-07-31 23:48:59.621 [info] [command][77f0aea3-2697-4aa3-8505-211f42d0df23] Socket close event received
2025-07-31 23:48:59.621 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][00a40f47-9977-47b4-8edf-c32cfa51eae7] socks connection closed
2025-07-31 23:48:59.735 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60923 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:49:59.625 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:49:59.627 [info] [command][dd8e7559-0cbe-4301-b3bd-6274d7329b32] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dd8e7559-0cbe-4301-b3bd-6274d7329b32""}
2025-07-31 23:49:59.628 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][897c48f9-cdfb-4ef9-9ba8-f6870e6d4d7d] received connection request
2025-07-31 23:49:59.628 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:49:59.669 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][897c48f9-cdfb-4ef9-9ba8-f6870e6d4d7d] socks forwarding established
2025-07-31 23:49:59.714 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][897c48f9-cdfb-4ef9-9ba8-f6870e6d4d7d] socks connection closed
2025-07-31 23:49:59.714 [info] [command][dd8e7559-0cbe-4301-b3bd-6274d7329b32] Process exited with code 0
2025-07-31 23:49:59.714 [info] [command][dd8e7559-0cbe-4301-b3bd-6274d7329b32] Socket close event received
2025-07-31 23:49:59.759 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60956 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:50:59.720 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:50:59.722 [info] [command][c6f59e84-5ea3-474f-b16b-64a9decdeac1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c6f59e84-5ea3-474f-b16b-64a9decdeac1""}
2025-07-31 23:50:59.723 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][abac9518-2f08-4086-b2e4-e8734ef7cf0b] received connection request
2025-07-31 23:50:59.725 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:50:59.774 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][abac9518-2f08-4086-b2e4-e8734ef7cf0b] socks forwarding established
2025-07-31 23:50:59.820 [info] [command][c6f59e84-5ea3-474f-b16b-64a9decdeac1] Process exited with code 0
2025-07-31 23:50:59.820 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][abac9518-2f08-4086-b2e4-e8734ef7cf0b] socks connection closed
2025-07-31 23:50:59.821 [info] [command][c6f59e84-5ea3-474f-b16b-64a9decdeac1] Socket close event received
2025-07-31 23:50:59.860 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 60988 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:51:59.823 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:51:59.826 [info] [command][d5df62ad-79a3-49aa-9f03-74a26462d0fb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d5df62ad-79a3-49aa-9f03-74a26462d0fb""}
2025-07-31 23:51:59.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3c39ebd-8a01-4191-8291-b15532aea8ba] received connection request
2025-07-31 23:51:59.827 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:51:59.869 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3c39ebd-8a01-4191-8291-b15532aea8ba] socks forwarding established
2025-07-31 23:51:59.915 [info] [command][d5df62ad-79a3-49aa-9f03-74a26462d0fb] Process exited with code 0
2025-07-31 23:51:59.915 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3c39ebd-8a01-4191-8291-b15532aea8ba] socks connection closed
2025-07-31 23:51:59.915 [info] [command][d5df62ad-79a3-49aa-9f03-74a26462d0fb] Socket close event received
2025-07-31 23:51:59.954 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61031 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:52:59.920 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:52:59.921 [info] [command][f300c21a-12ac-4b58-9f2f-fc2f50f2e2d6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f300c21a-12ac-4b58-9f2f-fc2f50f2e2d6""}
2025-07-31 23:52:59.921 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][794d9900-c377-465a-8201-aacea4c4947e] received connection request
2025-07-31 23:52:59.922 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 23:52:59.922 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:52:59.964 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][794d9900-c377-465a-8201-aacea4c4947e] socks forwarding established
2025-07-31 23:53:00.008 [info] [command][f300c21a-12ac-4b58-9f2f-fc2f50f2e2d6] Process exited with code 0
2025-07-31 23:53:00.008 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][794d9900-c377-465a-8201-aacea4c4947e] socks connection closed
2025-07-31 23:53:00.008 [info] [command][f300c21a-12ac-4b58-9f2f-fc2f50f2e2d6] Socket close event received
2025-07-31 23:53:00.049 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61070 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:54:00.013 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:54:00.015 [info] [command][7993dd37-d3c5-4b8c-ad91-92faeef0013e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7993dd37-d3c5-4b8c-ad91-92faeef0013e""}
2025-07-31 23:54:00.017 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][15f4d441-6dde-42d9-9317-10ad5b738d71] received connection request
2025-07-31 23:54:00.017 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:54:00.059 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][15f4d441-6dde-42d9-9317-10ad5b738d71] socks forwarding established
2025-07-31 23:54:00.102 [info] [command][7993dd37-d3c5-4b8c-ad91-92faeef0013e] Process exited with code 0
2025-07-31 23:54:00.103 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][15f4d441-6dde-42d9-9317-10ad5b738d71] socks connection closed
2025-07-31 23:54:00.103 [info] [command][7993dd37-d3c5-4b8c-ad91-92faeef0013e] Socket close event received
2025-07-31 23:54:00.147 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61090 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:55:00.104 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:55:00.105 [info] [command][0ae862c3-a37c-44a1-a40f-eb009adac73b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0ae862c3-a37c-44a1-a40f-eb009adac73b""}
2025-07-31 23:55:00.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fac6409b-8a7e-490b-9289-75e30e41cae5] received connection request
2025-07-31 23:55:00.106 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:55:00.151 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fac6409b-8a7e-490b-9289-75e30e41cae5] socks forwarding established
2025-07-31 23:55:00.226 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fac6409b-8a7e-490b-9289-75e30e41cae5] socks connection closed
2025-07-31 23:55:00.226 [info] [command][0ae862c3-a37c-44a1-a40f-eb009adac73b] Process exited with code 0
2025-07-31 23:55:00.226 [info] [command][0ae862c3-a37c-44a1-a40f-eb009adac73b] Socket close event received
2025-07-31 23:55:00.272 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61117 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:56:00.227 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:56:00.228 [info] [command][7f8529f2-24ee-4d2a-bc97-12b5b05bf3ca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7f8529f2-24ee-4d2a-bc97-12b5b05bf3ca""}
2025-07-31 23:56:00.228 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a9746cb7-8725-407c-a36d-6773aaea4cca] received connection request
2025-07-31 23:56:00.228 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-07-31 23:56:00.228 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:56:00.270 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a9746cb7-8725-407c-a36d-6773aaea4cca] socks forwarding established
2025-07-31 23:56:00.315 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a9746cb7-8725-407c-a36d-6773aaea4cca] socks connection closed
2025-07-31 23:56:00.315 [info] [command][7f8529f2-24ee-4d2a-bc97-12b5b05bf3ca] Process exited with code 0
2025-07-31 23:56:00.315 [info] [command][7f8529f2-24ee-4d2a-bc97-12b5b05bf3ca] Socket close event received
2025-07-31 23:56:00.356 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61155 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:57:00.318 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:57:00.319 [info] [command][ac3330ee-04f8-4f21-8737-f9f8bb70e68d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ac3330ee-04f8-4f21-8737-f9f8bb70e68d""}
2025-07-31 23:57:00.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c102e438-aa4d-469e-bcf2-63719e704723] received connection request
2025-07-31 23:57:00.321 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:57:00.363 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c102e438-aa4d-469e-bcf2-63719e704723] socks forwarding established
2025-07-31 23:57:00.409 [info] [command][ac3330ee-04f8-4f21-8737-f9f8bb70e68d] Process exited with code 0
2025-07-31 23:57:00.409 [info] [command][ac3330ee-04f8-4f21-8737-f9f8bb70e68d] Socket close event received
2025-07-31 23:57:00.410 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c102e438-aa4d-469e-bcf2-63719e704723] socks connection closed
2025-07-31 23:57:00.450 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61195 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:58:00.411 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:58:00.414 [info] [command][7d8cd8af-219b-4072-a933-68d9a2e9bc13] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7d8cd8af-219b-4072-a933-68d9a2e9bc13""}
2025-07-31 23:58:00.415 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0441fb85-2d7f-4992-aa84-e5b9c2be8c75] received connection request
2025-07-31 23:58:00.416 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:58:00.458 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0441fb85-2d7f-4992-aa84-e5b9c2be8c75] socks forwarding established
2025-07-31 23:58:00.503 [info] [command][7d8cd8af-219b-4072-a933-68d9a2e9bc13] Process exited with code 0
2025-07-31 23:58:00.503 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0441fb85-2d7f-4992-aa84-e5b9c2be8c75] socks connection closed
2025-07-31 23:58:00.503 [info] [command][7d8cd8af-219b-4072-a933-68d9a2e9bc13] Socket close event received
2025-07-31 23:58:00.543 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61228 to 127.0.0.1 port 53495, nchannels 6

2025-07-31 23:59:00.509 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-07-31 23:59:00.511 [info] [command][a6e839a5-bb74-4268-ae97-d552b16118e4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a6e839a5-bb74-4268-ae97-d552b16118e4""}
2025-07-31 23:59:00.512 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][39488625-e18e-4532-af6b-75cc1f354dcc] received connection request
2025-07-31 23:59:00.513 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-07-31 23:59:00.592 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39488625-e18e-4532-af6b-75cc1f354dcc] socks forwarding established
2025-07-31 23:59:00.637 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39488625-e18e-4532-af6b-75cc1f354dcc] socks connection closed
2025-07-31 23:59:00.638 [info] [command][a6e839a5-bb74-4268-ae97-d552b16118e4] Process exited with code 0
2025-07-31 23:59:00.638 [info] [command][a6e839a5-bb74-4268-ae97-d552b16118e4] Socket close event received
2025-07-31 23:59:00.679 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61264 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:00:00.642 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:00:00.645 [info] [command][a3e4b686-9559-4eac-8055-0a8aa7bbe977] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a3e4b686-9559-4eac-8055-0a8aa7bbe977""}
2025-08-01 00:00:00.645 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b2a06a93-1a7a-4ef6-b192-1856b4c99f17] received connection request
2025-08-01 00:00:00.646 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:00:00.687 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2a06a93-1a7a-4ef6-b192-1856b4c99f17] socks forwarding established
2025-08-01 00:00:00.736 [info] [command][a3e4b686-9559-4eac-8055-0a8aa7bbe977] Process exited with code 0
2025-08-01 00:00:00.736 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2a06a93-1a7a-4ef6-b192-1856b4c99f17] socks connection closed
2025-08-01 00:00:00.736 [info] [command][a3e4b686-9559-4eac-8055-0a8aa7bbe977] Socket close event received
2025-08-01 00:00:00.780 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61286 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:01:00.740 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:01:00.741 [info] [command][59c2ada6-9d52-428d-97f4-8b66a4fab50f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""59c2ada6-9d52-428d-97f4-8b66a4fab50f""}
2025-08-01 00:01:00.741 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][87930a88-31d9-4109-8564-b1c0b36e778a] received connection request
2025-08-01 00:01:00.741 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:01:00.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87930a88-31d9-4109-8564-b1c0b36e778a] socks forwarding established
2025-08-01 00:01:00.826 [info] [command][59c2ada6-9d52-428d-97f4-8b66a4fab50f] Process exited with code 0
2025-08-01 00:01:00.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][87930a88-31d9-4109-8564-b1c0b36e778a] socks connection closed
2025-08-01 00:01:00.828 [info] [command][59c2ada6-9d52-428d-97f4-8b66a4fab50f] Socket close event received
2025-08-01 00:01:00.870 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61322 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:02:00.827 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:02:00.828 [info] [command][188e7535-06f7-4e9a-93ad-ad4e0cd45ade] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""188e7535-06f7-4e9a-93ad-ad4e0cd45ade""}
2025-08-01 00:02:00.829 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6b758e2f-c3d7-4275-a618-89a3aa924066] received connection request
2025-08-01 00:02:00.830 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:02:00.873 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b758e2f-c3d7-4275-a618-89a3aa924066] socks forwarding established
2025-08-01 00:02:00.917 [info] [command][188e7535-06f7-4e9a-93ad-ad4e0cd45ade] Process exited with code 0
2025-08-01 00:02:00.917 [info] [command][188e7535-06f7-4e9a-93ad-ad4e0cd45ade] Socket close event received
2025-08-01 00:02:00.918 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b758e2f-c3d7-4275-a618-89a3aa924066] socks connection closed
2025-08-01 00:02:00.959 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61363 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:03:00.920 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:03:00.924 [info] [command][769b260b-408a-411e-a33a-23d79863db16] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""769b260b-408a-411e-a33a-23d79863db16""}
2025-08-01 00:03:00.925 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bf5cae43-a905-47bf-a965-e765691bf6cb] received connection request
2025-08-01 00:03:00.925 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:03:00.971 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bf5cae43-a905-47bf-a965-e765691bf6cb] socks forwarding established
2025-08-01 00:03:01.014 [info] [command][769b260b-408a-411e-a33a-23d79863db16] Process exited with code 0
2025-08-01 00:03:01.014 [info] [command][769b260b-408a-411e-a33a-23d79863db16] Socket close event received
2025-08-01 00:03:01.014 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bf5cae43-a905-47bf-a965-e765691bf6cb] socks connection closed
2025-08-01 00:03:01.054 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61394 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:04:01.019 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:04:01.020 [info] [command][aaccfb73-0ac4-48f3-9777-057f2f0ea0e4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""aaccfb73-0ac4-48f3-9777-057f2f0ea0e4""}
2025-08-01 00:04:01.021 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dc50575a-205a-4e2c-8921-7a94367a1a45] received connection request
2025-08-01 00:04:01.021 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 00:04:01.021 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:04:01.066 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dc50575a-205a-4e2c-8921-7a94367a1a45] socks forwarding established
2025-08-01 00:04:01.112 [info] [command][aaccfb73-0ac4-48f3-9777-057f2f0ea0e4] Process exited with code 0
2025-08-01 00:04:01.112 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dc50575a-205a-4e2c-8921-7a94367a1a45] socks connection closed
2025-08-01 00:04:01.112 [info] [command][aaccfb73-0ac4-48f3-9777-057f2f0ea0e4] Socket close event received
2025-08-01 00:04:01.152 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61410 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:05:01.117 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:05:01.118 [info] [command][7382d3da-d202-42dd-aa35-055dd31a8269] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7382d3da-d202-42dd-aa35-055dd31a8269""}
2025-08-01 00:05:01.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d430138b-8af7-4386-8f3c-2dae39831909] received connection request
2025-08-01 00:05:01.118 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:05:01.161 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d430138b-8af7-4386-8f3c-2dae39831909] socks forwarding established
2025-08-01 00:05:01.206 [info] [command][7382d3da-d202-42dd-aa35-055dd31a8269] Process exited with code 0
2025-08-01 00:05:01.206 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d430138b-8af7-4386-8f3c-2dae39831909] socks connection closed
2025-08-01 00:05:01.206 [info] [command][7382d3da-d202-42dd-aa35-055dd31a8269] Socket close event received
2025-08-01 00:05:01.245 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61424 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:06:01.208 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:06:01.208 [info] [command][9c17426b-9990-494b-b3f1-4d95880fd12a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9c17426b-9990-494b-b3f1-4d95880fd12a""}
2025-08-01 00:06:01.209 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][840980c0-265c-4650-b7ff-242b025b6a30] received connection request
2025-08-01 00:06:01.209 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 00:06:01.209 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:06:01.258 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][840980c0-265c-4650-b7ff-242b025b6a30] socks forwarding established
2025-08-01 00:06:01.303 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][840980c0-265c-4650-b7ff-242b025b6a30] socks connection closed
2025-08-01 00:06:01.303 [info] [command][9c17426b-9990-494b-b3f1-4d95880fd12a] Process exited with code 0
2025-08-01 00:06:01.303 [info] [command][9c17426b-9990-494b-b3f1-4d95880fd12a] Socket close event received
2025-08-01 00:06:01.342 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61439 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:07:01.303 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:07:01.306 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9884c38f-8341-4424-9892-c314bb5ac2b4] received connection request
2025-08-01 00:07:01.306 [info] [command][42a76ba2-487f-4b87-b136-6d164ef1ec16] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""42a76ba2-487f-4b87-b136-6d164ef1ec16""}
2025-08-01 00:07:01.307 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:07:01.350 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9884c38f-8341-4424-9892-c314bb5ac2b4] socks forwarding established
2025-08-01 00:07:01.396 [info] [command][42a76ba2-487f-4b87-b136-6d164ef1ec16] Process exited with code 0
2025-08-01 00:07:01.396 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9884c38f-8341-4424-9892-c314bb5ac2b4] socks connection closed
2025-08-01 00:07:01.397 [info] [command][42a76ba2-487f-4b87-b136-6d164ef1ec16] Socket close event received
2025-08-01 00:07:01.446 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61483 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:08:01.401 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:08:01.402 [info] [command][8151f5a5-fa54-4af9-bfc5-5ecd66f86ee6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8151f5a5-fa54-4af9-bfc5-5ecd66f86ee6""}
2025-08-01 00:08:01.403 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][951cffbe-456e-450b-9fbf-6c3482befb30] received connection request
2025-08-01 00:08:01.403 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:08:01.477 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][951cffbe-456e-450b-9fbf-6c3482befb30] socks forwarding established
2025-08-01 00:08:01.523 [info] [command][8151f5a5-fa54-4af9-bfc5-5ecd66f86ee6] Process exited with code 0
2025-08-01 00:08:01.523 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][951cffbe-456e-450b-9fbf-6c3482befb30] socks connection closed
2025-08-01 00:08:01.523 [info] [command][8151f5a5-fa54-4af9-bfc5-5ecd66f86ee6] Socket close event received
2025-08-01 00:08:01.565 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61551 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:09:01.527 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:09:01.529 [info] [command][1afc69df-f171-480f-8360-f2ab827fd16d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1afc69df-f171-480f-8360-f2ab827fd16d""}
2025-08-01 00:09:01.530 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][01fb3d2f-3686-43a9-908a-42e20873b599] received connection request
2025-08-01 00:09:01.530 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:09:01.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][01fb3d2f-3686-43a9-908a-42e20873b599] socks forwarding established
2025-08-01 00:09:01.617 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][01fb3d2f-3686-43a9-908a-42e20873b599] socks connection closed
2025-08-01 00:09:01.617 [info] [command][1afc69df-f171-480f-8360-f2ab827fd16d] Process exited with code 0
2025-08-01 00:09:01.617 [info] [command][1afc69df-f171-480f-8360-f2ab827fd16d] Socket close event received
2025-08-01 00:09:01.658 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61606 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:10:01.622 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:10:01.624 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f842c6bb-6049-4681-bf63-af410cac7bcf] received connection request
2025-08-01 00:10:01.625 [info] [command][7056b372-33a9-4316-bfad-055ec8860170] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7056b372-33a9-4316-bfad-055ec8860170""}
2025-08-01 00:10:01.625 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:10:01.674 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f842c6bb-6049-4681-bf63-af410cac7bcf] socks forwarding established
2025-08-01 00:10:01.722 [info] [command][7056b372-33a9-4316-bfad-055ec8860170] Process exited with code 0
2025-08-01 00:10:01.722 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f842c6bb-6049-4681-bf63-af410cac7bcf] socks connection closed
2025-08-01 00:10:01.722 [info] [command][7056b372-33a9-4316-bfad-055ec8860170] Socket close event received
2025-08-01 00:10:01.761 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61664 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:11:01.727 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:11:01.729 [info] [command][690eb2fd-cb3f-4184-a369-985243287cb8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""690eb2fd-cb3f-4184-a369-985243287cb8""}
2025-08-01 00:11:01.730 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bdfd5cf9-9935-4149-ba16-82958a2623a6] received connection request
2025-08-01 00:11:01.731 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:11:01.788 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bdfd5cf9-9935-4149-ba16-82958a2623a6] socks forwarding established
2025-08-01 00:11:01.832 [info] [command][690eb2fd-cb3f-4184-a369-985243287cb8] Process exited with code 0
2025-08-01 00:11:01.832 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bdfd5cf9-9935-4149-ba16-82958a2623a6] socks connection closed
2025-08-01 00:11:01.832 [info] [command][690eb2fd-cb3f-4184-a369-985243287cb8] Socket close event received
2025-08-01 00:11:01.873 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61719 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:12:01.835 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:12:01.837 [info] [command][c6675ebd-9a26-4461-bcf1-28731e3bd434] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c6675ebd-9a26-4461-bcf1-28731e3bd434""}
2025-08-01 00:12:01.838 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][179f1b1a-2c58-4fab-9f28-8474957984ee] received connection request
2025-08-01 00:12:01.839 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:12:01.885 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][179f1b1a-2c58-4fab-9f28-8474957984ee] socks forwarding established
2025-08-01 00:12:01.934 [info] [command][c6675ebd-9a26-4461-bcf1-28731e3bd434] Process exited with code 0
2025-08-01 00:12:01.935 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][179f1b1a-2c58-4fab-9f28-8474957984ee] socks connection closed
2025-08-01 00:12:01.935 [info] [command][c6675ebd-9a26-4461-bcf1-28731e3bd434] Socket close event received
2025-08-01 00:12:01.975 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61768 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:13:01.937 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:13:01.939 [info] [command][6ea9a962-658f-4f02-88a4-38ac0c333013] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6ea9a962-658f-4f02-88a4-38ac0c333013""}
2025-08-01 00:13:01.940 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4ed3fb64-906d-484b-b9dd-30c08f7eaec8] received connection request
2025-08-01 00:13:01.940 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:13:01.983 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4ed3fb64-906d-484b-b9dd-30c08f7eaec8] socks forwarding established
2025-08-01 00:13:02.030 [info] [command][6ea9a962-658f-4f02-88a4-38ac0c333013] Process exited with code 0
2025-08-01 00:13:02.030 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4ed3fb64-906d-484b-b9dd-30c08f7eaec8] socks connection closed
2025-08-01 00:13:02.030 [info] [command][6ea9a962-658f-4f02-88a4-38ac0c333013] Socket close event received
2025-08-01 00:13:02.069 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61801 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:14:02.033 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:14:02.035 [info] [command][9a7caaaf-2b7f-48fd-b10c-b746d97f4b6a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9a7caaaf-2b7f-48fd-b10c-b746d97f4b6a""}
2025-08-01 00:14:02.035 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d53df853-01a8-473f-a17d-1fd2034d569f] received connection request
2025-08-01 00:14:02.036 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:14:02.085 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d53df853-01a8-473f-a17d-1fd2034d569f] socks forwarding established
2025-08-01 00:14:02.130 [info] [command][9a7caaaf-2b7f-48fd-b10c-b746d97f4b6a] Process exited with code 0
2025-08-01 00:14:02.131 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d53df853-01a8-473f-a17d-1fd2034d569f] socks connection closed
2025-08-01 00:14:02.131 [info] [command][9a7caaaf-2b7f-48fd-b10c-b746d97f4b6a] Socket close event received
2025-08-01 00:14:02.170 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61817 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:15:02.136 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:15:02.138 [info] [command][8ac1ca29-451e-428e-81fa-373856be0885] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8ac1ca29-451e-428e-81fa-373856be0885""}
2025-08-01 00:15:02.139 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e525288a-02cc-4767-a361-c4ff7ab9482e] received connection request
2025-08-01 00:15:02.139 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:15:02.307 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e525288a-02cc-4767-a361-c4ff7ab9482e] socks forwarding established
2025-08-01 00:15:02.351 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e525288a-02cc-4767-a361-c4ff7ab9482e] socks connection closed
2025-08-01 00:15:02.351 [info] [command][8ac1ca29-451e-428e-81fa-373856be0885] Process exited with code 0
2025-08-01 00:15:02.351 [info] [command][8ac1ca29-451e-428e-81fa-373856be0885] Socket close event received
2025-08-01 00:15:02.392 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61831 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:16:02.354 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:16:02.356 [info] [command][6ff386fb-7eae-4cbc-9a7f-5320647f4176] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6ff386fb-7eae-4cbc-9a7f-5320647f4176""}
2025-08-01 00:16:02.357 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5c79e43f-4109-43a9-896a-6e96d98d681e] received connection request
2025-08-01 00:16:02.358 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:16:02.400 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c79e43f-4109-43a9-896a-6e96d98d681e] socks forwarding established
2025-08-01 00:16:02.446 [info] [command][6ff386fb-7eae-4cbc-9a7f-5320647f4176] Process exited with code 0
2025-08-01 00:16:02.446 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c79e43f-4109-43a9-896a-6e96d98d681e] socks connection closed
2025-08-01 00:16:02.446 [info] [command][6ff386fb-7eae-4cbc-9a7f-5320647f4176] Socket close event received
2025-08-01 00:16:02.502 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61855 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:17:02.449 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:17:02.451 [info] [command][f50a052d-3a86-490b-978a-ec1809e50225] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f50a052d-3a86-490b-978a-ec1809e50225""}
2025-08-01 00:17:02.451 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4537e406-07e7-4594-9014-34661039716c] received connection request
2025-08-01 00:17:02.451 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 00:17:02.452 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:17:02.497 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4537e406-07e7-4594-9014-34661039716c] socks forwarding established
2025-08-01 00:17:02.544 [info] [command][f50a052d-3a86-490b-978a-ec1809e50225] Process exited with code 0
2025-08-01 00:17:02.544 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4537e406-07e7-4594-9014-34661039716c] socks connection closed
2025-08-01 00:17:02.544 [info] [command][f50a052d-3a86-490b-978a-ec1809e50225] Socket close event received
2025-08-01 00:17:02.585 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61880 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:18:02.545 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:18:02.546 [info] [command][578b21e6-7d6a-4dbd-ac96-f547028cb18a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""578b21e6-7d6a-4dbd-ac96-f547028cb18a""}
2025-08-01 00:18:02.546 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e5ffbf4d-e4c8-4ea7-8438-5e3b87842449] received connection request
2025-08-01 00:18:02.546 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:18:02.593 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e5ffbf4d-e4c8-4ea7-8438-5e3b87842449] socks forwarding established
2025-08-01 00:18:02.668 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e5ffbf4d-e4c8-4ea7-8438-5e3b87842449] socks connection closed
2025-08-01 00:18:02.669 [info] [command][578b21e6-7d6a-4dbd-ac96-f547028cb18a] Process exited with code 0
2025-08-01 00:18:02.669 [info] [command][578b21e6-7d6a-4dbd-ac96-f547028cb18a] Socket close event received
2025-08-01 00:18:02.709 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61911 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:19:02.674 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:19:02.675 [info] [command][5e717046-6cc2-4714-9df2-f385b3e8d4ee] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5e717046-6cc2-4714-9df2-f385b3e8d4ee""}
2025-08-01 00:19:02.675 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][07e43a7f-685b-47a8-b65f-57f405dcf3b4] received connection request
2025-08-01 00:19:02.676 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:19:02.720 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07e43a7f-685b-47a8-b65f-57f405dcf3b4] socks forwarding established
2025-08-01 00:19:02.766 [info] [command][5e717046-6cc2-4714-9df2-f385b3e8d4ee] Process exited with code 0
2025-08-01 00:19:02.766 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07e43a7f-685b-47a8-b65f-57f405dcf3b4] socks connection closed
2025-08-01 00:19:02.766 [info] [command][5e717046-6cc2-4714-9df2-f385b3e8d4ee] Socket close event received
2025-08-01 00:19:02.807 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61944 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:20:02.774 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:20:02.775 [info] [command][6e16e604-5e1f-4a44-b0c2-7d9d5597707c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6e16e604-5e1f-4a44-b0c2-7d9d5597707c""}
2025-08-01 00:20:02.776 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][830d6485-31b7-49fb-bf38-8a3061b04b3c] received connection request
2025-08-01 00:20:02.776 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:20:03.033 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][830d6485-31b7-49fb-bf38-8a3061b04b3c] socks forwarding established
2025-08-01 00:20:03.108 [info] [command][6e16e604-5e1f-4a44-b0c2-7d9d5597707c] Process exited with code 0
2025-08-01 00:20:03.108 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][830d6485-31b7-49fb-bf38-8a3061b04b3c] socks connection closed
2025-08-01 00:20:03.109 [info] [command][6e16e604-5e1f-4a44-b0c2-7d9d5597707c] Socket close event received
2025-08-01 00:20:03.150 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 61968 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:21:03.116 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:21:03.118 [info] [command][d6bb7a25-988a-4fbf-b750-27d2e06a6162] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d6bb7a25-988a-4fbf-b750-27d2e06a6162""}
2025-08-01 00:21:03.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2f794709-7c90-421d-b047-d328d2f4eed2] received connection request
2025-08-01 00:21:03.119 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:21:03.377 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2f794709-7c90-421d-b047-d328d2f4eed2] socks forwarding established
2025-08-01 00:21:03.544 [info] [command][d6bb7a25-988a-4fbf-b750-27d2e06a6162] Process exited with code 0
2025-08-01 00:21:03.545 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2f794709-7c90-421d-b047-d328d2f4eed2] socks connection closed
2025-08-01 00:21:03.545 [info] [command][d6bb7a25-988a-4fbf-b750-27d2e06a6162] Socket close event received
2025-08-01 00:21:03.721 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62001 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:22:03.553 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:22:03.555 [info] [command][21ddb6ab-1c01-4db9-b9bb-a72f6a0ebec0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""21ddb6ab-1c01-4db9-b9bb-a72f6a0ebec0""}
2025-08-01 00:22:03.556 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][11f61c3b-e02c-4926-b7cc-cc1bdb6e617f] received connection request
2025-08-01 00:22:03.557 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:22:03.599 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][11f61c3b-e02c-4926-b7cc-cc1bdb6e617f] socks forwarding established
2025-08-01 00:22:03.644 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][11f61c3b-e02c-4926-b7cc-cc1bdb6e617f] socks connection closed
2025-08-01 00:22:03.645 [info] [command][21ddb6ab-1c01-4db9-b9bb-a72f6a0ebec0] Process exited with code 0
2025-08-01 00:22:03.645 [info] [command][21ddb6ab-1c01-4db9-b9bb-a72f6a0ebec0] Socket close event received
2025-08-01 00:22:03.684 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62024 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:23:03.656 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:23:03.657 [info] [command][1bc67bc4-12f9-472a-9ab9-8775e8200475] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1bc67bc4-12f9-472a-9ab9-8775e8200475""}
2025-08-01 00:23:03.658 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0774aca8-5d81-487b-9ba7-3eb3e4158061] received connection request
2025-08-01 00:23:03.658 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:23:03.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0774aca8-5d81-487b-9ba7-3eb3e4158061] socks forwarding established
2025-08-01 00:23:03.750 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0774aca8-5d81-487b-9ba7-3eb3e4158061] socks connection closed
2025-08-01 00:23:03.750 [info] [command][1bc67bc4-12f9-472a-9ab9-8775e8200475] Process exited with code 0
2025-08-01 00:23:03.750 [info] [command][1bc67bc4-12f9-472a-9ab9-8775e8200475] Socket close event received
2025-08-01 00:23:03.789 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62056 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:24:03.759 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:24:03.761 [info] [command][ce61166c-1ca3-497a-a2da-a6e2f1f3edf5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ce61166c-1ca3-497a-a2da-a6e2f1f3edf5""}
2025-08-01 00:24:03.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c29acf91-11b1-47e9-804b-b4de9019f64d] received connection request
2025-08-01 00:24:03.762 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:24:03.804 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c29acf91-11b1-47e9-804b-b4de9019f64d] socks forwarding established
2025-08-01 00:24:03.850 [info] [command][ce61166c-1ca3-497a-a2da-a6e2f1f3edf5] Process exited with code 0
2025-08-01 00:24:03.851 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c29acf91-11b1-47e9-804b-b4de9019f64d] socks connection closed
2025-08-01 00:24:03.851 [info] [command][ce61166c-1ca3-497a-a2da-a6e2f1f3edf5] Socket close event received
2025-08-01 00:24:03.891 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62084 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:25:03.861 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:25:03.862 [info] [command][d73745f9-bab8-4f17-9b38-2587f80b6b2f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d73745f9-bab8-4f17-9b38-2587f80b6b2f""}
2025-08-01 00:25:03.863 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7bbf9065-3278-442d-9140-4bf52f5e4cc1] received connection request
2025-08-01 00:25:03.864 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:25:03.912 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7bbf9065-3278-442d-9140-4bf52f5e4cc1] socks forwarding established
2025-08-01 00:25:03.957 [info] [command][d73745f9-bab8-4f17-9b38-2587f80b6b2f] Process exited with code 0
2025-08-01 00:25:03.957 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7bbf9065-3278-442d-9140-4bf52f5e4cc1] socks connection closed
2025-08-01 00:25:03.957 [info] [command][d73745f9-bab8-4f17-9b38-2587f80b6b2f] Socket close event received
2025-08-01 00:25:03.997 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62096 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:26:03.967 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:26:03.968 [info] [command][df0d09f1-2451-4d5e-90e1-40c501e89520] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""df0d09f1-2451-4d5e-90e1-40c501e89520""}
2025-08-01 00:26:03.969 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5d853d10-2b6d-48f4-9e51-77ae1c6e68bf] received connection request
2025-08-01 00:26:03.970 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:26:04.014 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d853d10-2b6d-48f4-9e51-77ae1c6e68bf] socks forwarding established
2025-08-01 00:26:04.060 [info] [command][df0d09f1-2451-4d5e-90e1-40c501e89520] Process exited with code 0
2025-08-01 00:26:04.061 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d853d10-2b6d-48f4-9e51-77ae1c6e68bf] socks connection closed
2025-08-01 00:26:04.061 [info] [command][df0d09f1-2451-4d5e-90e1-40c501e89520] Socket close event received
2025-08-01 00:26:04.101 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62109 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:27:04.065 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:27:04.066 [info] [command][42e43fbb-6fd5-4ff7-8068-eeed5ba6e6a4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""42e43fbb-6fd5-4ff7-8068-eeed5ba6e6a4""}
2025-08-01 00:27:04.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ade3dd32-b0da-4452-b772-d9181aef2140] received connection request
2025-08-01 00:27:04.068 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:27:04.110 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ade3dd32-b0da-4452-b772-d9181aef2140] socks forwarding established
2025-08-01 00:27:04.154 [info] [command][42e43fbb-6fd5-4ff7-8068-eeed5ba6e6a4] Process exited with code 0
2025-08-01 00:27:04.155 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ade3dd32-b0da-4452-b772-d9181aef2140] socks connection closed
2025-08-01 00:27:04.155 [info] [command][42e43fbb-6fd5-4ff7-8068-eeed5ba6e6a4] Socket close event received
2025-08-01 00:27:04.194 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62130 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:28:04.157 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:28:04.158 [info] [command][c8d49d4d-0b20-44f6-9977-884995477f16] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c8d49d4d-0b20-44f6-9977-884995477f16""}
2025-08-01 00:28:04.159 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bce37fae-a76e-4c50-9e10-a8b10ae992d2] received connection request
2025-08-01 00:28:04.159 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:28:04.199 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bce37fae-a76e-4c50-9e10-a8b10ae992d2] socks forwarding established
2025-08-01 00:28:04.243 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bce37fae-a76e-4c50-9e10-a8b10ae992d2] socks connection closed
2025-08-01 00:28:04.244 [info] [command][c8d49d4d-0b20-44f6-9977-884995477f16] Process exited with code 0
2025-08-01 00:28:04.244 [info] [command][c8d49d4d-0b20-44f6-9977-884995477f16] Socket close event received
2025-08-01 00:28:04.285 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62158 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:29:04.249 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:29:04.250 [info] [command][29067894-02ab-4f15-a6f3-2964a70b3caa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""29067894-02ab-4f15-a6f3-2964a70b3caa""}
2025-08-01 00:29:04.251 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][19d7917f-957b-46bc-8565-efe995e93a05] received connection request
2025-08-01 00:29:04.252 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:29:04.310 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][19d7917f-957b-46bc-8565-efe995e93a05] socks forwarding established
2025-08-01 00:29:04.375 [info] [command][29067894-02ab-4f15-a6f3-2964a70b3caa] Process exited with code 0
2025-08-01 00:29:04.376 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][19d7917f-957b-46bc-8565-efe995e93a05] socks connection closed
2025-08-01 00:29:04.376 [info] [command][29067894-02ab-4f15-a6f3-2964a70b3caa] Socket close event received
2025-08-01 00:29:04.417 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62194 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:30:04.386 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:30:04.387 [info] [command][6f9c0bc3-17ba-446d-b01e-0bb2a175cfcf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6f9c0bc3-17ba-446d-b01e-0bb2a175cfcf""}
2025-08-01 00:30:04.388 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a513a98b-90ee-4993-b31d-69ecdfdf4e18] received connection request
2025-08-01 00:30:04.388 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:30:04.433 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a513a98b-90ee-4993-b31d-69ecdfdf4e18] socks forwarding established
2025-08-01 00:30:04.477 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a513a98b-90ee-4993-b31d-69ecdfdf4e18] socks connection closed
2025-08-01 00:30:04.477 [info] [command][6f9c0bc3-17ba-446d-b01e-0bb2a175cfcf] Process exited with code 0
2025-08-01 00:30:04.477 [info] [command][6f9c0bc3-17ba-446d-b01e-0bb2a175cfcf] Socket close event received
2025-08-01 00:30:04.518 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62209 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:31:04.483 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:31:04.484 [info] [command][94c1c4cb-4607-48a3-aa80-06dab99e7fe3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""94c1c4cb-4607-48a3-aa80-06dab99e7fe3""}
2025-08-01 00:31:04.484 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b65fe9b4-2a51-443b-9a4e-103b3f8ed0cf] received connection request
2025-08-01 00:31:04.484 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:31:04.524 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b65fe9b4-2a51-443b-9a4e-103b3f8ed0cf] socks forwarding established
2025-08-01 00:31:04.569 [info] [command][94c1c4cb-4607-48a3-aa80-06dab99e7fe3] Process exited with code 0
2025-08-01 00:31:04.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b65fe9b4-2a51-443b-9a4e-103b3f8ed0cf] socks connection closed
2025-08-01 00:31:04.570 [info] [command][94c1c4cb-4607-48a3-aa80-06dab99e7fe3] Socket close event received
2025-08-01 00:31:04.609 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62229 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:32:04.571 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:32:04.572 [info] [command][ca11f9ad-3dc7-459d-8f82-7b08fa8ce257] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ca11f9ad-3dc7-459d-8f82-7b08fa8ce257""}
2025-08-01 00:32:04.573 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4e794662-7079-492b-a0b6-8b6ac47a426e] received connection request
2025-08-01 00:32:04.573 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:32:04.616 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4e794662-7079-492b-a0b6-8b6ac47a426e] socks forwarding established
2025-08-01 00:32:04.660 [info] [command][ca11f9ad-3dc7-459d-8f82-7b08fa8ce257] Process exited with code 0
2025-08-01 00:32:04.661 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4e794662-7079-492b-a0b6-8b6ac47a426e] socks connection closed
2025-08-01 00:32:04.661 [info] [command][ca11f9ad-3dc7-459d-8f82-7b08fa8ce257] Socket close event received
2025-08-01 00:32:04.701 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62253 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:33:04.664 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:33:04.665 [info] [command][99d5ed9a-6f5c-4f59-8a36-0ead8cb74533] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""99d5ed9a-6f5c-4f59-8a36-0ead8cb74533""}
2025-08-01 00:33:04.666 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e0f3db36-d5c7-46f4-960c-21a08b5891fd] received connection request
2025-08-01 00:33:04.668 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:33:04.714 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e0f3db36-d5c7-46f4-960c-21a08b5891fd] socks forwarding established
2025-08-01 00:33:04.757 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e0f3db36-d5c7-46f4-960c-21a08b5891fd] socks connection closed
2025-08-01 00:33:04.758 [info] [command][99d5ed9a-6f5c-4f59-8a36-0ead8cb74533] Process exited with code 0
2025-08-01 00:33:04.758 [info] [command][99d5ed9a-6f5c-4f59-8a36-0ead8cb74533] Socket close event received
2025-08-01 00:33:04.797 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62289 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:34:04.768 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:34:04.769 [info] [command][8a6abe84-27d4-4cf3-ab6e-4f4157d6941d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8a6abe84-27d4-4cf3-ab6e-4f4157d6941d""}
2025-08-01 00:34:04.770 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][97c5b15a-d599-45df-894d-79fa3230f4ab] received connection request
2025-08-01 00:34:04.771 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:34:04.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97c5b15a-d599-45df-894d-79fa3230f4ab] socks forwarding established
2025-08-01 00:34:04.903 [info] [command][8a6abe84-27d4-4cf3-ab6e-4f4157d6941d] Process exited with code 0
2025-08-01 00:34:04.904 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97c5b15a-d599-45df-894d-79fa3230f4ab] socks connection closed
2025-08-01 00:34:04.904 [info] [command][8a6abe84-27d4-4cf3-ab6e-4f4157d6941d] Socket close event received
2025-08-01 00:34:05.024 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62305 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:35:04.913 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:35:04.915 [info] [command][3398f8bb-4490-42ea-969d-9f66ce7981a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3398f8bb-4490-42ea-969d-9f66ce7981a2""}
2025-08-01 00:35:04.916 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c8420905-4bfe-48d5-ab39-b0537516a269] received connection request
2025-08-01 00:35:04.917 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:35:04.958 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c8420905-4bfe-48d5-ab39-b0537516a269] socks forwarding established
2025-08-01 00:35:05.006 [info] [command][3398f8bb-4490-42ea-969d-9f66ce7981a2] Process exited with code 0
2025-08-01 00:35:05.007 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c8420905-4bfe-48d5-ab39-b0537516a269] socks connection closed
2025-08-01 00:35:05.007 [info] [command][3398f8bb-4490-42ea-969d-9f66ce7981a2] Socket close event received
2025-08-01 00:35:05.048 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62318 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:36:05.016 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:36:05.018 [info] [command][58983a92-1f4f-4e20-bc93-c65191cca483] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""58983a92-1f4f-4e20-bc93-c65191cca483""}
2025-08-01 00:36:05.019 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][271a2256-fa9f-484f-b0d9-00e4b8631235] received connection request
2025-08-01 00:36:05.019 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:36:05.062 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][271a2256-fa9f-484f-b0d9-00e4b8631235] socks forwarding established
2025-08-01 00:36:05.106 [info] [command][58983a92-1f4f-4e20-bc93-c65191cca483] Process exited with code 0
2025-08-01 00:36:05.107 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][271a2256-fa9f-484f-b0d9-00e4b8631235] socks connection closed
2025-08-01 00:36:05.107 [info] [command][58983a92-1f4f-4e20-bc93-c65191cca483] Socket close event received
2025-08-01 00:36:05.147 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62336 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:37:05.117 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:37:05.119 [info] [command][53a2b68f-5d27-4ec8-8f49-8fcdd9cd24ae] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""53a2b68f-5d27-4ec8-8f49-8fcdd9cd24ae""}
2025-08-01 00:37:05.119 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1a9ef138-54d0-4230-8e6a-a54ff292445a] received connection request
2025-08-01 00:37:05.120 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:37:05.161 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a9ef138-54d0-4230-8e6a-a54ff292445a] socks forwarding established
2025-08-01 00:37:05.205 [info] [command][53a2b68f-5d27-4ec8-8f49-8fcdd9cd24ae] Process exited with code 0
2025-08-01 00:37:05.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a9ef138-54d0-4230-8e6a-a54ff292445a] socks connection closed
2025-08-01 00:37:05.205 [info] [command][53a2b68f-5d27-4ec8-8f49-8fcdd9cd24ae] Socket close event received
2025-08-01 00:37:05.245 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62365 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:38:05.205 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:38:05.207 [info] [command][37652d9b-afeb-4b81-86e2-4cc58f0b5153] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""37652d9b-afeb-4b81-86e2-4cc58f0b5153""}
2025-08-01 00:38:05.208 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][907db749-624d-4fca-919d-9bee9cf6d6b2] received connection request
2025-08-01 00:38:05.209 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:38:05.252 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][907db749-624d-4fca-919d-9bee9cf6d6b2] socks forwarding established
2025-08-01 00:38:05.298 [info] [command][37652d9b-afeb-4b81-86e2-4cc58f0b5153] Process exited with code 0
2025-08-01 00:38:05.298 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][907db749-624d-4fca-919d-9bee9cf6d6b2] socks connection closed
2025-08-01 00:38:05.298 [info] [command][37652d9b-afeb-4b81-86e2-4cc58f0b5153] Socket close event received
2025-08-01 00:38:05.400 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62396 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:39:05.309 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:39:05.311 [info] [command][4befc340-0ea9-4b34-a019-17c3e6fc3f77] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4befc340-0ea9-4b34-a019-17c3e6fc3f77""}
2025-08-01 00:39:05.311 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d3d2e22a-4652-4302-8b0b-f03402ace604] received connection request
2025-08-01 00:39:05.312 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:39:05.358 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3d2e22a-4652-4302-8b0b-f03402ace604] socks forwarding established
2025-08-01 00:39:05.403 [info] [command][4befc340-0ea9-4b34-a019-17c3e6fc3f77] Process exited with code 0
2025-08-01 00:39:05.403 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3d2e22a-4652-4302-8b0b-f03402ace604] socks connection closed
2025-08-01 00:39:05.403 [info] [command][4befc340-0ea9-4b34-a019-17c3e6fc3f77] Socket close event received
2025-08-01 00:39:05.444 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62413 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:40:05.411 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:40:05.412 [info] [command][dbec3eb2-c756-4b54-9a0a-7c5e12ef5fb0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dbec3eb2-c756-4b54-9a0a-7c5e12ef5fb0""}
2025-08-01 00:40:05.413 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][42da25aa-def8-4039-9d62-8f79a1978e37] received connection request
2025-08-01 00:40:05.414 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:40:05.456 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][42da25aa-def8-4039-9d62-8f79a1978e37] socks forwarding established
2025-08-01 00:40:05.501 [info] [command][dbec3eb2-c756-4b54-9a0a-7c5e12ef5fb0] Process exited with code 0
2025-08-01 00:40:05.501 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][42da25aa-def8-4039-9d62-8f79a1978e37] socks connection closed
2025-08-01 00:40:05.502 [info] [command][dbec3eb2-c756-4b54-9a0a-7c5e12ef5fb0] Socket close event received
2025-08-01 00:40:05.541 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62428 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:41:05.505 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:41:05.507 [info] [command][f3d0d704-18cd-41a0-b665-7c2b1063d96f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f3d0d704-18cd-41a0-b665-7c2b1063d96f""}
2025-08-01 00:41:05.507 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a3c00c2b-9a5a-458f-8577-c2c82cb9b245] received connection request
2025-08-01 00:41:05.508 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:41:05.553 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a3c00c2b-9a5a-458f-8577-c2c82cb9b245] socks forwarding established
2025-08-01 00:41:05.598 [info] [command][f3d0d704-18cd-41a0-b665-7c2b1063d96f] Process exited with code 0
2025-08-01 00:41:05.598 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a3c00c2b-9a5a-458f-8577-c2c82cb9b245] socks connection closed
2025-08-01 00:41:05.598 [info] [command][f3d0d704-18cd-41a0-b665-7c2b1063d96f] Socket close event received
2025-08-01 00:41:05.639 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62449 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:42:05.609 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:42:05.611 [info] [command][c1181b02-32fd-46f8-99d4-596fabc97f96] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c1181b02-32fd-46f8-99d4-596fabc97f96""}
2025-08-01 00:42:05.611 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][828a6f96-fcef-43ea-aa03-e0e04c7407ad] received connection request
2025-08-01 00:42:05.612 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:42:05.653 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][828a6f96-fcef-43ea-aa03-e0e04c7407ad] socks forwarding established
2025-08-01 00:42:05.699 [info] [command][c1181b02-32fd-46f8-99d4-596fabc97f96] Process exited with code 0
2025-08-01 00:42:05.699 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][828a6f96-fcef-43ea-aa03-e0e04c7407ad] socks connection closed
2025-08-01 00:42:05.700 [info] [command][c1181b02-32fd-46f8-99d4-596fabc97f96] Socket close event received
2025-08-01 00:42:05.739 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62479 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:43:05.709 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:43:05.711 [info] [command][7cb62d5e-9457-46f4-8fdf-db7b0f691d00] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7cb62d5e-9457-46f4-8fdf-db7b0f691d00""}
2025-08-01 00:43:05.712 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][844d51b2-44ba-41b4-b09b-f54e1c5f7d1a] received connection request
2025-08-01 00:43:05.712 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:43:05.756 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][844d51b2-44ba-41b4-b09b-f54e1c5f7d1a] socks forwarding established
2025-08-01 00:43:05.800 [info] [command][7cb62d5e-9457-46f4-8fdf-db7b0f691d00] Process exited with code 0
2025-08-01 00:43:05.800 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][844d51b2-44ba-41b4-b09b-f54e1c5f7d1a] socks connection closed
2025-08-01 00:43:05.800 [info] [command][7cb62d5e-9457-46f4-8fdf-db7b0f691d00] Socket close event received
2025-08-01 00:43:05.840 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62507 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:44:05.803 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:44:05.806 [info] [command][0b4d990c-5020-4e23-95f0-94df53ec8092] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0b4d990c-5020-4e23-95f0-94df53ec8092""}
2025-08-01 00:44:05.807 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8a3f0e6e-e877-4c3c-8787-8852d3853eed] received connection request
2025-08-01 00:44:05.807 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:44:05.865 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8a3f0e6e-e877-4c3c-8787-8852d3853eed] socks forwarding established
2025-08-01 00:44:05.910 [info] [command][0b4d990c-5020-4e23-95f0-94df53ec8092] Process exited with code 0
2025-08-01 00:44:05.910 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8a3f0e6e-e877-4c3c-8787-8852d3853eed] socks connection closed
2025-08-01 00:44:05.910 [info] [command][0b4d990c-5020-4e23-95f0-94df53ec8092] Socket close event received
2025-08-01 00:44:05.952 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62530 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:45:05.921 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:45:05.922 [info] [command][5a815f8c-2744-4a03-a716-955671d730b2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5a815f8c-2744-4a03-a716-955671d730b2""}
2025-08-01 00:45:05.922 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][52aace33-7b73-461f-ad88-af44d2bacdbb] received connection request
2025-08-01 00:45:05.922 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:45:05.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52aace33-7b73-461f-ad88-af44d2bacdbb] socks forwarding established
2025-08-01 00:45:06.009 [info] [command][5a815f8c-2744-4a03-a716-955671d730b2] Process exited with code 0
2025-08-01 00:45:06.009 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52aace33-7b73-461f-ad88-af44d2bacdbb] socks connection closed
2025-08-01 00:45:06.009 [info] [command][5a815f8c-2744-4a03-a716-955671d730b2] Socket close event received
2025-08-01 00:45:06.050 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62542 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:46:06.014 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:46:06.015 [info] [command][e53760db-620f-43c8-a799-e5fbe3aadb34] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e53760db-620f-43c8-a799-e5fbe3aadb34""}
2025-08-01 00:46:06.016 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0a965d67-c14d-4765-a1e6-120d53ea34d9] received connection request
2025-08-01 00:46:06.017 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:46:06.059 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0a965d67-c14d-4765-a1e6-120d53ea34d9] socks forwarding established
2025-08-01 00:46:06.102 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0a965d67-c14d-4765-a1e6-120d53ea34d9] socks connection closed
2025-08-01 00:46:06.103 [info] [command][e53760db-620f-43c8-a799-e5fbe3aadb34] Process exited with code 0
2025-08-01 00:46:06.103 [info] [command][e53760db-620f-43c8-a799-e5fbe3aadb34] Socket close event received
2025-08-01 00:46:06.143 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62554 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:47:06.109 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:47:06.111 [info] [command][c27150d9-0137-40fb-b75c-c5038952c081] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c27150d9-0137-40fb-b75c-c5038952c081""}
2025-08-01 00:47:06.111 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][48df9ca6-167b-4449-b68f-32a31bb92b1f] received connection request
2025-08-01 00:47:06.112 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:47:06.154 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48df9ca6-167b-4449-b68f-32a31bb92b1f] socks forwarding established
2025-08-01 00:47:06.199 [info] [command][c27150d9-0137-40fb-b75c-c5038952c081] Process exited with code 0
2025-08-01 00:47:06.199 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48df9ca6-167b-4449-b68f-32a31bb92b1f] socks connection closed
2025-08-01 00:47:06.199 [info] [command][c27150d9-0137-40fb-b75c-c5038952c081] Socket close event received
2025-08-01 00:47:06.239 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62589 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:48:06.205 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:48:06.207 [info] [command][c51bab02-018f-4280-962d-4e8cd230b36c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c51bab02-018f-4280-962d-4e8cd230b36c""}
2025-08-01 00:48:06.208 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5c17dd0a-2c9c-49b3-92fc-acc06a50902c] received connection request
2025-08-01 00:48:06.208 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:48:06.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c17dd0a-2c9c-49b3-92fc-acc06a50902c] socks forwarding established
2025-08-01 00:48:06.313 [info] [command][c51bab02-018f-4280-962d-4e8cd230b36c] Process exited with code 0
2025-08-01 00:48:06.313 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c17dd0a-2c9c-49b3-92fc-acc06a50902c] socks connection closed
2025-08-01 00:48:06.313 [info] [command][c51bab02-018f-4280-962d-4e8cd230b36c] Socket close event received
2025-08-01 00:48:06.427 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62623 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:49:06.317 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:49:06.320 [info] [command][1aea54e3-9e56-48b6-ac57-4753af86800c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1aea54e3-9e56-48b6-ac57-4753af86800c""}
2025-08-01 00:49:06.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5ee23520-410f-4dd3-9e61-f42dd75c7371] received connection request
2025-08-01 00:49:06.321 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:49:06.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ee23520-410f-4dd3-9e61-f42dd75c7371] socks forwarding established
2025-08-01 00:49:06.409 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ee23520-410f-4dd3-9e61-f42dd75c7371] socks connection closed
2025-08-01 00:49:06.409 [info] [command][1aea54e3-9e56-48b6-ac57-4753af86800c] Process exited with code 0
2025-08-01 00:49:06.409 [info] [command][1aea54e3-9e56-48b6-ac57-4753af86800c] Socket close event received
2025-08-01 00:49:06.448 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62652 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:50:06.420 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:50:06.422 [info] [command][75ab9bb9-e53c-465b-aef2-301cffc31fa2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""75ab9bb9-e53c-465b-aef2-301cffc31fa2""}
2025-08-01 00:50:06.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5222296e-8f9c-445e-88f9-033c68caae55] received connection request
2025-08-01 00:50:06.423 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:50:06.464 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5222296e-8f9c-445e-88f9-033c68caae55] socks forwarding established
2025-08-01 00:50:06.508 [info] [command][75ab9bb9-e53c-465b-aef2-301cffc31fa2] Process exited with code 0
2025-08-01 00:50:06.508 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5222296e-8f9c-445e-88f9-033c68caae55] socks connection closed
2025-08-01 00:50:06.508 [info] [command][75ab9bb9-e53c-465b-aef2-301cffc31fa2] Socket close event received
2025-08-01 00:50:06.553 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62668 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:51:06.514 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:51:06.516 [info] [command][d7d9ae9c-8202-4374-91aa-9b52d62d0cd9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d7d9ae9c-8202-4374-91aa-9b52d62d0cd9""}
2025-08-01 00:51:06.517 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][65c0ab48-2db8-41a7-b783-4ecb8f28fdf2] received connection request
2025-08-01 00:51:06.517 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:51:06.557 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][65c0ab48-2db8-41a7-b783-4ecb8f28fdf2] socks forwarding established
2025-08-01 00:51:06.601 [info] [command][d7d9ae9c-8202-4374-91aa-9b52d62d0cd9] Process exited with code 0
2025-08-01 00:51:06.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][65c0ab48-2db8-41a7-b783-4ecb8f28fdf2] socks connection closed
2025-08-01 00:51:06.601 [info] [command][d7d9ae9c-8202-4374-91aa-9b52d62d0cd9] Socket close event received
2025-08-01 00:51:06.640 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62683 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:52:06.608 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:52:06.609 [info] [command][b7d711e6-4306-409e-8517-705998ea3a9e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7d711e6-4306-409e-8517-705998ea3a9e""}
2025-08-01 00:52:06.610 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][790eccd4-3820-433d-b778-beebc9f53494] received connection request
2025-08-01 00:52:06.610 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:52:06.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][790eccd4-3820-433d-b778-beebc9f53494] socks forwarding established
2025-08-01 00:52:06.697 [info] [command][b7d711e6-4306-409e-8517-705998ea3a9e] Process exited with code 0
2025-08-01 00:52:06.697 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][790eccd4-3820-433d-b778-beebc9f53494] socks connection closed
2025-08-01 00:52:06.697 [info] [command][b7d711e6-4306-409e-8517-705998ea3a9e] Socket close event received
2025-08-01 00:52:06.738 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62709 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:53:06.698 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:53:06.700 [info] [command][94ce349d-aff7-444b-becd-7b1c4410eda4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""94ce349d-aff7-444b-becd-7b1c4410eda4""}
2025-08-01 00:53:06.700 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0766ee66-c50a-4951-9fe1-388b3cc71bc4] received connection request
2025-08-01 00:53:06.700 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:53:06.741 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0766ee66-c50a-4951-9fe1-388b3cc71bc4] socks forwarding established
2025-08-01 00:53:06.785 [info] [command][94ce349d-aff7-444b-becd-7b1c4410eda4] Process exited with code 0
2025-08-01 00:53:06.786 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0766ee66-c50a-4951-9fe1-388b3cc71bc4] socks connection closed
2025-08-01 00:53:06.786 [info] [command][94ce349d-aff7-444b-becd-7b1c4410eda4] Socket close event received
2025-08-01 00:53:06.825 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62741 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:54:06.795 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:54:06.797 [info] [command][8054fa70-0846-4a64-8db6-7ad95b8bde08] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8054fa70-0846-4a64-8db6-7ad95b8bde08""}
2025-08-01 00:54:06.798 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8ad63908-a805-4a16-a416-24f8896de687] received connection request
2025-08-01 00:54:06.798 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:54:06.845 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ad63908-a805-4a16-a416-24f8896de687] socks forwarding established
2025-08-01 00:54:06.892 [info] [command][8054fa70-0846-4a64-8db6-7ad95b8bde08] Process exited with code 0
2025-08-01 00:54:06.893 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ad63908-a805-4a16-a416-24f8896de687] socks connection closed
2025-08-01 00:54:06.893 [info] [command][8054fa70-0846-4a64-8db6-7ad95b8bde08] Socket close event received
2025-08-01 00:54:06.932 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62758 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:55:06.902 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:55:06.904 [info] [command][10254796-36f4-4660-ad72-ec819a1ed38b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""10254796-36f4-4660-ad72-ec819a1ed38b""}
2025-08-01 00:55:06.905 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a5bf7eeb-4abc-4cbd-ae03-84a03e5146cd] received connection request
2025-08-01 00:55:06.905 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:55:06.950 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a5bf7eeb-4abc-4cbd-ae03-84a03e5146cd] socks forwarding established
2025-08-01 00:55:06.995 [info] [command][10254796-36f4-4660-ad72-ec819a1ed38b] Process exited with code 0
2025-08-01 00:55:06.995 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a5bf7eeb-4abc-4cbd-ae03-84a03e5146cd] socks connection closed
2025-08-01 00:55:06.995 [info] [command][10254796-36f4-4660-ad72-ec819a1ed38b] Socket close event received
2025-08-01 00:55:07.034 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62773 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:56:06.998 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:56:07.000 [info] [command][48180943-ceff-43a3-b773-076de9a4bbf6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""48180943-ceff-43a3-b773-076de9a4bbf6""}
2025-08-01 00:56:07.001 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c1b86839-8c9d-40dc-ad15-2bbf5d7b2334] received connection request
2025-08-01 00:56:07.001 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:56:07.046 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1b86839-8c9d-40dc-ad15-2bbf5d7b2334] socks forwarding established
2025-08-01 00:56:07.091 [info] [command][48180943-ceff-43a3-b773-076de9a4bbf6] Process exited with code 0
2025-08-01 00:56:07.092 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1b86839-8c9d-40dc-ad15-2bbf5d7b2334] socks connection closed
2025-08-01 00:56:07.092 [info] [command][48180943-ceff-43a3-b773-076de9a4bbf6] Socket close event received
2025-08-01 00:56:07.132 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62787 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:57:07.102 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:57:07.104 [info] [command][e5e27c9e-5b07-4364-a8a1-26c9ef399b72] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e5e27c9e-5b07-4364-a8a1-26c9ef399b72""}
2025-08-01 00:57:07.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c5007d73-0cab-491c-95f9-52575f551e11] received connection request
2025-08-01 00:57:07.105 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:57:07.199 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c5007d73-0cab-491c-95f9-52575f551e11] socks forwarding established
2025-08-01 00:57:07.244 [info] [command][e5e27c9e-5b07-4364-a8a1-26c9ef399b72] Process exited with code 0
2025-08-01 00:57:07.245 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c5007d73-0cab-491c-95f9-52575f551e11] socks connection closed
2025-08-01 00:57:07.245 [info] [command][e5e27c9e-5b07-4364-a8a1-26c9ef399b72] Socket close event received
2025-08-01 00:57:07.358 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62823 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:58:07.247 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:58:07.249 [info] [command][2dfaba16-2fb9-49ee-bcd7-13d088d7883c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2dfaba16-2fb9-49ee-bcd7-13d088d7883c""}
2025-08-01 00:58:07.250 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f1fd300d-cad7-4a34-a0e8-49df08776caa] received connection request
2025-08-01 00:58:07.250 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:58:07.298 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f1fd300d-cad7-4a34-a0e8-49df08776caa] socks forwarding established
2025-08-01 00:58:07.341 [info] [command][2dfaba16-2fb9-49ee-bcd7-13d088d7883c] Process exited with code 0
2025-08-01 00:58:07.341 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f1fd300d-cad7-4a34-a0e8-49df08776caa] socks connection closed
2025-08-01 00:58:07.341 [info] [command][2dfaba16-2fb9-49ee-bcd7-13d088d7883c] Socket close event received
2025-08-01 00:58:07.379 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62849 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 00:59:07.349 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 00:59:07.351 [info] [command][404d9ec8-c178-4edd-9600-15a608836499] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""404d9ec8-c178-4edd-9600-15a608836499""}
2025-08-01 00:59:07.351 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][30605d49-2006-4a4b-91aa-617f64b371e9] received connection request
2025-08-01 00:59:07.351 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 00:59:07.395 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][30605d49-2006-4a4b-91aa-617f64b371e9] socks forwarding established
2025-08-01 00:59:07.439 [info] [command][404d9ec8-c178-4edd-9600-15a608836499] Process exited with code 0
2025-08-01 00:59:07.439 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][30605d49-2006-4a4b-91aa-617f64b371e9] socks connection closed
2025-08-01 00:59:07.439 [info] [command][404d9ec8-c178-4edd-9600-15a608836499] Socket close event received
2025-08-01 00:59:07.480 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62866 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:00:07.444 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:00:07.446 [info] [command][7acd0df6-a369-4398-8bc0-125262caccb6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7acd0df6-a369-4398-8bc0-125262caccb6""}
2025-08-01 01:00:07.447 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][92691e94-5049-4e16-95fd-173f79593ad3] received connection request
2025-08-01 01:00:07.448 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:00:07.490 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][92691e94-5049-4e16-95fd-173f79593ad3] socks forwarding established
2025-08-01 01:00:07.536 [info] [command][7acd0df6-a369-4398-8bc0-125262caccb6] Process exited with code 0
2025-08-01 01:00:07.536 [info] [command][7acd0df6-a369-4398-8bc0-125262caccb6] Socket close event received
2025-08-01 01:00:07.536 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][92691e94-5049-4e16-95fd-173f79593ad3] socks connection closed
2025-08-01 01:00:07.582 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62878 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:01:07.542 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:01:07.544 [info] [command][b14c29dc-0b07-42f1-9f5b-6fd29c72e61b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b14c29dc-0b07-42f1-9f5b-6fd29c72e61b""}
2025-08-01 01:01:07.544 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a85124fc-112d-4c05-88af-ffd0b0107bb6] received connection request
2025-08-01 01:01:07.545 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:01:07.585 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a85124fc-112d-4c05-88af-ffd0b0107bb6] socks forwarding established
2025-08-01 01:01:07.630 [info] [command][b14c29dc-0b07-42f1-9f5b-6fd29c72e61b] Process exited with code 0
2025-08-01 01:01:07.630 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a85124fc-112d-4c05-88af-ffd0b0107bb6] socks connection closed
2025-08-01 01:01:07.630 [info] [command][b14c29dc-0b07-42f1-9f5b-6fd29c72e61b] Socket close event received
2025-08-01 01:01:07.673 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62892 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:02:07.636 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:02:07.637 [info] [command][0217b2eb-7926-45e2-b9f2-5fd804ceaaf9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0217b2eb-7926-45e2-b9f2-5fd804ceaaf9""}
2025-08-01 01:02:07.638 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d7015107-1e46-4ccf-ba7a-1929c790d20d] received connection request
2025-08-01 01:02:07.638 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:02:07.751 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d7015107-1e46-4ccf-ba7a-1929c790d20d] socks forwarding established
2025-08-01 01:02:07.796 [info] [command][0217b2eb-7926-45e2-b9f2-5fd804ceaaf9] Process exited with code 0
2025-08-01 01:02:07.797 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d7015107-1e46-4ccf-ba7a-1929c790d20d] socks connection closed
2025-08-01 01:02:07.797 [info] [command][0217b2eb-7926-45e2-b9f2-5fd804ceaaf9] Socket close event received
2025-08-01 01:02:07.906 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62940 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:03:07.800 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:03:07.802 [info] [command][0e3150e5-2e9a-4ff2-a9b4-9788ee8d6ff1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0e3150e5-2e9a-4ff2-a9b4-9788ee8d6ff1""}
2025-08-01 01:03:07.802 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e466e022-a1f6-4885-adff-21730d44d83e] received connection request
2025-08-01 01:03:07.802 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:03:07.845 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e466e022-a1f6-4885-adff-21730d44d83e] socks forwarding established
2025-08-01 01:03:07.889 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e466e022-a1f6-4885-adff-21730d44d83e] socks connection closed
2025-08-01 01:03:07.889 [info] [command][0e3150e5-2e9a-4ff2-a9b4-9788ee8d6ff1] Process exited with code 0
2025-08-01 01:03:07.889 [info] [command][0e3150e5-2e9a-4ff2-a9b4-9788ee8d6ff1] Socket close event received
2025-08-01 01:03:07.929 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62968 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:04:07.896 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:04:07.897 [info] [command][e04fc98f-1edb-42e2-8ffa-0fd96d3880a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e04fc98f-1edb-42e2-8ffa-0fd96d3880a2""}
2025-08-01 01:04:07.897 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2b2dbce6-6f02-4e9d-b9c2-44e4cbe61259] received connection request
2025-08-01 01:04:07.898 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:04:07.937 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2b2dbce6-6f02-4e9d-b9c2-44e4cbe61259] socks forwarding established
2025-08-01 01:04:07.982 [info] [command][e04fc98f-1edb-42e2-8ffa-0fd96d3880a2] Process exited with code 0
2025-08-01 01:04:07.982 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2b2dbce6-6f02-4e9d-b9c2-44e4cbe61259] socks connection closed
2025-08-01 01:04:07.982 [info] [command][e04fc98f-1edb-42e2-8ffa-0fd96d3880a2] Socket close event received
2025-08-01 01:04:08.024 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 62988 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:05:07.987 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:05:07.990 [info] [command][f8e07df9-5545-47b5-b55b-0a1c8e3f2ad4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f8e07df9-5545-47b5-b55b-0a1c8e3f2ad4""}
2025-08-01 01:05:07.990 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6ab2033a-c356-4a88-b20e-9ec769ef29cc] received connection request
2025-08-01 01:05:07.991 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:05:08.038 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6ab2033a-c356-4a88-b20e-9ec769ef29cc] socks forwarding established
2025-08-01 01:05:08.083 [info] [command][f8e07df9-5545-47b5-b55b-0a1c8e3f2ad4] Process exited with code 0
2025-08-01 01:05:08.084 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6ab2033a-c356-4a88-b20e-9ec769ef29cc] socks connection closed
2025-08-01 01:05:08.084 [info] [command][f8e07df9-5545-47b5-b55b-0a1c8e3f2ad4] Socket close event received
2025-08-01 01:05:08.122 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63006 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:06:08.093 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:06:08.096 [info] [command][a8934c81-447d-4842-ab76-94ef61ab5712] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a8934c81-447d-4842-ab76-94ef61ab5712""}
2025-08-01 01:06:08.097 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][afed9888-1df9-47d7-a2d6-748c66bbf82b] received connection request
2025-08-01 01:06:08.097 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:06:08.226 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][afed9888-1df9-47d7-a2d6-748c66bbf82b] socks forwarding established
2025-08-01 01:06:08.271 [info] [command][a8934c81-447d-4842-ab76-94ef61ab5712] Process exited with code 0
2025-08-01 01:06:08.271 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][afed9888-1df9-47d7-a2d6-748c66bbf82b] socks connection closed
2025-08-01 01:06:08.271 [info] [command][a8934c81-447d-4842-ab76-94ef61ab5712] Socket close event received
2025-08-01 01:06:08.312 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63024 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:07:08.278 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:07:08.280 [info] [command][f7da37ce-f129-4fcc-aba7-b515aeb87e39] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f7da37ce-f129-4fcc-aba7-b515aeb87e39""}
2025-08-01 01:07:08.280 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b0eaad2c-693e-4a2a-b279-65281014e7d8] received connection request
2025-08-01 01:07:08.281 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:07:08.333 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b0eaad2c-693e-4a2a-b279-65281014e7d8] socks forwarding established
2025-08-01 01:07:08.377 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b0eaad2c-693e-4a2a-b279-65281014e7d8] socks connection closed
2025-08-01 01:07:08.377 [info] [command][f7da37ce-f129-4fcc-aba7-b515aeb87e39] Process exited with code 0
2025-08-01 01:07:08.377 [info] [command][f7da37ce-f129-4fcc-aba7-b515aeb87e39] Socket close event received
2025-08-01 01:07:08.440 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63057 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:08:08.387 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:08:08.388 [info] [command][866ab438-a7ce-4484-958a-e372449c2075] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""866ab438-a7ce-4484-958a-e372449c2075""}
2025-08-01 01:08:08.389 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][06679f1c-7a5a-4929-a296-b331ea2c8e44] received connection request
2025-08-01 01:08:08.389 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:08:08.441 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06679f1c-7a5a-4929-a296-b331ea2c8e44] socks forwarding established
2025-08-01 01:08:08.486 [info] [command][866ab438-a7ce-4484-958a-e372449c2075] Process exited with code 0
2025-08-01 01:08:08.486 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][06679f1c-7a5a-4929-a296-b331ea2c8e44] socks connection closed
2025-08-01 01:08:08.486 [info] [command][866ab438-a7ce-4484-958a-e372449c2075] Socket close event received
2025-08-01 01:08:08.525 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63086 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:09:08.492 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:09:08.494 [info] [command][987a603a-5cc8-4b81-9f7d-b36549ae40cf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""987a603a-5cc8-4b81-9f7d-b36549ae40cf""}
2025-08-01 01:09:08.495 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][69c985fe-3ac3-4255-a3d2-b066fcdae19c] received connection request
2025-08-01 01:09:08.495 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:09:08.541 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69c985fe-3ac3-4255-a3d2-b066fcdae19c] socks forwarding established
2025-08-01 01:09:08.585 [info] [command][987a603a-5cc8-4b81-9f7d-b36549ae40cf] Process exited with code 0
2025-08-01 01:09:08.585 [info] [command][987a603a-5cc8-4b81-9f7d-b36549ae40cf] Socket close event received
2025-08-01 01:09:08.585 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][69c985fe-3ac3-4255-a3d2-b066fcdae19c] socks connection closed
2025-08-01 01:09:08.624 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63105 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:10:08.593 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:10:08.595 [info] [command][671f6005-3052-4af2-a0ce-694e7fd037f3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""671f6005-3052-4af2-a0ce-694e7fd037f3""}
2025-08-01 01:10:08.595 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e42bc4db-d617-4705-b1c4-7b60f661456f] received connection request
2025-08-01 01:10:08.596 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:10:08.641 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e42bc4db-d617-4705-b1c4-7b60f661456f] socks forwarding established
2025-08-01 01:10:08.687 [info] [command][671f6005-3052-4af2-a0ce-694e7fd037f3] Process exited with code 0
2025-08-01 01:10:08.687 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e42bc4db-d617-4705-b1c4-7b60f661456f] socks connection closed
2025-08-01 01:10:08.688 [info] [command][671f6005-3052-4af2-a0ce-694e7fd037f3] Socket close event received
2025-08-01 01:10:08.745 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63139 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:11:08.688 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:11:08.690 [info] [command][8f6a443f-1003-4281-8d57-b554f20cfaa1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8f6a443f-1003-4281-8d57-b554f20cfaa1""}
2025-08-01 01:11:08.691 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c36e0d1c-9710-4e38-bdc6-7dbc850d8ade] received connection request
2025-08-01 01:11:08.691 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:11:08.734 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c36e0d1c-9710-4e38-bdc6-7dbc850d8ade] socks forwarding established
2025-08-01 01:11:08.780 [info] [command][8f6a443f-1003-4281-8d57-b554f20cfaa1] Process exited with code 0
2025-08-01 01:11:08.780 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c36e0d1c-9710-4e38-bdc6-7dbc850d8ade] socks connection closed
2025-08-01 01:11:08.780 [info] [command][8f6a443f-1003-4281-8d57-b554f20cfaa1] Socket close event received
2025-08-01 01:11:08.821 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63153 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:12:08.783 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:12:08.786 [info] [command][7377a5ee-0b0a-4233-b7bc-9469cec25212] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7377a5ee-0b0a-4233-b7bc-9469cec25212""}
2025-08-01 01:12:08.787 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eec4fa66-45cc-4ccf-ae0a-f5da1ac97ad1] received connection request
2025-08-01 01:12:08.787 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:12:08.830 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eec4fa66-45cc-4ccf-ae0a-f5da1ac97ad1] socks forwarding established
2025-08-01 01:12:08.875 [info] [command][7377a5ee-0b0a-4233-b7bc-9469cec25212] Process exited with code 0
2025-08-01 01:12:08.875 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eec4fa66-45cc-4ccf-ae0a-f5da1ac97ad1] socks connection closed
2025-08-01 01:12:08.875 [info] [command][7377a5ee-0b0a-4233-b7bc-9469cec25212] Socket close event received
2025-08-01 01:12:08.915 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63191 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:13:08.879 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:13:08.882 [info] [command][46a793b8-6dd8-4aea-8159-bbca5e25ef03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""46a793b8-6dd8-4aea-8159-bbca5e25ef03""}
2025-08-01 01:13:08.882 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5a543875-8723-4e43-89ab-66b16c5a5bc0] received connection request
2025-08-01 01:13:08.883 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:13:08.932 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5a543875-8723-4e43-89ab-66b16c5a5bc0] socks forwarding established
2025-08-01 01:13:08.977 [info] [command][46a793b8-6dd8-4aea-8159-bbca5e25ef03] Process exited with code 0
2025-08-01 01:13:08.977 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5a543875-8723-4e43-89ab-66b16c5a5bc0] socks connection closed
2025-08-01 01:13:08.977 [info] [command][46a793b8-6dd8-4aea-8159-bbca5e25ef03] Socket close event received
2025-08-01 01:13:09.017 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63221 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:14:08.981 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:14:08.983 [info] [command][d35005d1-8573-414c-9be3-6401d80d02a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d35005d1-8573-414c-9be3-6401d80d02a2""}
2025-08-01 01:14:08.984 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][984ba4d2-4847-4b4d-9b62-513f9e916fc9] received connection request
2025-08-01 01:14:08.984 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:14:09.031 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][984ba4d2-4847-4b4d-9b62-513f9e916fc9] socks forwarding established
2025-08-01 01:14:09.077 [info] [command][d35005d1-8573-414c-9be3-6401d80d02a2] Process exited with code 0
2025-08-01 01:14:09.077 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][984ba4d2-4847-4b4d-9b62-513f9e916fc9] socks connection closed
2025-08-01 01:14:09.077 [info] [command][d35005d1-8573-414c-9be3-6401d80d02a2] Socket close event received
2025-08-01 01:14:09.118 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63238 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:15:09.077 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:15:09.079 [info] [command][b442c77b-9411-4167-98eb-1c48e3cb9e6e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b442c77b-9411-4167-98eb-1c48e3cb9e6e""}
2025-08-01 01:15:09.080 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a1f1f0c4-f746-45bd-8266-676635b33993] received connection request
2025-08-01 01:15:09.081 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:15:09.123 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1f1f0c4-f746-45bd-8266-676635b33993] socks forwarding established
2025-08-01 01:15:09.170 [info] [command][b442c77b-9411-4167-98eb-1c48e3cb9e6e] Process exited with code 0
2025-08-01 01:15:09.170 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1f1f0c4-f746-45bd-8266-676635b33993] socks connection closed
2025-08-01 01:15:09.170 [info] [command][b442c77b-9411-4167-98eb-1c48e3cb9e6e] Socket close event received
2025-08-01 01:15:09.210 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63254 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:16:09.178 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:16:09.179 [info] [command][0c0db1db-a349-4d4d-b916-62e43210144b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0c0db1db-a349-4d4d-b916-62e43210144b""}
2025-08-01 01:16:09.180 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][30f1280d-9de1-44b6-af01-b15aad5b2e71] received connection request
2025-08-01 01:16:09.181 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:16:09.224 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][30f1280d-9de1-44b6-af01-b15aad5b2e71] socks forwarding established
2025-08-01 01:16:09.270 [info] [command][0c0db1db-a349-4d4d-b916-62e43210144b] Process exited with code 0
2025-08-01 01:16:09.270 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][30f1280d-9de1-44b6-af01-b15aad5b2e71] socks connection closed
2025-08-01 01:16:09.270 [info] [command][0c0db1db-a349-4d4d-b916-62e43210144b] Socket close event received
2025-08-01 01:16:09.314 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63270 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:17:09.272 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:17:09.274 [info] [command][d6a510b0-e959-476a-8159-2898206e136b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d6a510b0-e959-476a-8159-2898206e136b""}
2025-08-01 01:17:09.275 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c71869d9-5ceb-47d3-afa4-c1ded01ce03c] received connection request
2025-08-01 01:17:09.275 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:17:09.317 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c71869d9-5ceb-47d3-afa4-c1ded01ce03c] socks forwarding established
2025-08-01 01:17:09.361 [info] [command][d6a510b0-e959-476a-8159-2898206e136b] Process exited with code 0
2025-08-01 01:17:09.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c71869d9-5ceb-47d3-afa4-c1ded01ce03c] socks connection closed
2025-08-01 01:17:09.362 [info] [command][d6a510b0-e959-476a-8159-2898206e136b] Socket close event received
2025-08-01 01:17:09.402 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63299 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:18:09.371 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:18:09.372 [info] [command][ee383789-663e-42d4-ae4b-8ffa83c70036] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ee383789-663e-42d4-ae4b-8ffa83c70036""}
2025-08-01 01:18:09.373 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][70649bee-ad6a-4910-8f95-c7e5032bba7d] received connection request
2025-08-01 01:18:09.374 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:18:09.417 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][70649bee-ad6a-4910-8f95-c7e5032bba7d] socks forwarding established
2025-08-01 01:18:09.462 [info] [command][ee383789-663e-42d4-ae4b-8ffa83c70036] Process exited with code 0
2025-08-01 01:18:09.463 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][70649bee-ad6a-4910-8f95-c7e5032bba7d] socks connection closed
2025-08-01 01:18:09.463 [info] [command][ee383789-663e-42d4-ae4b-8ffa83c70036] Socket close event received
2025-08-01 01:18:09.504 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63331 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:19:09.470 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:19:09.472 [info] [command][450bda9f-5c0e-4373-a9b7-89aad282595b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""450bda9f-5c0e-4373-a9b7-89aad282595b""}
2025-08-01 01:19:09.472 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eff0a157-b43c-460c-909d-cfe789efe069] received connection request
2025-08-01 01:19:09.473 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:19:09.515 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eff0a157-b43c-460c-909d-cfe789efe069] socks forwarding established
2025-08-01 01:19:09.563 [info] [command][450bda9f-5c0e-4373-a9b7-89aad282595b] Process exited with code 0
2025-08-01 01:19:09.564 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eff0a157-b43c-460c-909d-cfe789efe069] socks connection closed
2025-08-01 01:19:09.564 [info] [command][450bda9f-5c0e-4373-a9b7-89aad282595b] Socket close event received
2025-08-01 01:19:09.603 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63348 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:20:09.567 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:20:09.569 [info] [command][e52828ad-29cf-4d6d-8868-829a65ebf9eb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e52828ad-29cf-4d6d-8868-829a65ebf9eb""}
2025-08-01 01:20:09.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d1bd0add-dc8e-4b06-904e-b2e88b22700a] received connection request
2025-08-01 01:20:09.571 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:20:09.613 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d1bd0add-dc8e-4b06-904e-b2e88b22700a] socks forwarding established
2025-08-01 01:20:09.660 [info] [command][e52828ad-29cf-4d6d-8868-829a65ebf9eb] Process exited with code 0
2025-08-01 01:20:09.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d1bd0add-dc8e-4b06-904e-b2e88b22700a] socks connection closed
2025-08-01 01:20:09.660 [info] [command][e52828ad-29cf-4d6d-8868-829a65ebf9eb] Socket close event received
2025-08-01 01:20:09.702 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63362 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:21:09.664 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:21:09.666 [info] [command][bd6dec5c-b993-4b5c-8277-b3919c20fe0c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bd6dec5c-b993-4b5c-8277-b3919c20fe0c""}
2025-08-01 01:21:09.666 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cf8c2740-f093-47b8-becd-1036b48e411e] received connection request
2025-08-01 01:21:09.667 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:21:09.716 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cf8c2740-f093-47b8-becd-1036b48e411e] socks forwarding established
2025-08-01 01:21:09.762 [info] [command][bd6dec5c-b993-4b5c-8277-b3919c20fe0c] Process exited with code 0
2025-08-01 01:21:09.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cf8c2740-f093-47b8-becd-1036b48e411e] socks connection closed
2025-08-01 01:21:09.762 [info] [command][bd6dec5c-b993-4b5c-8277-b3919c20fe0c] Socket close event received
2025-08-01 01:21:09.804 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63380 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:22:09.768 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:22:09.769 [info] [command][0f989f60-c20e-4525-b600-7f120a4d8a9f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0f989f60-c20e-4525-b600-7f120a4d8a9f""}
2025-08-01 01:22:09.770 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fdbccb1d-06c0-49f1-a16d-2efe41bfca57] received connection request
2025-08-01 01:22:09.771 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:22:09.820 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fdbccb1d-06c0-49f1-a16d-2efe41bfca57] socks forwarding established
2025-08-01 01:22:09.865 [info] [command][0f989f60-c20e-4525-b600-7f120a4d8a9f] Process exited with code 0
2025-08-01 01:22:09.865 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fdbccb1d-06c0-49f1-a16d-2efe41bfca57] socks connection closed
2025-08-01 01:22:09.865 [info] [command][0f989f60-c20e-4525-b600-7f120a4d8a9f] Socket close event received
2025-08-01 01:22:09.906 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63429 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:23:09.872 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:23:09.873 [info] [command][fe3369d6-0543-4e4d-b3e3-b01c91b1c59f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fe3369d6-0543-4e4d-b3e3-b01c91b1c59f""}
2025-08-01 01:23:09.873 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f7abcb4a-a68d-48c2-8aae-c208d0dff150] received connection request
2025-08-01 01:23:09.874 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:23:09.919 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f7abcb4a-a68d-48c2-8aae-c208d0dff150] socks forwarding established
2025-08-01 01:23:09.964 [info] [command][fe3369d6-0543-4e4d-b3e3-b01c91b1c59f] Process exited with code 0
2025-08-01 01:23:09.965 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f7abcb4a-a68d-48c2-8aae-c208d0dff150] socks connection closed
2025-08-01 01:23:09.965 [info] [command][fe3369d6-0543-4e4d-b3e3-b01c91b1c59f] Socket close event received
2025-08-01 01:23:10.005 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63460 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:24:09.966 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:24:09.966 [info] [command][c2dd839c-d17d-46b8-a3b3-7677391c0a41] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c2dd839c-d17d-46b8-a3b3-7677391c0a41""}
2025-08-01 01:24:09.966 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ba023b3e-b233-422f-9edc-3407d39de613] received connection request
2025-08-01 01:24:09.967 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:24:10.014 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba023b3e-b233-422f-9edc-3407d39de613] socks forwarding established
2025-08-01 01:24:10.059 [info] [command][c2dd839c-d17d-46b8-a3b3-7677391c0a41] Process exited with code 0
2025-08-01 01:24:10.060 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba023b3e-b233-422f-9edc-3407d39de613] socks connection closed
2025-08-01 01:24:10.060 [info] [command][c2dd839c-d17d-46b8-a3b3-7677391c0a41] Socket close event received
2025-08-01 01:24:10.101 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63477 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:25:10.069 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:25:10.070 [info] [command][b116567f-317b-4b52-9fa0-520a2ed77339] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b116567f-317b-4b52-9fa0-520a2ed77339""}
2025-08-01 01:25:10.071 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1e37d3db-71b7-4743-8142-bade1ea79684] received connection request
2025-08-01 01:25:10.071 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:25:10.182 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1e37d3db-71b7-4743-8142-bade1ea79684] socks forwarding established
2025-08-01 01:25:10.227 [info] [command][b116567f-317b-4b52-9fa0-520a2ed77339] Process exited with code 0
2025-08-01 01:25:10.228 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1e37d3db-71b7-4743-8142-bade1ea79684] socks connection closed
2025-08-01 01:25:10.228 [info] [command][b116567f-317b-4b52-9fa0-520a2ed77339] Socket close event received
2025-08-01 01:25:10.268 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63497 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:26:10.237 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:26:10.238 [info] [command][1b344bfa-f917-45e0-877d-03429a347b5e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1b344bfa-f917-45e0-877d-03429a347b5e""}
2025-08-01 01:26:10.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fa8c874c-c942-44d0-8854-36df52ad44b5] received connection request
2025-08-01 01:26:10.239 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 01:26:10.239 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:26:10.286 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fa8c874c-c942-44d0-8854-36df52ad44b5] socks forwarding established
2025-08-01 01:26:10.330 [info] [command][1b344bfa-f917-45e0-877d-03429a347b5e] Process exited with code 0
2025-08-01 01:26:10.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fa8c874c-c942-44d0-8854-36df52ad44b5] socks connection closed
2025-08-01 01:26:10.331 [info] [command][1b344bfa-f917-45e0-877d-03429a347b5e] Socket close event received
2025-08-01 01:26:10.371 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63515 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:27:10.337 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:27:10.338 [info] [command][3f93a764-5e90-40a6-9124-61025f60e6d6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3f93a764-5e90-40a6-9124-61025f60e6d6""}
2025-08-01 01:27:10.338 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][82578560-de7b-4a5c-8117-b7df48eb1830] received connection request
2025-08-01 01:27:10.339 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:27:10.382 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82578560-de7b-4a5c-8117-b7df48eb1830] socks forwarding established
2025-08-01 01:27:10.704 [info] [command][3f93a764-5e90-40a6-9124-61025f60e6d6] Process exited with code 0
2025-08-01 01:27:10.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82578560-de7b-4a5c-8117-b7df48eb1830] socks connection closed
2025-08-01 01:27:10.705 [info] [command][3f93a764-5e90-40a6-9124-61025f60e6d6] Socket close event received
2025-08-01 01:27:10.744 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63547 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:28:10.713 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:28:10.714 [info] [command][61aeb32e-abd6-4943-9834-684b22ba0517] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""61aeb32e-abd6-4943-9834-684b22ba0517""}
2025-08-01 01:28:10.714 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cfa7c94c-8232-4a87-8a2d-383743d3139d] received connection request
2025-08-01 01:28:10.715 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:28:10.757 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfa7c94c-8232-4a87-8a2d-383743d3139d] socks forwarding established
2025-08-01 01:28:10.803 [info] [command][61aeb32e-abd6-4943-9834-684b22ba0517] Process exited with code 0
2025-08-01 01:28:10.804 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfa7c94c-8232-4a87-8a2d-383743d3139d] socks connection closed
2025-08-01 01:28:10.804 [info] [command][61aeb32e-abd6-4943-9834-684b22ba0517] Socket close event received
2025-08-01 01:28:10.843 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63581 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:29:10.811 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:29:10.812 [info] [command][c0deb70e-8a65-4a30-aa59-18031721b0bc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c0deb70e-8a65-4a30-aa59-18031721b0bc""}
2025-08-01 01:29:10.813 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a745e2e6-93f6-440f-962f-f95c1ed57900] received connection request
2025-08-01 01:29:10.814 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:29:10.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a745e2e6-93f6-440f-962f-f95c1ed57900] socks forwarding established
2025-08-01 01:29:10.902 [info] [command][c0deb70e-8a65-4a30-aa59-18031721b0bc] Process exited with code 0
2025-08-01 01:29:10.902 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a745e2e6-93f6-440f-962f-f95c1ed57900] socks connection closed
2025-08-01 01:29:10.902 [info] [command][c0deb70e-8a65-4a30-aa59-18031721b0bc] Socket close event received
2025-08-01 01:29:10.942 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63617 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:30:10.913 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:30:10.914 [info] [command][344658fb-6fc5-499f-8ede-336e88ed1d13] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""344658fb-6fc5-499f-8ede-336e88ed1d13""}
2025-08-01 01:30:10.914 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][27031d4a-2717-4a23-907b-8572b025fc61] received connection request
2025-08-01 01:30:10.915 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:30:11.097 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27031d4a-2717-4a23-907b-8572b025fc61] socks forwarding established
2025-08-01 01:30:11.143 [info] [command][344658fb-6fc5-499f-8ede-336e88ed1d13] Process exited with code 0
2025-08-01 01:30:11.144 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][27031d4a-2717-4a23-907b-8572b025fc61] socks connection closed
2025-08-01 01:30:11.144 [info] [command][344658fb-6fc5-499f-8ede-336e88ed1d13] Socket close event received
2025-08-01 01:30:11.187 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63635 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:31:11.152 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:31:11.153 [info] [command][58c17086-eb30-4190-9bb7-8a321b89c8ab] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""58c17086-eb30-4190-9bb7-8a321b89c8ab""}
2025-08-01 01:31:11.154 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][64b29541-5ac4-49cb-833e-1d108982b490] received connection request
2025-08-01 01:31:11.155 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:31:11.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][64b29541-5ac4-49cb-833e-1d108982b490] socks forwarding established
2025-08-01 01:31:11.251 [info] [command][58c17086-eb30-4190-9bb7-8a321b89c8ab] Process exited with code 0
2025-08-01 01:31:11.252 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][64b29541-5ac4-49cb-833e-1d108982b490] socks connection closed
2025-08-01 01:31:11.252 [info] [command][58c17086-eb30-4190-9bb7-8a321b89c8ab] Socket close event received
2025-08-01 01:31:11.293 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63667 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:32:11.253 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:32:11.254 [info] [command][8408a617-6c06-4aab-8188-789a8b5cb931] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8408a617-6c06-4aab-8188-789a8b5cb931""}
2025-08-01 01:32:11.255 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ba65ecca-031f-479f-adeb-858f6d89883b] received connection request
2025-08-01 01:32:11.255 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:32:11.302 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba65ecca-031f-479f-adeb-858f6d89883b] socks forwarding established
2025-08-01 01:32:11.348 [info] [command][8408a617-6c06-4aab-8188-789a8b5cb931] Process exited with code 0
2025-08-01 01:32:11.348 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ba65ecca-031f-479f-adeb-858f6d89883b] socks connection closed
2025-08-01 01:32:11.348 [info] [command][8408a617-6c06-4aab-8188-789a8b5cb931] Socket close event received
2025-08-01 01:32:11.389 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63703 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:33:11.354 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:33:11.355 [info] [command][1f5ddfec-f744-471c-a235-1c0aee4d4cc9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1f5ddfec-f744-471c-a235-1c0aee4d4cc9""}
2025-08-01 01:33:11.356 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cf521903-9b2f-4621-8ed0-16f4f9f767fd] received connection request
2025-08-01 01:33:11.357 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:33:11.402 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cf521903-9b2f-4621-8ed0-16f4f9f767fd] socks forwarding established
2025-08-01 01:33:11.447 [info] [command][1f5ddfec-f744-471c-a235-1c0aee4d4cc9] Process exited with code 0
2025-08-01 01:33:11.447 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cf521903-9b2f-4621-8ed0-16f4f9f767fd] socks connection closed
2025-08-01 01:33:11.447 [info] [command][1f5ddfec-f744-471c-a235-1c0aee4d4cc9] Socket close event received
2025-08-01 01:33:11.486 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63733 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:34:11.458 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:34:11.459 [info] [command][a358146a-9d7f-473b-ad4b-fc25e76b1be9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a358146a-9d7f-473b-ad4b-fc25e76b1be9""}
2025-08-01 01:34:11.460 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][af5ab51f-2604-47e2-a51f-29bb10e5fbd3] received connection request
2025-08-01 01:34:11.460 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:34:11.574 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][af5ab51f-2604-47e2-a51f-29bb10e5fbd3] socks forwarding established
2025-08-01 01:34:11.620 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][af5ab51f-2604-47e2-a51f-29bb10e5fbd3] socks connection closed
2025-08-01 01:34:11.620 [info] [command][a358146a-9d7f-473b-ad4b-fc25e76b1be9] Process exited with code 0
2025-08-01 01:34:11.621 [info] [command][a358146a-9d7f-473b-ad4b-fc25e76b1be9] Socket close event received
2025-08-01 01:34:11.660 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63752 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:35:11.629 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:35:11.630 [info] [command][a6f9090a-4ab5-493f-ad42-0f8bc6df916e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a6f9090a-4ab5-493f-ad42-0f8bc6df916e""}
2025-08-01 01:35:11.631 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][22e28fff-bd64-4434-bc46-ae3e431204ab] received connection request
2025-08-01 01:35:11.632 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:35:11.903 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22e28fff-bd64-4434-bc46-ae3e431204ab] socks forwarding established
2025-08-01 01:35:11.948 [info] [command][a6f9090a-4ab5-493f-ad42-0f8bc6df916e] Process exited with code 0
2025-08-01 01:35:11.948 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22e28fff-bd64-4434-bc46-ae3e431204ab] socks connection closed
2025-08-01 01:35:11.948 [info] [command][a6f9090a-4ab5-493f-ad42-0f8bc6df916e] Socket close event received
2025-08-01 01:35:11.987 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63767 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:36:11.950 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:36:11.952 [info] [command][ffe1be7e-aebb-4d10-9c14-56f446d58f94] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ffe1be7e-aebb-4d10-9c14-56f446d58f94""}
2025-08-01 01:36:11.952 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9e26c331-8a68-4675-947e-adce0ccd130d] received connection request
2025-08-01 01:36:11.953 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:36:11.999 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9e26c331-8a68-4675-947e-adce0ccd130d] socks forwarding established
2025-08-01 01:36:12.044 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9e26c331-8a68-4675-947e-adce0ccd130d] socks connection closed
2025-08-01 01:36:12.045 [info] [command][ffe1be7e-aebb-4d10-9c14-56f446d58f94] Process exited with code 0
2025-08-01 01:36:12.045 [info] [command][ffe1be7e-aebb-4d10-9c14-56f446d58f94] Socket close event received
2025-08-01 01:36:12.083 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63786 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:37:12.054 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:37:12.055 [info] [command][afd2b046-a1c8-4180-909b-e7e98e3a7255] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""afd2b046-a1c8-4180-909b-e7e98e3a7255""}
2025-08-01 01:37:12.056 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][961bc1a3-acd4-4b4b-83a6-61d97f91319d] received connection request
2025-08-01 01:37:12.057 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:37:12.098 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][961bc1a3-acd4-4b4b-83a6-61d97f91319d] socks forwarding established
2025-08-01 01:37:12.144 [info] [command][afd2b046-a1c8-4180-909b-e7e98e3a7255] Process exited with code 0
2025-08-01 01:37:12.144 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][961bc1a3-acd4-4b4b-83a6-61d97f91319d] socks connection closed
2025-08-01 01:37:12.145 [info] [command][afd2b046-a1c8-4180-909b-e7e98e3a7255] Socket close event received
2025-08-01 01:37:12.185 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63820 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:38:12.150 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:38:12.151 [info] [command][83eda586-28fb-40e8-9b39-cc92ecfca806] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""83eda586-28fb-40e8-9b39-cc92ecfca806""}
2025-08-01 01:38:12.152 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6bc7c357-5590-4078-8b42-415d9c1e6f54] received connection request
2025-08-01 01:38:12.152 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:38:12.195 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6bc7c357-5590-4078-8b42-415d9c1e6f54] socks forwarding established
2025-08-01 01:38:12.242 [info] [command][83eda586-28fb-40e8-9b39-cc92ecfca806] Process exited with code 0
2025-08-01 01:38:12.242 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6bc7c357-5590-4078-8b42-415d9c1e6f54] socks connection closed
2025-08-01 01:38:12.242 [info] [command][83eda586-28fb-40e8-9b39-cc92ecfca806] Socket close event received
2025-08-01 01:38:12.282 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63865 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:39:12.245 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:39:12.246 [info] [command][4af221e4-1b97-447d-b131-3567461622e4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4af221e4-1b97-447d-b131-3567461622e4""}
2025-08-01 01:39:12.247 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4a6ab4d7-3961-4ede-8b51-b84f647593d1] received connection request
2025-08-01 01:39:12.248 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:39:12.295 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4a6ab4d7-3961-4ede-8b51-b84f647593d1] socks forwarding established
2025-08-01 01:39:12.340 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4a6ab4d7-3961-4ede-8b51-b84f647593d1] socks connection closed
2025-08-01 01:39:12.340 [info] [command][4af221e4-1b97-447d-b131-3567461622e4] Process exited with code 0
2025-08-01 01:39:12.340 [info] [command][4af221e4-1b97-447d-b131-3567461622e4] Socket close event received
2025-08-01 01:39:12.381 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63882 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:40:12.345 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:40:12.346 [info] [command][23777c2d-a31e-46d8-adae-6f9c80dec753] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""23777c2d-a31e-46d8-adae-6f9c80dec753""}
2025-08-01 01:40:12.346 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c1688283-ab3c-41fc-b257-8220f15be8b6] received connection request
2025-08-01 01:40:12.346 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:40:12.394 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1688283-ab3c-41fc-b257-8220f15be8b6] socks forwarding established
2025-08-01 01:40:12.438 [info] [command][23777c2d-a31e-46d8-adae-6f9c80dec753] Process exited with code 0
2025-08-01 01:40:12.439 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1688283-ab3c-41fc-b257-8220f15be8b6] socks connection closed
2025-08-01 01:40:12.439 [info] [command][23777c2d-a31e-46d8-adae-6f9c80dec753] Socket close event received
2025-08-01 01:40:12.479 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63900 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:41:12.449 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:41:12.450 [info] [command][eeac370d-8bd3-4d44-b379-792cb2048900] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eeac370d-8bd3-4d44-b379-792cb2048900""}
2025-08-01 01:41:12.450 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][341abe72-389c-495a-b6bf-8801a6e383ff] received connection request
2025-08-01 01:41:12.451 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:41:12.495 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][341abe72-389c-495a-b6bf-8801a6e383ff] socks forwarding established
2025-08-01 01:41:12.550 [info] [command][eeac370d-8bd3-4d44-b379-792cb2048900] Process exited with code 0
2025-08-01 01:41:12.551 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][341abe72-389c-495a-b6bf-8801a6e383ff] socks connection closed
2025-08-01 01:41:12.551 [info] [command][eeac370d-8bd3-4d44-b379-792cb2048900] Socket close event received
2025-08-01 01:41:12.592 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63921 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:42:12.552 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:42:12.554 [info] [command][7e7cb1b2-0c26-480e-b25c-2b6235505459] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7e7cb1b2-0c26-480e-b25c-2b6235505459""}
2025-08-01 01:42:12.555 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e20cb6a2-7a3c-4a87-be2e-22492ed6f974] received connection request
2025-08-01 01:42:12.555 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:42:12.604 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e20cb6a2-7a3c-4a87-be2e-22492ed6f974] socks forwarding established
2025-08-01 01:42:12.651 [info] [command][7e7cb1b2-0c26-480e-b25c-2b6235505459] Process exited with code 0
2025-08-01 01:42:12.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e20cb6a2-7a3c-4a87-be2e-22492ed6f974] socks connection closed
2025-08-01 01:42:12.651 [info] [command][7e7cb1b2-0c26-480e-b25c-2b6235505459] Socket close event received
2025-08-01 01:42:12.693 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63953 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:43:12.656 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:43:12.657 [info] [command][4cfe9d24-e78d-42f6-a177-f8f056ae027d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4cfe9d24-e78d-42f6-a177-f8f056ae027d""}
2025-08-01 01:43:12.658 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dedb7e2e-3583-4e49-9cec-39a935c39eb5] received connection request
2025-08-01 01:43:12.658 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:43:12.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dedb7e2e-3583-4e49-9cec-39a935c39eb5] socks forwarding established
2025-08-01 01:43:12.750 [info] [command][4cfe9d24-e78d-42f6-a177-f8f056ae027d] Process exited with code 0
2025-08-01 01:43:12.750 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dedb7e2e-3583-4e49-9cec-39a935c39eb5] socks connection closed
2025-08-01 01:43:12.750 [info] [command][4cfe9d24-e78d-42f6-a177-f8f056ae027d] Socket close event received
2025-08-01 01:43:12.827 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63981 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:44:12.760 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:44:12.762 [info] [command][aecd7e78-362d-403a-a87c-e62718311b03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""aecd7e78-362d-403a-a87c-e62718311b03""}
2025-08-01 01:44:12.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3e1fda7-d2b6-4547-b6bb-3e6e9d9e8c7a] received connection request
2025-08-01 01:44:12.763 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:44:12.839 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3e1fda7-d2b6-4547-b6bb-3e6e9d9e8c7a] socks forwarding established
2025-08-01 01:44:12.883 [info] [command][aecd7e78-362d-403a-a87c-e62718311b03] Process exited with code 0
2025-08-01 01:44:12.883 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3e1fda7-d2b6-4547-b6bb-3e6e9d9e8c7a] socks connection closed
2025-08-01 01:44:12.883 [info] [command][aecd7e78-362d-403a-a87c-e62718311b03] Socket close event received
2025-08-01 01:44:12.926 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 63997 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:45:12.885 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:45:12.886 [info] [command][c9950b2d-3f8b-4d64-a727-dd57345eda39] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c9950b2d-3f8b-4d64-a727-dd57345eda39""}
2025-08-01 01:45:12.887 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][12315e23-2798-4939-b971-1fb45283a048] received connection request
2025-08-01 01:45:12.887 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:45:12.935 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][12315e23-2798-4939-b971-1fb45283a048] socks forwarding established
2025-08-01 01:45:12.981 [info] [command][c9950b2d-3f8b-4d64-a727-dd57345eda39] Process exited with code 0
2025-08-01 01:45:12.981 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][12315e23-2798-4939-b971-1fb45283a048] socks connection closed
2025-08-01 01:45:12.981 [info] [command][c9950b2d-3f8b-4d64-a727-dd57345eda39] Socket close event received
2025-08-01 01:45:13.020 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64029 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:46:12.988 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:46:12.989 [info] [command][2e2c9bf6-9693-4b30-bfd2-f29255ddb071] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2e2c9bf6-9693-4b30-bfd2-f29255ddb071""}
2025-08-01 01:46:12.990 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][836de5a2-d5f3-495f-b22d-ff3a69a7ff3e] received connection request
2025-08-01 01:46:12.991 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:46:13.040 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][836de5a2-d5f3-495f-b22d-ff3a69a7ff3e] socks forwarding established
2025-08-01 01:46:13.083 [info] [command][2e2c9bf6-9693-4b30-bfd2-f29255ddb071] Process exited with code 0
2025-08-01 01:46:13.083 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][836de5a2-d5f3-495f-b22d-ff3a69a7ff3e] socks connection closed
2025-08-01 01:46:13.083 [info] [command][2e2c9bf6-9693-4b30-bfd2-f29255ddb071] Socket close event received
2025-08-01 01:46:13.123 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64049 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:47:13.088 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:47:13.090 [info] [command][78bd4986-0e7d-4541-8b61-f2a046c86ba3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""78bd4986-0e7d-4541-8b61-f2a046c86ba3""}
2025-08-01 01:47:13.091 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][396df133-b18c-4606-8f4e-45919fd43ab6] received connection request
2025-08-01 01:47:13.092 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:47:13.137 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][396df133-b18c-4606-8f4e-45919fd43ab6] socks forwarding established
2025-08-01 01:47:13.181 [info] [command][78bd4986-0e7d-4541-8b61-f2a046c86ba3] Process exited with code 0
2025-08-01 01:47:13.181 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][396df133-b18c-4606-8f4e-45919fd43ab6] socks connection closed
2025-08-01 01:47:13.181 [info] [command][78bd4986-0e7d-4541-8b61-f2a046c86ba3] Socket close event received
2025-08-01 01:47:13.222 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64087 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:48:13.189 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:48:13.190 [info] [command][f9c82154-68e4-45ef-b135-0787975c723f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f9c82154-68e4-45ef-b135-0787975c723f""}
2025-08-01 01:48:13.191 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a1d19bdc-0146-4fa5-81d1-abde47602b36] received connection request
2025-08-01 01:48:13.191 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:48:13.237 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1d19bdc-0146-4fa5-81d1-abde47602b36] socks forwarding established
2025-08-01 01:48:13.282 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1d19bdc-0146-4fa5-81d1-abde47602b36] socks connection closed
2025-08-01 01:48:13.283 [info] [command][f9c82154-68e4-45ef-b135-0787975c723f] Process exited with code 0
2025-08-01 01:48:13.283 [info] [command][f9c82154-68e4-45ef-b135-0787975c723f] Socket close event received
2025-08-01 01:48:13.322 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64115 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:49:13.284 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:49:13.284 [info] [command][da9eb05f-2ef4-41c4-8897-b1d3f04b8728] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""da9eb05f-2ef4-41c4-8897-b1d3f04b8728""}
2025-08-01 01:49:13.285 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a6bd584c-f33a-4d0b-a268-d5b5b231388c] received connection request
2025-08-01 01:49:13.285 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:49:13.397 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a6bd584c-f33a-4d0b-a268-d5b5b231388c] socks forwarding established
2025-08-01 01:49:13.440 [info] [command][da9eb05f-2ef4-41c4-8897-b1d3f04b8728] Process exited with code 0
2025-08-01 01:49:13.440 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a6bd584c-f33a-4d0b-a268-d5b5b231388c] socks connection closed
2025-08-01 01:49:13.440 [info] [command][da9eb05f-2ef4-41c4-8897-b1d3f04b8728] Socket close event received
2025-08-01 01:49:13.503 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64133 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:50:13.444 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:50:13.446 [info] [command][a4bbb666-9720-4a82-ad51-6e46037a4959] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a4bbb666-9720-4a82-ad51-6e46037a4959""}
2025-08-01 01:50:13.446 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][366e53fb-18e3-4c4c-9913-34b16bddf495] received connection request
2025-08-01 01:50:13.447 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:50:13.492 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][366e53fb-18e3-4c4c-9913-34b16bddf495] socks forwarding established
2025-08-01 01:50:13.540 [info] [command][a4bbb666-9720-4a82-ad51-6e46037a4959] Process exited with code 0
2025-08-01 01:50:13.540 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][366e53fb-18e3-4c4c-9913-34b16bddf495] socks connection closed
2025-08-01 01:50:13.540 [info] [command][a4bbb666-9720-4a82-ad51-6e46037a4959] Socket close event received
2025-08-01 01:50:13.580 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64150 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:51:13.545 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:51:13.546 [info] [command][829472cf-f8e2-425e-838e-ed1ff216e106] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""829472cf-f8e2-425e-838e-ed1ff216e106""}
2025-08-01 01:51:13.547 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1d80cc8c-dfbc-4f1d-b22b-4a46bb1ccbad] received connection request
2025-08-01 01:51:13.548 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:51:13.593 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d80cc8c-dfbc-4f1d-b22b-4a46bb1ccbad] socks forwarding established
2025-08-01 01:51:13.639 [info] [command][829472cf-f8e2-425e-838e-ed1ff216e106] Process exited with code 0
2025-08-01 01:51:13.639 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d80cc8c-dfbc-4f1d-b22b-4a46bb1ccbad] socks connection closed
2025-08-01 01:51:13.639 [info] [command][829472cf-f8e2-425e-838e-ed1ff216e106] Socket close event received
2025-08-01 01:51:13.679 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64167 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:52:13.645 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:52:13.646 [info] [command][e39a246d-90eb-4254-8e0f-684942e7ea1a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e39a246d-90eb-4254-8e0f-684942e7ea1a""}
2025-08-01 01:52:13.647 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c4600261-97fc-4707-8622-dfd199536cf1] received connection request
2025-08-01 01:52:13.647 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:52:13.687 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4600261-97fc-4707-8622-dfd199536cf1] socks forwarding established
2025-08-01 01:52:13.732 [info] [command][e39a246d-90eb-4254-8e0f-684942e7ea1a] Process exited with code 0
2025-08-01 01:52:13.733 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4600261-97fc-4707-8622-dfd199536cf1] socks connection closed
2025-08-01 01:52:13.733 [info] [command][e39a246d-90eb-4254-8e0f-684942e7ea1a] Socket close event received
2025-08-01 01:52:13.773 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64194 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:53:13.742 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:53:13.744 [info] [command][eab3afb2-3133-468a-8b58-e487e9aa7e75] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eab3afb2-3133-468a-8b58-e487e9aa7e75""}
2025-08-01 01:53:13.744 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5aa56771-1bea-4d28-8d6e-b538bb95340c] received connection request
2025-08-01 01:53:13.744 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:53:13.786 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5aa56771-1bea-4d28-8d6e-b538bb95340c] socks forwarding established
2025-08-01 01:53:13.833 [info] [command][eab3afb2-3133-468a-8b58-e487e9aa7e75] Process exited with code 0
2025-08-01 01:53:13.833 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5aa56771-1bea-4d28-8d6e-b538bb95340c] socks connection closed
2025-08-01 01:53:13.833 [info] [command][eab3afb2-3133-468a-8b58-e487e9aa7e75] Socket close event received
2025-08-01 01:53:13.928 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64241 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:54:13.837 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:54:13.837 [info] [command][c01239cb-28fe-401f-88a5-a7b9a508bfc1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c01239cb-28fe-401f-88a5-a7b9a508bfc1""}
2025-08-01 01:54:13.838 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6f8c91f3-f357-4d13-87fe-4e3c682327c1] received connection request
2025-08-01 01:54:13.838 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:54:13.883 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f8c91f3-f357-4d13-87fe-4e3c682327c1] socks forwarding established
2025-08-01 01:54:13.928 [info] [command][c01239cb-28fe-401f-88a5-a7b9a508bfc1] Process exited with code 0
2025-08-01 01:54:13.928 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f8c91f3-f357-4d13-87fe-4e3c682327c1] socks connection closed
2025-08-01 01:54:13.928 [info] [command][c01239cb-28fe-401f-88a5-a7b9a508bfc1] Socket close event received
2025-08-01 01:54:13.968 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64253 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:55:13.937 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:55:13.939 [info] [command][aa9d2e6d-9471-41fa-871f-9e38d3b10062] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""aa9d2e6d-9471-41fa-871f-9e38d3b10062""}
2025-08-01 01:55:13.939 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ef345225-bd27-4b26-901e-ed2e358c7321] received connection request
2025-08-01 01:55:13.940 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:55:13.981 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ef345225-bd27-4b26-901e-ed2e358c7321] socks forwarding established
2025-08-01 01:55:14.025 [info] [command][aa9d2e6d-9471-41fa-871f-9e38d3b10062] Process exited with code 0
2025-08-01 01:55:14.026 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ef345225-bd27-4b26-901e-ed2e358c7321] socks connection closed
2025-08-01 01:55:14.026 [info] [command][aa9d2e6d-9471-41fa-871f-9e38d3b10062] Socket close event received
2025-08-01 01:55:14.065 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64266 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:56:14.034 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:56:14.035 [info] [command][7fecb6d3-e501-4151-8edb-9da47c2565ed] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7fecb6d3-e501-4151-8edb-9da47c2565ed""}
2025-08-01 01:56:14.036 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8ff73cd5-3e46-410d-9c55-6ae7921f3cfd] received connection request
2025-08-01 01:56:14.036 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:56:14.079 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ff73cd5-3e46-410d-9c55-6ae7921f3cfd] socks forwarding established
2025-08-01 01:56:14.124 [info] [command][7fecb6d3-e501-4151-8edb-9da47c2565ed] Process exited with code 0
2025-08-01 01:56:14.125 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ff73cd5-3e46-410d-9c55-6ae7921f3cfd] socks connection closed
2025-08-01 01:56:14.125 [info] [command][7fecb6d3-e501-4151-8edb-9da47c2565ed] Socket close event received
2025-08-01 01:56:14.163 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64286 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:57:14.134 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:57:14.135 [info] [command][794b5822-ed0b-4891-8718-cbada824b27a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""794b5822-ed0b-4891-8718-cbada824b27a""}
2025-08-01 01:57:14.136 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1092c163-a861-4b52-bec9-531ee97b15ae] received connection request
2025-08-01 01:57:14.137 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:57:14.180 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1092c163-a861-4b52-bec9-531ee97b15ae] socks forwarding established
2025-08-01 01:57:14.226 [info] [command][794b5822-ed0b-4891-8718-cbada824b27a] Process exited with code 0
2025-08-01 01:57:14.226 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1092c163-a861-4b52-bec9-531ee97b15ae] socks connection closed
2025-08-01 01:57:14.226 [info] [command][794b5822-ed0b-4891-8718-cbada824b27a] Socket close event received
2025-08-01 01:57:14.265 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64323 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:58:14.230 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:58:14.231 [info] [command][f3a040b1-0cd6-4b80-bfee-d8e0b6254dd5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f3a040b1-0cd6-4b80-bfee-d8e0b6254dd5""}
2025-08-01 01:58:14.232 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1ae51432-e28a-4617-b723-5b43878bebdd] received connection request
2025-08-01 01:58:14.232 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:58:14.272 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1ae51432-e28a-4617-b723-5b43878bebdd] socks forwarding established
2025-08-01 01:58:14.320 [info] [command][f3a040b1-0cd6-4b80-bfee-d8e0b6254dd5] Process exited with code 0
2025-08-01 01:58:14.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1ae51432-e28a-4617-b723-5b43878bebdd] socks connection closed
2025-08-01 01:58:14.320 [info] [command][f3a040b1-0cd6-4b80-bfee-d8e0b6254dd5] Socket close event received
2025-08-01 01:58:14.361 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64354 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 01:59:14.323 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 01:59:14.325 [info] [command][809878c0-5e80-4e46-8dbd-bba3dcf27c87] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""809878c0-5e80-4e46-8dbd-bba3dcf27c87""}
2025-08-01 01:59:14.326 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][89e62db8-272a-48af-aaf5-78298c8bddef] received connection request
2025-08-01 01:59:14.326 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 01:59:14.367 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89e62db8-272a-48af-aaf5-78298c8bddef] socks forwarding established
2025-08-01 01:59:14.412 [info] [command][809878c0-5e80-4e46-8dbd-bba3dcf27c87] Process exited with code 0
2025-08-01 01:59:14.413 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89e62db8-272a-48af-aaf5-78298c8bddef] socks connection closed
2025-08-01 01:59:14.413 [info] [command][809878c0-5e80-4e46-8dbd-bba3dcf27c87] Socket close event received
2025-08-01 01:59:14.452 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64382 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:00:14.415 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:00:14.416 [info] [command][db70ab3c-e136-4302-b900-697e30ee1feb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""db70ab3c-e136-4302-b900-697e30ee1feb""}
2025-08-01 02:00:14.417 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][07d7fa49-6ed5-4abe-890d-dcac76f5528a] received connection request
2025-08-01 02:00:14.418 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:00:14.463 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07d7fa49-6ed5-4abe-890d-dcac76f5528a] socks forwarding established
2025-08-01 02:00:14.508 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07d7fa49-6ed5-4abe-890d-dcac76f5528a] socks connection closed
2025-08-01 02:00:14.508 [info] [command][db70ab3c-e136-4302-b900-697e30ee1feb] Process exited with code 0
2025-08-01 02:00:14.508 [info] [command][db70ab3c-e136-4302-b900-697e30ee1feb] Socket close event received
2025-08-01 02:00:14.547 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64398 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:01:14.519 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:01:14.521 [info] [command][b06ecabd-8691-4f05-8bf5-e72e5fc23a83] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b06ecabd-8691-4f05-8bf5-e72e5fc23a83""}
2025-08-01 02:01:14.521 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2c936507-f2cc-4aa6-9d53-864ddb12d922] received connection request
2025-08-01 02:01:14.522 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:01:14.571 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c936507-f2cc-4aa6-9d53-864ddb12d922] socks forwarding established
2025-08-01 02:01:14.616 [info] [command][b06ecabd-8691-4f05-8bf5-e72e5fc23a83] Process exited with code 0
2025-08-01 02:01:14.616 [info] [command][b06ecabd-8691-4f05-8bf5-e72e5fc23a83] Socket close event received
2025-08-01 02:01:14.655 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c936507-f2cc-4aa6-9d53-864ddb12d922] socks connection closed
2025-08-01 02:01:14.656 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64412 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:02:14.620 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:02:14.623 [info] [command][62656ce7-2486-47e5-96ab-212faf303deb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""62656ce7-2486-47e5-96ab-212faf303deb""}
2025-08-01 02:02:14.623 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cbe0e623-d556-4fa9-9927-bcbb822111a7] received connection request
2025-08-01 02:02:14.624 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:02:14.667 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cbe0e623-d556-4fa9-9927-bcbb822111a7] socks forwarding established
2025-08-01 02:02:14.714 [info] [command][62656ce7-2486-47e5-96ab-212faf303deb] Process exited with code 0
2025-08-01 02:02:14.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cbe0e623-d556-4fa9-9927-bcbb822111a7] socks connection closed
2025-08-01 02:02:14.715 [info] [command][62656ce7-2486-47e5-96ab-212faf303deb] Socket close event received
2025-08-01 02:02:14.757 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64442 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:03:14.719 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:03:14.720 [info] [command][f76c7d36-843f-4001-913c-613560616a0e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f76c7d36-843f-4001-913c-613560616a0e""}
2025-08-01 02:03:14.721 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bcfa034f-1d90-4549-a8a8-3b1591a3b892] received connection request
2025-08-01 02:03:14.722 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:03:14.990 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bcfa034f-1d90-4549-a8a8-3b1591a3b892] socks forwarding established
2025-08-01 02:03:15.034 [info] [command][f76c7d36-843f-4001-913c-613560616a0e] Process exited with code 0
2025-08-01 02:03:15.034 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bcfa034f-1d90-4549-a8a8-3b1591a3b892] socks connection closed
2025-08-01 02:03:15.034 [info] [command][f76c7d36-843f-4001-913c-613560616a0e] Socket close event received
2025-08-01 02:03:15.074 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64470 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:04:15.041 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:04:15.042 [info] [command][3a32bf33-9d32-4ca3-a9c1-f6aa0d969ff7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3a32bf33-9d32-4ca3-a9c1-f6aa0d969ff7""}
2025-08-01 02:04:15.042 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0c6b9bcf-edd6-4fec-9bed-d530a78633cb] received connection request
2025-08-01 02:04:15.043 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:04:15.089 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c6b9bcf-edd6-4fec-9bed-d530a78633cb] socks forwarding established
2025-08-01 02:04:15.135 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c6b9bcf-edd6-4fec-9bed-d530a78633cb] socks connection closed
2025-08-01 02:04:15.136 [info] [command][3a32bf33-9d32-4ca3-a9c1-f6aa0d969ff7] Process exited with code 0
2025-08-01 02:04:15.136 [info] [command][3a32bf33-9d32-4ca3-a9c1-f6aa0d969ff7] Socket close event received
2025-08-01 02:04:15.176 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64486 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:05:15.136 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:05:15.138 [info] [command][f19839ef-80a6-4984-b46d-0cd8d786f548] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f19839ef-80a6-4984-b46d-0cd8d786f548""}
2025-08-01 02:05:15.138 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5b1af534-1c4a-4099-8cb4-9b6e037c3f6f] received connection request
2025-08-01 02:05:15.139 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:05:15.182 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b1af534-1c4a-4099-8cb4-9b6e037c3f6f] socks forwarding established
2025-08-01 02:05:15.231 [info] [command][f19839ef-80a6-4984-b46d-0cd8d786f548] Process exited with code 0
2025-08-01 02:05:15.231 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b1af534-1c4a-4099-8cb4-9b6e037c3f6f] socks connection closed
2025-08-01 02:05:15.231 [info] [command][f19839ef-80a6-4984-b46d-0cd8d786f548] Socket close event received
2025-08-01 02:05:15.270 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64499 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:06:15.239 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:06:15.241 [info] [command][69c7ab57-3939-433c-829c-ab11730879fc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""69c7ab57-3939-433c-829c-ab11730879fc""}
2025-08-01 02:06:15.242 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][627f24f6-4eee-4fb8-9aac-a1f4e5c67cf6] received connection request
2025-08-01 02:06:15.243 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:06:15.284 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][627f24f6-4eee-4fb8-9aac-a1f4e5c67cf6] socks forwarding established
2025-08-01 02:06:15.328 [info] [command][69c7ab57-3939-433c-829c-ab11730879fc] Process exited with code 0
2025-08-01 02:06:15.328 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][627f24f6-4eee-4fb8-9aac-a1f4e5c67cf6] socks connection closed
2025-08-01 02:06:15.328 [info] [command][69c7ab57-3939-433c-829c-ab11730879fc] Socket close event received
2025-08-01 02:06:15.366 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64512 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:07:15.336 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:07:15.338 [info] [command][85554a89-04e8-4d85-bb6d-f471b9660989] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""85554a89-04e8-4d85-bb6d-f471b9660989""}
2025-08-01 02:07:15.339 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5a6e7a39-bf65-4cec-a90e-2e711b244080] received connection request
2025-08-01 02:07:15.340 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:07:15.440 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5a6e7a39-bf65-4cec-a90e-2e711b244080] socks forwarding established
2025-08-01 02:07:15.489 [info] [command][85554a89-04e8-4d85-bb6d-f471b9660989] Process exited with code 0
2025-08-01 02:07:15.489 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5a6e7a39-bf65-4cec-a90e-2e711b244080] socks connection closed
2025-08-01 02:07:15.490 [info] [command][85554a89-04e8-4d85-bb6d-f471b9660989] Socket close event received
2025-08-01 02:07:15.530 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64543 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:08:15.499 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:08:15.500 [info] [command][cec86da1-c061-4ffa-8d81-6d4e284e8581] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cec86da1-c061-4ffa-8d81-6d4e284e8581""}
2025-08-01 02:08:15.501 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a8858d48-2af8-4bdc-a2ba-8ea10eed82b1] received connection request
2025-08-01 02:08:15.501 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:08:15.546 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8858d48-2af8-4bdc-a2ba-8ea10eed82b1] socks forwarding established
2025-08-01 02:08:15.591 [info] [command][cec86da1-c061-4ffa-8d81-6d4e284e8581] Process exited with code 0
2025-08-01 02:08:15.591 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8858d48-2af8-4bdc-a2ba-8ea10eed82b1] socks connection closed
2025-08-01 02:08:15.591 [info] [command][cec86da1-c061-4ffa-8d81-6d4e284e8581] Socket close event received
2025-08-01 02:08:15.632 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64573 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:09:15.599 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:09:15.600 [info] [command][9ff4a21d-3781-48f7-b53a-dd9cb33f443e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9ff4a21d-3781-48f7-b53a-dd9cb33f443e""}
2025-08-01 02:09:15.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5d61007c-0952-482b-9a02-95a10a6f4510] received connection request
2025-08-01 02:09:15.601 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:09:15.645 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d61007c-0952-482b-9a02-95a10a6f4510] socks forwarding established
2025-08-01 02:09:15.689 [info] [command][9ff4a21d-3781-48f7-b53a-dd9cb33f443e] Process exited with code 0
2025-08-01 02:09:15.689 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d61007c-0952-482b-9a02-95a10a6f4510] socks connection closed
2025-08-01 02:09:15.689 [info] [command][9ff4a21d-3781-48f7-b53a-dd9cb33f443e] Socket close event received
2025-08-01 02:09:15.729 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64593 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:10:15.699 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:10:15.701 [info] [command][de4d6af2-a083-4a25-b873-225c46dc6bd2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""de4d6af2-a083-4a25-b873-225c46dc6bd2""}
2025-08-01 02:10:15.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][14d5ae80-e329-405c-b21d-69d7105299b5] received connection request
2025-08-01 02:10:15.702 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:10:15.749 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14d5ae80-e329-405c-b21d-69d7105299b5] socks forwarding established
2025-08-01 02:10:15.794 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14d5ae80-e329-405c-b21d-69d7105299b5] socks connection closed
2025-08-01 02:10:15.794 [info] [command][de4d6af2-a083-4a25-b873-225c46dc6bd2] Process exited with code 0
2025-08-01 02:10:15.794 [info] [command][de4d6af2-a083-4a25-b873-225c46dc6bd2] Socket close event received
2025-08-01 02:10:15.837 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64612 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:11:15.805 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:11:15.807 [info] [command][1b1e56c3-0966-4170-ab58-d7139d6fd56c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1b1e56c3-0966-4170-ab58-d7139d6fd56c""}
2025-08-01 02:11:15.808 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][103b7f1a-7cc7-410c-80e5-137d37faa25f] received connection request
2025-08-01 02:11:15.808 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:11:15.855 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][103b7f1a-7cc7-410c-80e5-137d37faa25f] socks forwarding established
2025-08-01 02:11:15.899 [info] [command][1b1e56c3-0966-4170-ab58-d7139d6fd56c] Process exited with code 0
2025-08-01 02:11:15.900 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][103b7f1a-7cc7-410c-80e5-137d37faa25f] socks connection closed
2025-08-01 02:11:15.900 [info] [command][1b1e56c3-0966-4170-ab58-d7139d6fd56c] Socket close event received
2025-08-01 02:11:15.940 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64624 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:12:15.909 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:12:15.911 [info] [command][b5d05877-c829-4421-a522-0d00a7d2216d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b5d05877-c829-4421-a522-0d00a7d2216d""}
2025-08-01 02:12:15.912 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5738d1cc-bb0a-4a59-8c8f-c56b89cb6572] received connection request
2025-08-01 02:12:15.913 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:12:15.990 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5738d1cc-bb0a-4a59-8c8f-c56b89cb6572] socks forwarding established
2025-08-01 02:12:16.034 [info] [command][b5d05877-c829-4421-a522-0d00a7d2216d] Process exited with code 0
2025-08-01 02:12:16.034 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5738d1cc-bb0a-4a59-8c8f-c56b89cb6572] socks connection closed
2025-08-01 02:12:16.034 [info] [command][b5d05877-c829-4421-a522-0d00a7d2216d] Socket close event received
2025-08-01 02:12:16.076 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64657 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:13:16.041 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:13:16.043 [info] [command][6ba884a8-d17c-40f0-8646-4f38714aa04e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6ba884a8-d17c-40f0-8646-4f38714aa04e""}
2025-08-01 02:13:16.044 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0c7819e3-ed0e-436a-9ce4-6a30a58b16c8] received connection request
2025-08-01 02:13:16.044 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:13:16.086 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c7819e3-ed0e-436a-9ce4-6a30a58b16c8] socks forwarding established
2025-08-01 02:13:16.130 [info] [command][6ba884a8-d17c-40f0-8646-4f38714aa04e] Process exited with code 0
2025-08-01 02:13:16.130 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c7819e3-ed0e-436a-9ce4-6a30a58b16c8] socks connection closed
2025-08-01 02:13:16.130 [info] [command][6ba884a8-d17c-40f0-8646-4f38714aa04e] Socket close event received
2025-08-01 02:13:16.171 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64702 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:14:16.131 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:14:16.132 [info] [command][118d7b48-6dc0-4604-af55-2d95e59aee6f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""118d7b48-6dc0-4604-af55-2d95e59aee6f""}
2025-08-01 02:14:16.133 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2a7fd5bd-c2bc-4747-9abb-2e7363f951e9] received connection request
2025-08-01 02:14:16.134 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:14:16.182 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2a7fd5bd-c2bc-4747-9abb-2e7363f951e9] socks forwarding established
2025-08-01 02:14:16.229 [info] [command][118d7b48-6dc0-4604-af55-2d95e59aee6f] Process exited with code 0
2025-08-01 02:14:16.229 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2a7fd5bd-c2bc-4747-9abb-2e7363f951e9] socks connection closed
2025-08-01 02:14:16.229 [info] [command][118d7b48-6dc0-4604-af55-2d95e59aee6f] Socket close event received
2025-08-01 02:14:16.271 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64718 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:15:16.238 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:15:16.239 [info] [command][ef474535-69cd-4b31-b41e-df48c247b43d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ef474535-69cd-4b31-b41e-df48c247b43d""}
2025-08-01 02:15:16.240 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b30c996e-4c4f-4686-bd16-22480ce6bc89] received connection request
2025-08-01 02:15:16.240 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:15:16.287 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b30c996e-4c4f-4686-bd16-22480ce6bc89] socks forwarding established
2025-08-01 02:15:16.333 [info] [command][ef474535-69cd-4b31-b41e-df48c247b43d] Process exited with code 0
2025-08-01 02:15:16.333 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b30c996e-4c4f-4686-bd16-22480ce6bc89] socks connection closed
2025-08-01 02:15:16.334 [info] [command][ef474535-69cd-4b31-b41e-df48c247b43d] Socket close event received
2025-08-01 02:15:16.373 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64736 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:16:16.341 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:16:16.344 [info] [command][515c9cef-c167-4c7c-981b-47411e9bcf64] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""515c9cef-c167-4c7c-981b-47411e9bcf64""}
2025-08-01 02:16:16.345 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][33f7cd8b-cf62-4590-96cc-5959c7d432ac] received connection request
2025-08-01 02:16:16.346 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:16:16.438 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33f7cd8b-cf62-4590-96cc-5959c7d432ac] socks forwarding established
2025-08-01 02:16:16.484 [info] [command][515c9cef-c167-4c7c-981b-47411e9bcf64] Process exited with code 0
2025-08-01 02:16:16.484 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33f7cd8b-cf62-4590-96cc-5959c7d432ac] socks connection closed
2025-08-01 02:16:16.485 [info] [command][515c9cef-c167-4c7c-981b-47411e9bcf64] Socket close event received
2025-08-01 02:16:16.524 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64751 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:17:16.493 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:17:16.495 [info] [command][eb7dbf2e-8ad5-44fd-bb6f-376cf3d9d950] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eb7dbf2e-8ad5-44fd-bb6f-376cf3d9d950""}
2025-08-01 02:17:16.496 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4d93283d-a4ce-4155-8384-e599443a19c1] received connection request
2025-08-01 02:17:16.497 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:17:16.539 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d93283d-a4ce-4155-8384-e599443a19c1] socks forwarding established
2025-08-01 02:17:16.603 [info] [command][eb7dbf2e-8ad5-44fd-bb6f-376cf3d9d950] Process exited with code 0
2025-08-01 02:17:16.604 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d93283d-a4ce-4155-8384-e599443a19c1] socks connection closed
2025-08-01 02:17:16.604 [info] [command][eb7dbf2e-8ad5-44fd-bb6f-376cf3d9d950] Socket close event received
2025-08-01 02:17:16.646 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64784 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:18:16.614 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:18:16.616 [info] [command][0375b2e6-aa03-4700-a0c4-ece9e4580f26] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0375b2e6-aa03-4700-a0c4-ece9e4580f26""}
2025-08-01 02:18:16.616 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][add2273b-df98-497a-b73b-ec1314811729] received connection request
2025-08-01 02:18:16.617 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:18:16.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][add2273b-df98-497a-b73b-ec1314811729] socks forwarding established
2025-08-01 02:18:16.706 [info] [command][0375b2e6-aa03-4700-a0c4-ece9e4580f26] Process exited with code 0
2025-08-01 02:18:16.706 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][add2273b-df98-497a-b73b-ec1314811729] socks connection closed
2025-08-01 02:18:16.706 [info] [command][0375b2e6-aa03-4700-a0c4-ece9e4580f26] Socket close event received
2025-08-01 02:18:16.750 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64812 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:19:16.707 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:19:16.709 [info] [command][d6fa29f4-19df-442d-86fa-9a953e68eda9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d6fa29f4-19df-442d-86fa-9a953e68eda9""}
2025-08-01 02:19:16.709 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7e43b752-a93b-474e-9fe7-bbcc5bd8bd4a] received connection request
2025-08-01 02:19:16.710 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:19:16.756 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7e43b752-a93b-474e-9fe7-bbcc5bd8bd4a] socks forwarding established
2025-08-01 02:19:16.806 [info] [command][d6fa29f4-19df-442d-86fa-9a953e68eda9] Process exited with code 0
2025-08-01 02:19:16.806 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7e43b752-a93b-474e-9fe7-bbcc5bd8bd4a] socks connection closed
2025-08-01 02:19:16.806 [info] [command][d6fa29f4-19df-442d-86fa-9a953e68eda9] Socket close event received
2025-08-01 02:19:16.846 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64826 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:20:16.812 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:20:16.814 [info] [command][e2090bbd-efe2-488c-8af2-409b777e8467] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e2090bbd-efe2-488c-8af2-409b777e8467""}
2025-08-01 02:20:16.815 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][59e81c7d-7e35-4bc7-ae6a-7f9578c4ad47] received connection request
2025-08-01 02:20:16.816 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 02:20:16.816 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:20:16.857 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][59e81c7d-7e35-4bc7-ae6a-7f9578c4ad47] socks forwarding established
2025-08-01 02:20:16.903 [info] [command][e2090bbd-efe2-488c-8af2-409b777e8467] Process exited with code 0
2025-08-01 02:20:16.904 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][59e81c7d-7e35-4bc7-ae6a-7f9578c4ad47] socks connection closed
2025-08-01 02:20:16.904 [info] [command][e2090bbd-efe2-488c-8af2-409b777e8467] Socket close event received
2025-08-01 02:20:16.944 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64842 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:21:16.905 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:21:16.907 [info] [command][84371b7c-3416-4352-b150-5258cc312769] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""84371b7c-3416-4352-b150-5258cc312769""}
2025-08-01 02:21:16.908 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][423fbf28-4a2c-49b3-851c-19a548b5554c] received connection request
2025-08-01 02:21:16.908 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:21:16.951 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][423fbf28-4a2c-49b3-851c-19a548b5554c] socks forwarding established
2025-08-01 02:21:17.081 [info] [command][84371b7c-3416-4352-b150-5258cc312769] Process exited with code 0
2025-08-01 02:21:17.081 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][423fbf28-4a2c-49b3-851c-19a548b5554c] socks connection closed
2025-08-01 02:21:17.081 [info] [command][84371b7c-3416-4352-b150-5258cc312769] Socket close event received
2025-08-01 02:21:17.139 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64862 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:22:17.083 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:22:17.083 [info] [command][18b49bb2-728a-4674-a6bb-50de7a014d3b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""18b49bb2-728a-4674-a6bb-50de7a014d3b""}
2025-08-01 02:22:17.084 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3e3fed4d-0abe-4190-ac9a-3c6e3d377e76] received connection request
2025-08-01 02:22:17.084 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:22:17.127 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3e3fed4d-0abe-4190-ac9a-3c6e3d377e76] socks forwarding established
2025-08-01 02:22:17.172 [info] [command][18b49bb2-728a-4674-a6bb-50de7a014d3b] Process exited with code 0
2025-08-01 02:22:17.173 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3e3fed4d-0abe-4190-ac9a-3c6e3d377e76] socks connection closed
2025-08-01 02:22:17.173 [info] [command][18b49bb2-728a-4674-a6bb-50de7a014d3b] Socket close event received
2025-08-01 02:22:17.216 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64896 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:23:17.182 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:23:17.184 [info] [command][46c7d259-6097-40ef-aab9-30713eb85c5a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""46c7d259-6097-40ef-aab9-30713eb85c5a""}
2025-08-01 02:23:17.184 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][180907e1-58ad-4d30-a4bc-e4934cbb0866] received connection request
2025-08-01 02:23:17.185 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:23:17.231 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][180907e1-58ad-4d30-a4bc-e4934cbb0866] socks forwarding established
2025-08-01 02:23:17.276 [info] [command][46c7d259-6097-40ef-aab9-30713eb85c5a] Process exited with code 0
2025-08-01 02:23:17.277 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][180907e1-58ad-4d30-a4bc-e4934cbb0866] socks connection closed
2025-08-01 02:23:17.277 [info] [command][46c7d259-6097-40ef-aab9-30713eb85c5a] Socket close event received
2025-08-01 02:23:17.315 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64920 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:24:17.283 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:24:17.284 [info] [command][b4ba88f1-a86f-452c-aa2d-aea36cc81148] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b4ba88f1-a86f-452c-aa2d-aea36cc81148""}
2025-08-01 02:24:17.285 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a09f1275-54f5-49cd-8ce5-efbfec435554] received connection request
2025-08-01 02:24:17.285 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:24:17.333 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a09f1275-54f5-49cd-8ce5-efbfec435554] socks forwarding established
2025-08-01 02:24:17.378 [info] [command][b4ba88f1-a86f-452c-aa2d-aea36cc81148] Process exited with code 0
2025-08-01 02:24:17.379 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a09f1275-54f5-49cd-8ce5-efbfec435554] socks connection closed
2025-08-01 02:24:17.379 [info] [command][b4ba88f1-a86f-452c-aa2d-aea36cc81148] Socket close event received
2025-08-01 02:24:17.419 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64944 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:25:17.386 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:25:17.387 [info] [command][b5164b1c-3b39-4653-8886-8a5803ea8594] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b5164b1c-3b39-4653-8886-8a5803ea8594""}
2025-08-01 02:25:17.387 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][541215d1-80b9-47a5-b6dc-5a1e576ad7fc] received connection request
2025-08-01 02:25:17.387 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:25:17.428 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][541215d1-80b9-47a5-b6dc-5a1e576ad7fc] socks forwarding established
2025-08-01 02:25:17.472 [info] [command][b5164b1c-3b39-4653-8886-8a5803ea8594] Process exited with code 0
2025-08-01 02:25:17.473 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][541215d1-80b9-47a5-b6dc-5a1e576ad7fc] socks connection closed
2025-08-01 02:25:17.473 [info] [command][b5164b1c-3b39-4653-8886-8a5803ea8594] Socket close event received
2025-08-01 02:25:17.513 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64967 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:26:17.480 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:26:17.481 [info] [command][3463720d-5353-4fa0-8137-a2e1e09e54d4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3463720d-5353-4fa0-8137-a2e1e09e54d4""}
2025-08-01 02:26:17.482 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1a2639f3-a340-4169-9ba0-993f73683e70] received connection request
2025-08-01 02:26:17.483 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:26:17.601 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a2639f3-a340-4169-9ba0-993f73683e70] socks forwarding established
2025-08-01 02:26:17.645 [info] [command][3463720d-5353-4fa0-8137-a2e1e09e54d4] Process exited with code 0
2025-08-01 02:26:17.646 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a2639f3-a340-4169-9ba0-993f73683e70] socks connection closed
2025-08-01 02:26:17.646 [info] [command][3463720d-5353-4fa0-8137-a2e1e09e54d4] Socket close event received
2025-08-01 02:26:17.685 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 64984 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:27:17.651 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:27:17.652 [info] [command][39d44373-6048-4478-aa9f-d4a3f2f5802f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""39d44373-6048-4478-aa9f-d4a3f2f5802f""}
2025-08-01 02:27:17.653 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dc9a9ff3-a5e1-4835-920d-eceba9c99076] received connection request
2025-08-01 02:27:17.654 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:27:17.696 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dc9a9ff3-a5e1-4835-920d-eceba9c99076] socks forwarding established
2025-08-01 02:27:17.741 [info] [command][39d44373-6048-4478-aa9f-d4a3f2f5802f] Process exited with code 0
2025-08-01 02:27:17.741 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dc9a9ff3-a5e1-4835-920d-eceba9c99076] socks connection closed
2025-08-01 02:27:17.741 [info] [command][39d44373-6048-4478-aa9f-d4a3f2f5802f] Socket close event received
2025-08-01 02:27:17.781 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65024 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:28:17.749 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:28:17.750 [info] [command][5352951a-ffe0-4f81-abe2-581c1558a58e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5352951a-ffe0-4f81-abe2-581c1558a58e""}
2025-08-01 02:28:17.751 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ee96b451-120a-4d2d-a6a6-ce503717a856] received connection request
2025-08-01 02:28:17.752 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:28:17.856 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee96b451-120a-4d2d-a6a6-ce503717a856] socks forwarding established
2025-08-01 02:28:17.903 [info] [command][5352951a-ffe0-4f81-abe2-581c1558a58e] Process exited with code 0
2025-08-01 02:28:17.903 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee96b451-120a-4d2d-a6a6-ce503717a856] socks connection closed
2025-08-01 02:28:17.903 [info] [command][5352951a-ffe0-4f81-abe2-581c1558a58e] Socket close event received
2025-08-01 02:28:17.956 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65057 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:29:17.906 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:29:17.907 [info] [command][0225e71f-12db-4bc4-bc41-3a599413a64a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0225e71f-12db-4bc4-bc41-3a599413a64a""}
2025-08-01 02:29:17.908 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5d215a96-5518-4022-aa50-acb9bab23c21] received connection request
2025-08-01 02:29:17.908 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:29:17.950 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d215a96-5518-4022-aa50-acb9bab23c21] socks forwarding established
2025-08-01 02:29:17.993 [info] [command][0225e71f-12db-4bc4-bc41-3a599413a64a] Process exited with code 0
2025-08-01 02:29:17.993 [info] [command][0225e71f-12db-4bc4-bc41-3a599413a64a] Socket close event received
2025-08-01 02:29:17.993 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d215a96-5518-4022-aa50-acb9bab23c21] socks connection closed
2025-08-01 02:29:18.032 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65079 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:30:17.996 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:30:17.997 [info] [command][d636a75b-2f5e-4bee-b577-d2df52f9158e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d636a75b-2f5e-4bee-b577-d2df52f9158e""}
2025-08-01 02:30:17.998 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6426b33f-6598-4cc7-ae7f-5a39e2370041] received connection request
2025-08-01 02:30:17.999 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:30:18.114 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6426b33f-6598-4cc7-ae7f-5a39e2370041] socks forwarding established
2025-08-01 02:30:18.164 [info] [command][d636a75b-2f5e-4bee-b577-d2df52f9158e] Process exited with code 0
2025-08-01 02:30:18.164 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6426b33f-6598-4cc7-ae7f-5a39e2370041] socks connection closed
2025-08-01 02:30:18.165 [info] [command][d636a75b-2f5e-4bee-b577-d2df52f9158e] Socket close event received
2025-08-01 02:30:18.209 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65104 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:31:18.165 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:31:18.166 [info] [command][02e4dba5-4a66-4f00-8174-ee1ac928fb1a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""02e4dba5-4a66-4f00-8174-ee1ac928fb1a""}
2025-08-01 02:31:18.167 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5c9f02b3-a67f-4e3f-a113-4657dc0d404d] received connection request
2025-08-01 02:31:18.167 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:31:18.427 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c9f02b3-a67f-4e3f-a113-4657dc0d404d] socks forwarding established
2025-08-01 02:31:18.473 [info] [command][02e4dba5-4a66-4f00-8174-ee1ac928fb1a] Process exited with code 0
2025-08-01 02:31:18.473 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5c9f02b3-a67f-4e3f-a113-4657dc0d404d] socks connection closed
2025-08-01 02:31:18.473 [info] [command][02e4dba5-4a66-4f00-8174-ee1ac928fb1a] Socket close event received
2025-08-01 02:31:18.513 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65116 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:32:18.475 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:32:18.476 [info] [command][f5c61b7f-d366-46a5-97a4-a79913f1cdeb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f5c61b7f-d366-46a5-97a4-a79913f1cdeb""}
2025-08-01 02:32:18.477 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][33cf3977-5455-4c1f-9a9b-7ce526c83e8e] received connection request
2025-08-01 02:32:18.478 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:32:18.521 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33cf3977-5455-4c1f-9a9b-7ce526c83e8e] socks forwarding established
2025-08-01 02:32:18.566 [info] [command][f5c61b7f-d366-46a5-97a4-a79913f1cdeb] Process exited with code 0
2025-08-01 02:32:18.566 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33cf3977-5455-4c1f-9a9b-7ce526c83e8e] socks connection closed
2025-08-01 02:32:18.566 [info] [command][f5c61b7f-d366-46a5-97a4-a79913f1cdeb] Socket close event received
2025-08-01 02:32:18.605 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65143 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:33:18.569 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:33:18.572 [info] [command][093c0ed2-5c10-4b3f-bffb-57285b1a8da6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""093c0ed2-5c10-4b3f-bffb-57285b1a8da6""}
2025-08-01 02:33:18.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bae7fc3f-28a5-4502-bbfe-197340880282] received connection request
2025-08-01 02:33:18.573 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:33:18.618 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bae7fc3f-28a5-4502-bbfe-197340880282] socks forwarding established
2025-08-01 02:33:18.664 [info] [command][093c0ed2-5c10-4b3f-bffb-57285b1a8da6] Process exited with code 0
2025-08-01 02:33:18.665 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bae7fc3f-28a5-4502-bbfe-197340880282] socks connection closed
2025-08-01 02:33:18.665 [info] [command][093c0ed2-5c10-4b3f-bffb-57285b1a8da6] Socket close event received
2025-08-01 02:33:18.706 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65180 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:34:18.672 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:34:18.673 [info] [command][6035858f-c291-452e-8523-7c408e5e2864] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6035858f-c291-452e-8523-7c408e5e2864""}
2025-08-01 02:34:18.673 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9a0e295c-5505-4a06-9797-48141eb2b5af] received connection request
2025-08-01 02:34:18.674 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:34:18.717 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9a0e295c-5505-4a06-9797-48141eb2b5af] socks forwarding established
2025-08-01 02:34:18.762 [info] [command][6035858f-c291-452e-8523-7c408e5e2864] Process exited with code 0
2025-08-01 02:34:18.762 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9a0e295c-5505-4a06-9797-48141eb2b5af] socks connection closed
2025-08-01 02:34:18.762 [info] [command][6035858f-c291-452e-8523-7c408e5e2864] Socket close event received
2025-08-01 02:34:18.803 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65197 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:35:18.768 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:35:18.770 [info] [command][5bc77a63-d002-4617-a309-59db1d4f02ae] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5bc77a63-d002-4617-a309-59db1d4f02ae""}
2025-08-01 02:35:18.770 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][28d2c485-e728-48b5-bff6-340e99c93a32] received connection request
2025-08-01 02:35:18.771 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:35:18.887 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][28d2c485-e728-48b5-bff6-340e99c93a32] socks forwarding established
2025-08-01 02:35:18.931 [info] [command][5bc77a63-d002-4617-a309-59db1d4f02ae] Process exited with code 0
2025-08-01 02:35:18.931 [info] [command][5bc77a63-d002-4617-a309-59db1d4f02ae] Socket close event received
2025-08-01 02:35:18.932 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][28d2c485-e728-48b5-bff6-340e99c93a32] socks connection closed
2025-08-01 02:35:18.972 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65213 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:36:18.933 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:36:18.934 [info] [command][2c850cec-9eb7-4b5f-86c1-48b7632b1dc2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2c850cec-9eb7-4b5f-86c1-48b7632b1dc2""}
2025-08-01 02:36:18.935 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][46b3fd2f-d6ce-4db0-97bd-97f9fb764bc1] received connection request
2025-08-01 02:36:18.935 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:36:18.979 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][46b3fd2f-d6ce-4db0-97bd-97f9fb764bc1] socks forwarding established
2025-08-01 02:36:19.025 [info] [command][2c850cec-9eb7-4b5f-86c1-48b7632b1dc2] Process exited with code 0
2025-08-01 02:36:19.025 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][46b3fd2f-d6ce-4db0-97bd-97f9fb764bc1] socks connection closed
2025-08-01 02:36:19.025 [info] [command][2c850cec-9eb7-4b5f-86c1-48b7632b1dc2] Socket close event received
2025-08-01 02:36:19.067 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65237 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:37:19.028 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:37:19.031 [info] [command][3163cd46-3f12-4bfa-8b23-3ca5ed05930f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3163cd46-3f12-4bfa-8b23-3ca5ed05930f""}
2025-08-01 02:37:19.031 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cb742c49-1aee-44a0-97b2-4115456d8818] received connection request
2025-08-01 02:37:19.032 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:37:19.073 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb742c49-1aee-44a0-97b2-4115456d8818] socks forwarding established
2025-08-01 02:37:19.117 [info] [command][3163cd46-3f12-4bfa-8b23-3ca5ed05930f] Process exited with code 0
2025-08-01 02:37:19.117 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb742c49-1aee-44a0-97b2-4115456d8818] socks connection closed
2025-08-01 02:37:19.117 [info] [command][3163cd46-3f12-4bfa-8b23-3ca5ed05930f] Socket close event received
2025-08-01 02:37:19.157 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65272 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:38:19.118 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:38:19.120 [info] [command][2a1e5469-e3b9-4eee-9870-16568cf50ceb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2a1e5469-e3b9-4eee-9870-16568cf50ceb""}
2025-08-01 02:38:19.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b9eae8b4-5fc0-4039-bfb2-90d4d4730fae] received connection request
2025-08-01 02:38:19.121 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:38:19.168 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9eae8b4-5fc0-4039-bfb2-90d4d4730fae] socks forwarding established
2025-08-01 02:38:19.211 [info] [command][2a1e5469-e3b9-4eee-9870-16568cf50ceb] Process exited with code 0
2025-08-01 02:38:19.211 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9eae8b4-5fc0-4039-bfb2-90d4d4730fae] socks connection closed
2025-08-01 02:38:19.212 [info] [command][2a1e5469-e3b9-4eee-9870-16568cf50ceb] Socket close event received
2025-08-01 02:38:19.251 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65304 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:39:19.214 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:39:19.215 [info] [command][816a5c83-ed45-432c-bab4-f2a114e9a6c3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""816a5c83-ed45-432c-bab4-f2a114e9a6c3""}
2025-08-01 02:39:19.216 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][48c13ae3-5602-43b0-8ab1-94fa72318aec] received connection request
2025-08-01 02:39:19.216 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:39:19.265 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48c13ae3-5602-43b0-8ab1-94fa72318aec] socks forwarding established
2025-08-01 02:39:19.310 [info] [command][816a5c83-ed45-432c-bab4-f2a114e9a6c3] Process exited with code 0
2025-08-01 02:39:19.310 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48c13ae3-5602-43b0-8ab1-94fa72318aec] socks connection closed
2025-08-01 02:39:19.310 [info] [command][816a5c83-ed45-432c-bab4-f2a114e9a6c3] Socket close event received
2025-08-01 02:39:19.350 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65325 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:40:19.321 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:40:19.323 [info] [command][659d7dcc-917b-4757-97fb-0b3cc37c0cd5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""659d7dcc-917b-4757-97fb-0b3cc37c0cd5""}
2025-08-01 02:40:19.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6ad15946-2242-4436-a2dc-232ce0fe63b5] received connection request
2025-08-01 02:40:19.324 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:40:19.371 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6ad15946-2242-4436-a2dc-232ce0fe63b5] socks forwarding established
2025-08-01 02:40:19.563 [info] [command][659d7dcc-917b-4757-97fb-0b3cc37c0cd5] Process exited with code 0
2025-08-01 02:40:19.563 [info] [command][659d7dcc-917b-4757-97fb-0b3cc37c0cd5] Socket close event received
2025-08-01 02:40:19.564 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6ad15946-2242-4436-a2dc-232ce0fe63b5] socks connection closed
2025-08-01 02:40:19.606 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65342 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:41:19.572 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:41:19.573 [info] [command][deac2993-3293-4242-b634-55c29432762d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""deac2993-3293-4242-b634-55c29432762d""}
2025-08-01 02:41:19.574 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6f810df9-1f5e-482c-90d5-525026bd0878] received connection request
2025-08-01 02:41:19.575 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:41:19.617 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f810df9-1f5e-482c-90d5-525026bd0878] socks forwarding established
2025-08-01 02:41:19.663 [info] [command][deac2993-3293-4242-b634-55c29432762d] Process exited with code 0
2025-08-01 02:41:19.664 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f810df9-1f5e-482c-90d5-525026bd0878] socks connection closed
2025-08-01 02:41:19.664 [info] [command][deac2993-3293-4242-b634-55c29432762d] Socket close event received
2025-08-01 02:41:19.703 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65360 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:42:19.668 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:42:19.670 [info] [command][0c42fb6c-bfb0-4c6a-b49e-4d680921ff8b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0c42fb6c-bfb0-4c6a-b49e-4d680921ff8b""}
2025-08-01 02:42:19.671 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4fc6af5a-2227-478f-9aa4-c16b527706e9] received connection request
2025-08-01 02:42:19.672 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:42:19.718 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4fc6af5a-2227-478f-9aa4-c16b527706e9] socks forwarding established
2025-08-01 02:42:19.765 [info] [command][0c42fb6c-bfb0-4c6a-b49e-4d680921ff8b] Process exited with code 0
2025-08-01 02:42:19.766 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4fc6af5a-2227-478f-9aa4-c16b527706e9] socks connection closed
2025-08-01 02:42:19.766 [info] [command][0c42fb6c-bfb0-4c6a-b49e-4d680921ff8b] Socket close event received
2025-08-01 02:42:19.807 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65390 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:43:19.770 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:43:19.772 [info] [command][a01fa096-24f0-4989-9671-6da7318eddbf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a01fa096-24f0-4989-9671-6da7318eddbf""}
2025-08-01 02:43:19.773 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][35da1c4d-60ce-4d06-825e-77a481925674] received connection request
2025-08-01 02:43:19.774 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:43:19.815 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35da1c4d-60ce-4d06-825e-77a481925674] socks forwarding established
2025-08-01 02:43:19.860 [info] [command][a01fa096-24f0-4989-9671-6da7318eddbf] Process exited with code 0
2025-08-01 02:43:19.861 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35da1c4d-60ce-4d06-825e-77a481925674] socks connection closed
2025-08-01 02:43:19.861 [info] [command][a01fa096-24f0-4989-9671-6da7318eddbf] Socket close event received
2025-08-01 02:43:19.950 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65414 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:44:19.865 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:44:19.867 [info] [command][b7bcf698-32d2-4bb9-a2dc-8e2680271659] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7bcf698-32d2-4bb9-a2dc-8e2680271659""}
2025-08-01 02:44:19.868 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][476224fe-c8c0-4e06-9acb-6af5baaee52d] received connection request
2025-08-01 02:44:19.868 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:44:19.992 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476224fe-c8c0-4e06-9acb-6af5baaee52d] socks forwarding established
2025-08-01 02:44:20.077 [info] [command][b7bcf698-32d2-4bb9-a2dc-8e2680271659] Process exited with code 0
2025-08-01 02:44:20.077 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][476224fe-c8c0-4e06-9acb-6af5baaee52d] socks connection closed
2025-08-01 02:44:20.077 [info] [command][b7bcf698-32d2-4bb9-a2dc-8e2680271659] Socket close event received
2025-08-01 02:44:20.224 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65433 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:45:20.087 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:45:20.088 [info] [command][856eed6c-3d5a-426a-a4ba-5fe7ad968671] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""856eed6c-3d5a-426a-a4ba-5fe7ad968671""}
2025-08-01 02:45:20.088 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][57a51e0f-021f-4cd5-a085-78cefce0d778] received connection request
2025-08-01 02:45:20.088 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:45:20.277 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][57a51e0f-021f-4cd5-a085-78cefce0d778] socks forwarding established
2025-08-01 02:45:20.322 [info] [command][856eed6c-3d5a-426a-a4ba-5fe7ad968671] Process exited with code 0
2025-08-01 02:45:20.322 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][57a51e0f-021f-4cd5-a085-78cefce0d778] socks connection closed
2025-08-01 02:45:20.322 [info] [command][856eed6c-3d5a-426a-a4ba-5fe7ad968671] Socket close event received
2025-08-01 02:45:20.383 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65447 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:46:20.322 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:46:20.324 [info] [command][84fa3082-f88e-4d7f-b06e-62c05c709caf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""84fa3082-f88e-4d7f-b06e-62c05c709caf""}
2025-08-01 02:46:20.325 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d5b4f695-2e05-4b56-96ea-f858e169bb50] received connection request
2025-08-01 02:46:20.326 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:46:20.370 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d5b4f695-2e05-4b56-96ea-f858e169bb50] socks forwarding established
2025-08-01 02:46:20.415 [info] [command][84fa3082-f88e-4d7f-b06e-62c05c709caf] Process exited with code 0
2025-08-01 02:46:20.415 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d5b4f695-2e05-4b56-96ea-f858e169bb50] socks connection closed
2025-08-01 02:46:20.415 [info] [command][84fa3082-f88e-4d7f-b06e-62c05c709caf] Socket close event received
2025-08-01 02:46:20.462 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65461 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:47:20.424 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:47:20.427 [info] [command][b33a4775-461f-4cd7-95f0-59baaf1a0893] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b33a4775-461f-4cd7-95f0-59baaf1a0893""}
2025-08-01 02:47:20.428 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1f1b1b3c-f3d2-4a61-8f8b-01bf93d7409d] received connection request
2025-08-01 02:47:20.428 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:47:20.469 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f1b1b3c-f3d2-4a61-8f8b-01bf93d7409d] socks forwarding established
2025-08-01 02:47:20.514 [info] [command][b33a4775-461f-4cd7-95f0-59baaf1a0893] Process exited with code 0
2025-08-01 02:47:20.514 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1f1b1b3c-f3d2-4a61-8f8b-01bf93d7409d] socks connection closed
2025-08-01 02:47:20.514 [info] [command][b33a4775-461f-4cd7-95f0-59baaf1a0893] Socket close event received
2025-08-01 02:47:20.556 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65487 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:48:20.515 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:48:20.518 [info] [command][9b5cb905-1183-4f55-931e-622848b080e9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9b5cb905-1183-4f55-931e-622848b080e9""}
2025-08-01 02:48:20.519 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1cc3c232-b464-4439-9785-e9054f2df085] received connection request
2025-08-01 02:48:20.519 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:48:20.563 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1cc3c232-b464-4439-9785-e9054f2df085] socks forwarding established
2025-08-01 02:48:20.608 [info] [command][9b5cb905-1183-4f55-931e-622848b080e9] Process exited with code 0
2025-08-01 02:48:20.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1cc3c232-b464-4439-9785-e9054f2df085] socks connection closed
2025-08-01 02:48:20.608 [info] [command][9b5cb905-1183-4f55-931e-622848b080e9] Socket close event received
2025-08-01 02:48:20.648 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65512 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:49:20.609 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:49:20.611 [info] [command][2a24a6aa-4369-4e78-ad18-9d87615414b8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2a24a6aa-4369-4e78-ad18-9d87615414b8""}
2025-08-01 02:49:20.612 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][005e3f21-43ae-4b95-a56d-c6350ebf8269] received connection request
2025-08-01 02:49:20.613 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:49:20.654 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][005e3f21-43ae-4b95-a56d-c6350ebf8269] socks forwarding established
2025-08-01 02:49:20.700 [info] [command][2a24a6aa-4369-4e78-ad18-9d87615414b8] Process exited with code 0
2025-08-01 02:49:20.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][005e3f21-43ae-4b95-a56d-c6350ebf8269] socks connection closed
2025-08-01 02:49:20.701 [info] [command][2a24a6aa-4369-4e78-ad18-9d87615414b8] Socket close event received
2025-08-01 02:49:20.743 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 65531 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:50:20.706 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:50:20.707 [info] [command][362af661-961b-411b-bc6c-610a6d3ceeff] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""362af661-961b-411b-bc6c-610a6d3ceeff""}
2025-08-01 02:50:20.708 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][984425d9-034f-4632-9903-314b8e64a562] received connection request
2025-08-01 02:50:20.709 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:50:20.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][984425d9-034f-4632-9903-314b8e64a562] socks forwarding established
2025-08-01 02:50:20.837 [info] [command][362af661-961b-411b-bc6c-610a6d3ceeff] Process exited with code 0
2025-08-01 02:50:20.838 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][984425d9-034f-4632-9903-314b8e64a562] socks connection closed
2025-08-01 02:50:20.838 [info] [command][362af661-961b-411b-bc6c-610a6d3ceeff] Socket close event received
2025-08-01 02:50:20.876 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49164 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:51:20.846 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:51:20.848 [info] [command][0a92287d-76e9-4df1-951b-d4a195c5b04b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0a92287d-76e9-4df1-951b-d4a195c5b04b""}
2025-08-01 02:51:20.849 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ac2b2aa8-b9e3-4d3a-a74e-c8242f5c1c19] received connection request
2025-08-01 02:51:20.850 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:51:20.894 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac2b2aa8-b9e3-4d3a-a74e-c8242f5c1c19] socks forwarding established
2025-08-01 02:51:20.937 [info] [command][0a92287d-76e9-4df1-951b-d4a195c5b04b] Process exited with code 0
2025-08-01 02:51:20.937 [info] [command][0a92287d-76e9-4df1-951b-d4a195c5b04b] Socket close event received
2025-08-01 02:51:20.955 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac2b2aa8-b9e3-4d3a-a74e-c8242f5c1c19] socks connection closed
2025-08-01 02:51:20.977 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49194 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:52:20.948 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:52:20.949 [info] [command][cbbb90c7-6090-4707-b5df-47e29ba4426a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cbbb90c7-6090-4707-b5df-47e29ba4426a""}
2025-08-01 02:52:20.950 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d37e9b9c-9f8d-4267-9fa4-be4dfade16d6] received connection request
2025-08-01 02:52:20.950 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:52:20.996 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d37e9b9c-9f8d-4267-9fa4-be4dfade16d6] socks forwarding established
2025-08-01 02:52:21.041 [info] [command][cbbb90c7-6090-4707-b5df-47e29ba4426a] Process exited with code 0
2025-08-01 02:52:21.041 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d37e9b9c-9f8d-4267-9fa4-be4dfade16d6] socks connection closed
2025-08-01 02:52:21.041 [info] [command][cbbb90c7-6090-4707-b5df-47e29ba4426a] Socket close event received
2025-08-01 02:52:21.079 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49229 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:53:21.046 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:53:21.048 [info] [command][308dbde5-097b-4568-b846-81b48ff793b3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""308dbde5-097b-4568-b846-81b48ff793b3""}
2025-08-01 02:53:21.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][09675245-8df1-4904-850e-e76cfdc2a7a4] received connection request
2025-08-01 02:53:21.048 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:53:21.091 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][09675245-8df1-4904-850e-e76cfdc2a7a4] socks forwarding established
2025-08-01 02:53:21.135 [info] [command][308dbde5-097b-4568-b846-81b48ff793b3] Process exited with code 0
2025-08-01 02:53:21.136 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][09675245-8df1-4904-850e-e76cfdc2a7a4] socks connection closed
2025-08-01 02:53:21.136 [info] [command][308dbde5-097b-4568-b846-81b48ff793b3] Socket close event received
2025-08-01 02:53:21.177 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49268 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:54:21.145 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:54:21.148 [info] [command][74d5ecc9-5a69-4b26-a5a6-2dd53a20e1a1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""74d5ecc9-5a69-4b26-a5a6-2dd53a20e1a1""}
2025-08-01 02:54:21.149 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5b80b6ba-3dad-413b-b0b9-2ae51e3c6197] received connection request
2025-08-01 02:54:21.149 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:54:21.191 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b80b6ba-3dad-413b-b0b9-2ae51e3c6197] socks forwarding established
2025-08-01 02:54:21.236 [info] [command][74d5ecc9-5a69-4b26-a5a6-2dd53a20e1a1] Process exited with code 0
2025-08-01 02:54:21.237 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5b80b6ba-3dad-413b-b0b9-2ae51e3c6197] socks connection closed
2025-08-01 02:54:21.237 [info] [command][74d5ecc9-5a69-4b26-a5a6-2dd53a20e1a1] Socket close event received
2025-08-01 02:54:21.297 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49288 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:55:21.239 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:55:21.240 [info] [command][825d6da1-c813-47e6-a3ba-e21b8f9600f8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""825d6da1-c813-47e6-a3ba-e21b8f9600f8""}
2025-08-01 02:55:21.240 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][252eb8d3-c326-4f14-a3c0-1bd5021b88f3] received connection request
2025-08-01 02:55:21.241 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:55:21.285 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][252eb8d3-c326-4f14-a3c0-1bd5021b88f3] socks forwarding established
2025-08-01 02:55:21.330 [info] [command][825d6da1-c813-47e6-a3ba-e21b8f9600f8] Process exited with code 0
2025-08-01 02:55:21.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][252eb8d3-c326-4f14-a3c0-1bd5021b88f3] socks connection closed
2025-08-01 02:55:21.330 [info] [command][825d6da1-c813-47e6-a3ba-e21b8f9600f8] Socket close event received
2025-08-01 02:55:21.370 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49319 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:56:21.338 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:56:21.340 [info] [command][9ccd37c4-91a0-4f92-bd08-d75ace1bcfa7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9ccd37c4-91a0-4f92-bd08-d75ace1bcfa7""}
2025-08-01 02:56:21.341 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b8d84e2d-d9e0-451c-ae65-d2fd7bda5d53] received connection request
2025-08-01 02:56:21.342 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:56:21.382 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b8d84e2d-d9e0-451c-ae65-d2fd7bda5d53] socks forwarding established
2025-08-01 02:56:21.427 [info] [command][9ccd37c4-91a0-4f92-bd08-d75ace1bcfa7] Process exited with code 0
2025-08-01 02:56:21.428 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b8d84e2d-d9e0-451c-ae65-d2fd7bda5d53] socks connection closed
2025-08-01 02:56:21.428 [info] [command][9ccd37c4-91a0-4f92-bd08-d75ace1bcfa7] Socket close event received
2025-08-01 02:56:21.467 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49350 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:57:21.437 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:57:21.438 [info] [command][ca7c1c52-4b46-4006-8299-4eb4e20a30a1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ca7c1c52-4b46-4006-8299-4eb4e20a30a1""}
2025-08-01 02:57:21.439 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eb408d6a-1285-40e0-8524-2c82859b0bec] received connection request
2025-08-01 02:57:21.440 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:57:21.484 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb408d6a-1285-40e0-8524-2c82859b0bec] socks forwarding established
2025-08-01 02:57:21.529 [info] [command][ca7c1c52-4b46-4006-8299-4eb4e20a30a1] Process exited with code 0
2025-08-01 02:57:21.530 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb408d6a-1285-40e0-8524-2c82859b0bec] socks connection closed
2025-08-01 02:57:21.530 [info] [command][ca7c1c52-4b46-4006-8299-4eb4e20a30a1] Socket close event received
2025-08-01 02:57:21.570 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49383 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:58:21.534 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:58:21.537 [info] [command][7ea75280-aa0f-4a2e-92ab-10af97dee742] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7ea75280-aa0f-4a2e-92ab-10af97dee742""}
2025-08-01 02:58:21.537 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][24eba784-9728-43fa-9620-4da32d89d90d] received connection request
2025-08-01 02:58:21.538 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:58:21.584 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][24eba784-9728-43fa-9620-4da32d89d90d] socks forwarding established
2025-08-01 02:58:21.629 [info] [command][7ea75280-aa0f-4a2e-92ab-10af97dee742] Process exited with code 0
2025-08-01 02:58:21.629 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][24eba784-9728-43fa-9620-4da32d89d90d] socks connection closed
2025-08-01 02:58:21.629 [info] [command][7ea75280-aa0f-4a2e-92ab-10af97dee742] Socket close event received
2025-08-01 02:58:21.672 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49413 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 02:59:21.630 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 02:59:21.633 [info] [command][f68a0513-6da3-4c2e-a4d2-4c1961489fd3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f68a0513-6da3-4c2e-a4d2-4c1961489fd3""}
2025-08-01 02:59:21.634 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d18f37d6-1fdc-4aab-ba5f-0c568656f5c1] received connection request
2025-08-01 02:59:21.634 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 02:59:21.675 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d18f37d6-1fdc-4aab-ba5f-0c568656f5c1] socks forwarding established
2025-08-01 02:59:21.723 [info] [command][f68a0513-6da3-4c2e-a4d2-4c1961489fd3] Process exited with code 0
2025-08-01 02:59:21.723 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d18f37d6-1fdc-4aab-ba5f-0c568656f5c1] socks connection closed
2025-08-01 02:59:21.723 [info] [command][f68a0513-6da3-4c2e-a4d2-4c1961489fd3] Socket close event received
2025-08-01 02:59:21.846 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49432 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:00:21.730 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:00:21.732 [info] [command][60fa3c9c-6c69-412f-8756-6a291e7b91a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""60fa3c9c-6c69-412f-8756-6a291e7b91a2""}
2025-08-01 03:00:21.733 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9097ecd9-dc2b-4a9b-8a1e-1f5f4a48efd8] received connection request
2025-08-01 03:00:21.733 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:00:21.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9097ecd9-dc2b-4a9b-8a1e-1f5f4a48efd8] socks forwarding established
2025-08-01 03:00:21.827 [info] [command][60fa3c9c-6c69-412f-8756-6a291e7b91a2] Process exited with code 0
2025-08-01 03:00:21.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9097ecd9-dc2b-4a9b-8a1e-1f5f4a48efd8] socks connection closed
2025-08-01 03:00:21.827 [info] [command][60fa3c9c-6c69-412f-8756-6a291e7b91a2] Socket close event received
2025-08-01 03:00:21.867 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49466 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:01:21.828 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:01:21.830 [info] [command][8d6b52c2-a121-4ba4-8c03-39f9a16eef66] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8d6b52c2-a121-4ba4-8c03-39f9a16eef66""}
2025-08-01 03:01:21.831 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][47a1955c-bfdf-40ae-a7bf-345ce5018bf3] received connection request
2025-08-01 03:01:21.832 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:01:21.873 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47a1955c-bfdf-40ae-a7bf-345ce5018bf3] socks forwarding established
2025-08-01 03:01:21.923 [info] [command][8d6b52c2-a121-4ba4-8c03-39f9a16eef66] Process exited with code 0
2025-08-01 03:01:21.923 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47a1955c-bfdf-40ae-a7bf-345ce5018bf3] socks connection closed
2025-08-01 03:01:21.923 [info] [command][8d6b52c2-a121-4ba4-8c03-39f9a16eef66] Socket close event received
2025-08-01 03:01:21.965 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49503 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:02:21.929 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:02:21.931 [info] [command][ae471622-6405-4976-98fb-3b3e8b731c29] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ae471622-6405-4976-98fb-3b3e8b731c29""}
2025-08-01 03:02:21.931 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c88e3900-70f3-4096-b0ec-f06440a21854] received connection request
2025-08-01 03:02:21.931 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:02:21.973 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c88e3900-70f3-4096-b0ec-f06440a21854] socks forwarding established
2025-08-01 03:02:22.018 [info] [command][ae471622-6405-4976-98fb-3b3e8b731c29] Process exited with code 0
2025-08-01 03:02:22.018 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c88e3900-70f3-4096-b0ec-f06440a21854] socks connection closed
2025-08-01 03:02:22.019 [info] [command][ae471622-6405-4976-98fb-3b3e8b731c29] Socket close event received
2025-08-01 03:02:22.060 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49534 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:03:22.020 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:03:22.021 [info] [command][0a388c1b-88e2-48e0-8296-2214d2a6842d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0a388c1b-88e2-48e0-8296-2214d2a6842d""}
2025-08-01 03:03:22.022 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b9720fd7-ac8d-465b-a47b-d977c6c60cf4] received connection request
2025-08-01 03:03:22.022 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:03:22.070 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9720fd7-ac8d-465b-a47b-d977c6c60cf4] socks forwarding established
2025-08-01 03:03:22.115 [info] [command][0a388c1b-88e2-48e0-8296-2214d2a6842d] Process exited with code 0
2025-08-01 03:03:22.116 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9720fd7-ac8d-465b-a47b-d977c6c60cf4] socks connection closed
2025-08-01 03:03:22.116 [info] [command][0a388c1b-88e2-48e0-8296-2214d2a6842d] Socket close event received
2025-08-01 03:03:22.232 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49578 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:04:22.116 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:04:22.119 [info] [command][9a815691-bc39-4e89-bcca-fc62d3533dc5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9a815691-bc39-4e89-bcca-fc62d3533dc5""}
2025-08-01 03:04:22.120 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1489c907-759d-4fd2-9a16-dbe137c51c67] received connection request
2025-08-01 03:04:22.120 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:04:22.169 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1489c907-759d-4fd2-9a16-dbe137c51c67] socks forwarding established
2025-08-01 03:04:22.215 [info] [command][9a815691-bc39-4e89-bcca-fc62d3533dc5] Process exited with code 0
2025-08-01 03:04:22.215 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1489c907-759d-4fd2-9a16-dbe137c51c67] socks connection closed
2025-08-01 03:04:22.215 [info] [command][9a815691-bc39-4e89-bcca-fc62d3533dc5] Socket close event received
2025-08-01 03:04:22.258 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49596 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:05:22.223 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:05:22.225 [info] [command][a61f9657-fca8-4141-a7be-e07433d6d342] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a61f9657-fca8-4141-a7be-e07433d6d342""}
2025-08-01 03:05:22.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d2e733fc-12c1-41f5-a1fa-9e250d1b4f0d] received connection request
2025-08-01 03:05:22.225 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:05:22.273 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d2e733fc-12c1-41f5-a1fa-9e250d1b4f0d] socks forwarding established
2025-08-01 03:05:22.321 [info] [command][a61f9657-fca8-4141-a7be-e07433d6d342] Process exited with code 0
2025-08-01 03:05:22.321 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d2e733fc-12c1-41f5-a1fa-9e250d1b4f0d] socks connection closed
2025-08-01 03:05:22.321 [info] [command][a61f9657-fca8-4141-a7be-e07433d6d342] Socket close event received
2025-08-01 03:05:22.361 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49614 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:06:22.332 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:06:22.334 [info] [command][c80d9ebc-287c-4f79-afda-47d7949a3d2c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c80d9ebc-287c-4f79-afda-47d7949a3d2c""}
2025-08-01 03:06:22.334 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b25fe46f-1972-4ec0-9d50-7a709317e340] received connection request
2025-08-01 03:06:22.335 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:06:22.377 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b25fe46f-1972-4ec0-9d50-7a709317e340] socks forwarding established
2025-08-01 03:06:22.420 [info] [command][c80d9ebc-287c-4f79-afda-47d7949a3d2c] Process exited with code 0
2025-08-01 03:06:22.420 [info] [command][c80d9ebc-287c-4f79-afda-47d7949a3d2c] Socket close event received
2025-08-01 03:06:22.421 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b25fe46f-1972-4ec0-9d50-7a709317e340] socks connection closed
2025-08-01 03:06:22.462 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49630 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:07:22.423 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:07:22.425 [info] [command][e46bd8d3-610e-4bbc-849a-95be6dc02576] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e46bd8d3-610e-4bbc-849a-95be6dc02576""}
2025-08-01 03:07:22.426 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a9f049dc-06f9-4699-b273-77e817fea759] received connection request
2025-08-01 03:07:22.426 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:07:22.471 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a9f049dc-06f9-4699-b273-77e817fea759] socks forwarding established
2025-08-01 03:07:22.515 [info] [command][e46bd8d3-610e-4bbc-849a-95be6dc02576] Process exited with code 0
2025-08-01 03:07:22.516 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a9f049dc-06f9-4699-b273-77e817fea759] socks connection closed
2025-08-01 03:07:22.516 [info] [command][e46bd8d3-610e-4bbc-849a-95be6dc02576] Socket close event received
2025-08-01 03:07:22.556 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49662 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:08:22.526 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:08:22.528 [info] [command][fdde8128-2803-4ffb-98fb-21619ae24841] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fdde8128-2803-4ffb-98fb-21619ae24841""}
2025-08-01 03:08:22.529 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][78aedbcb-2965-4576-b989-2921fdde312b] received connection request
2025-08-01 03:08:22.530 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:08:22.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78aedbcb-2965-4576-b989-2921fdde312b] socks forwarding established
2025-08-01 03:08:22.625 [info] [command][fdde8128-2803-4ffb-98fb-21619ae24841] Process exited with code 0
2025-08-01 03:08:22.625 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78aedbcb-2965-4576-b989-2921fdde312b] socks connection closed
2025-08-01 03:08:22.625 [info] [command][fdde8128-2803-4ffb-98fb-21619ae24841] Socket close event received
2025-08-01 03:08:22.667 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49692 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:09:22.636 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:09:22.638 [info] [command][616c1323-d3f9-43d0-83a4-78f36f957d0e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""616c1323-d3f9-43d0-83a4-78f36f957d0e""}
2025-08-01 03:09:22.639 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][100a4e30-d6ea-4eb1-a8d3-653e33bdd932] received connection request
2025-08-01 03:09:22.640 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:09:22.684 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][100a4e30-d6ea-4eb1-a8d3-653e33bdd932] socks forwarding established
2025-08-01 03:09:22.729 [info] [command][616c1323-d3f9-43d0-83a4-78f36f957d0e] Process exited with code 0
2025-08-01 03:09:22.729 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][100a4e30-d6ea-4eb1-a8d3-653e33bdd932] socks connection closed
2025-08-01 03:09:22.729 [info] [command][616c1323-d3f9-43d0-83a4-78f36f957d0e] Socket close event received
2025-08-01 03:09:22.768 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49721 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:10:22.739 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:10:22.741 [info] [command][9ba315b4-d65d-491c-b48c-646d33b01523] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9ba315b4-d65d-491c-b48c-646d33b01523""}
2025-08-01 03:10:22.741 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4e6af44f-739a-4288-ac51-e16ce1fc9122] received connection request
2025-08-01 03:10:22.741 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:10:22.789 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4e6af44f-739a-4288-ac51-e16ce1fc9122] socks forwarding established
2025-08-01 03:10:22.833 [info] [command][9ba315b4-d65d-491c-b48c-646d33b01523] Process exited with code 0
2025-08-01 03:10:22.834 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4e6af44f-739a-4288-ac51-e16ce1fc9122] socks connection closed
2025-08-01 03:10:22.834 [info] [command][9ba315b4-d65d-491c-b48c-646d33b01523] Socket close event received
2025-08-01 03:10:22.874 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49743 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:11:22.835 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:11:22.835 [info] [command][5fbe716b-046c-4902-86a4-dd611f9e7601] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5fbe716b-046c-4902-86a4-dd611f9e7601""}
2025-08-01 03:11:22.836 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][26a1d70e-67ab-40c7-844d-9aec332d609c] received connection request
2025-08-01 03:11:22.836 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:11:22.876 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][26a1d70e-67ab-40c7-844d-9aec332d609c] socks forwarding established
2025-08-01 03:11:22.920 [info] [command][5fbe716b-046c-4902-86a4-dd611f9e7601] Process exited with code 0
2025-08-01 03:11:22.920 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][26a1d70e-67ab-40c7-844d-9aec332d609c] socks connection closed
2025-08-01 03:11:22.920 [info] [command][5fbe716b-046c-4902-86a4-dd611f9e7601] Socket close event received
2025-08-01 03:11:22.963 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49767 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:12:22.922 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:12:22.924 [info] [command][6293cc84-7e5d-417a-bb6a-a6d50e42a0ca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6293cc84-7e5d-417a-bb6a-a6d50e42a0ca""}
2025-08-01 03:12:22.925 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][29cf4f1a-8cfe-44cc-9c3e-22f85e46d64a] received connection request
2025-08-01 03:12:22.925 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:12:22.965 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29cf4f1a-8cfe-44cc-9c3e-22f85e46d64a] socks forwarding established
2025-08-01 03:12:23.012 [info] [command][6293cc84-7e5d-417a-bb6a-a6d50e42a0ca] Process exited with code 0
2025-08-01 03:12:23.012 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29cf4f1a-8cfe-44cc-9c3e-22f85e46d64a] socks connection closed
2025-08-01 03:12:23.012 [info] [command][6293cc84-7e5d-417a-bb6a-a6d50e42a0ca] Socket close event received
2025-08-01 03:12:23.052 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49819 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:13:23.019 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:13:23.021 [info] [command][36c65910-c5ee-4a50-a9d8-ca9ad6e2c56c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""36c65910-c5ee-4a50-a9d8-ca9ad6e2c56c""}
2025-08-01 03:13:23.022 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f8f9453d-fae5-44a4-9a90-3623b978b486] received connection request
2025-08-01 03:13:23.022 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:13:23.102 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f8f9453d-fae5-44a4-9a90-3623b978b486] socks forwarding established
2025-08-01 03:13:23.149 [info] [command][36c65910-c5ee-4a50-a9d8-ca9ad6e2c56c] Process exited with code 0
2025-08-01 03:13:23.149 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f8f9453d-fae5-44a4-9a90-3623b978b486] socks connection closed
2025-08-01 03:13:23.149 [info] [command][36c65910-c5ee-4a50-a9d8-ca9ad6e2c56c] Socket close event received
2025-08-01 03:13:23.193 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49847 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:14:23.156 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:14:23.158 [info] [command][1a4a15cf-295c-4ba0-b176-c062611959b2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1a4a15cf-295c-4ba0-b176-c062611959b2""}
2025-08-01 03:14:23.159 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d9257d09-8a54-4bcb-908c-163f14e9b6de] received connection request
2025-08-01 03:14:23.159 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:14:23.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d9257d09-8a54-4bcb-908c-163f14e9b6de] socks forwarding established
2025-08-01 03:14:23.261 [info] [command][1a4a15cf-295c-4ba0-b176-c062611959b2] Process exited with code 0
2025-08-01 03:14:23.261 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d9257d09-8a54-4bcb-908c-163f14e9b6de] socks connection closed
2025-08-01 03:14:23.261 [info] [command][1a4a15cf-295c-4ba0-b176-c062611959b2] Socket close event received
2025-08-01 03:14:23.361 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49888 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:15:23.270 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:15:23.273 [info] [command][7eeb981c-e97f-4f9c-909d-8490a00ab058] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7eeb981c-e97f-4f9c-909d-8490a00ab058""}
2025-08-01 03:15:23.274 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][20ec42ba-f43e-42f8-8ccb-503a6cc99905] received connection request
2025-08-01 03:15:23.274 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:15:23.321 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][20ec42ba-f43e-42f8-8ccb-503a6cc99905] socks forwarding established
2025-08-01 03:15:23.367 [info] [command][7eeb981c-e97f-4f9c-909d-8490a00ab058] Process exited with code 0
2025-08-01 03:15:23.367 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][20ec42ba-f43e-42f8-8ccb-503a6cc99905] socks connection closed
2025-08-01 03:15:23.368 [info] [command][7eeb981c-e97f-4f9c-909d-8490a00ab058] Socket close event received
2025-08-01 03:15:23.407 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49914 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:16:23.376 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:16:23.378 [info] [command][12453968-c827-43b3-a599-e98305fcafb7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""12453968-c827-43b3-a599-e98305fcafb7""}
2025-08-01 03:16:23.378 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][da428155-1c51-4995-b254-547c973f89ec] received connection request
2025-08-01 03:16:23.379 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:16:23.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][da428155-1c51-4995-b254-547c973f89ec] socks forwarding established
2025-08-01 03:16:23.467 [info] [command][12453968-c827-43b3-a599-e98305fcafb7] Process exited with code 0
2025-08-01 03:16:23.467 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][da428155-1c51-4995-b254-547c973f89ec] socks connection closed
2025-08-01 03:16:23.467 [info] [command][12453968-c827-43b3-a599-e98305fcafb7] Socket close event received
2025-08-01 03:16:23.507 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49928 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:17:23.475 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:17:23.477 [info] [command][ea4b8810-c02e-45d5-ba3f-0446f5c79a30] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ea4b8810-c02e-45d5-ba3f-0446f5c79a30""}
2025-08-01 03:17:23.478 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][349e899f-be70-42d7-aa35-36c7dc8553c2] received connection request
2025-08-01 03:17:23.478 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:17:23.523 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][349e899f-be70-42d7-aa35-36c7dc8553c2] socks forwarding established
2025-08-01 03:17:23.568 [info] [command][ea4b8810-c02e-45d5-ba3f-0446f5c79a30] Process exited with code 0
2025-08-01 03:17:23.568 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][349e899f-be70-42d7-aa35-36c7dc8553c2] socks connection closed
2025-08-01 03:17:23.569 [info] [command][ea4b8810-c02e-45d5-ba3f-0446f5c79a30] Socket close event received
2025-08-01 03:17:23.609 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49957 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:18:23.570 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:18:23.571 [info] [command][be52b627-7d3f-4a97-9d5c-28796071af63] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""be52b627-7d3f-4a97-9d5c-28796071af63""}
2025-08-01 03:18:23.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][67e12392-8554-4e75-9f60-5ad7fb83c5e9] received connection request
2025-08-01 03:18:23.573 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:18:23.614 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67e12392-8554-4e75-9f60-5ad7fb83c5e9] socks forwarding established
2025-08-01 03:18:23.661 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67e12392-8554-4e75-9f60-5ad7fb83c5e9] socks connection closed
2025-08-01 03:18:23.662 [info] [command][be52b627-7d3f-4a97-9d5c-28796071af63] Process exited with code 0
2025-08-01 03:18:23.662 [info] [command][be52b627-7d3f-4a97-9d5c-28796071af63] Socket close event received
2025-08-01 03:18:23.701 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 49988 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:19:23.667 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:19:23.669 [info] [command][7260f59c-1ba0-4ee0-983f-15c1bb1a087d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7260f59c-1ba0-4ee0-983f-15c1bb1a087d""}
2025-08-01 03:19:23.670 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3cea5c5e-83e3-4f54-909e-1567e80d050a] received connection request
2025-08-01 03:19:23.670 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:19:23.717 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3cea5c5e-83e3-4f54-909e-1567e80d050a] socks forwarding established
2025-08-01 03:19:23.763 [info] [command][7260f59c-1ba0-4ee0-983f-15c1bb1a087d] Process exited with code 0
2025-08-01 03:19:23.763 [info] [command][7260f59c-1ba0-4ee0-983f-15c1bb1a087d] Socket close event received
2025-08-01 03:19:23.763 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3cea5c5e-83e3-4f54-909e-1567e80d050a] socks connection closed
2025-08-01 03:19:23.805 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50013 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:20:23.763 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:20:23.765 [info] [command][fa530dc9-2e68-4ee9-b219-dcdfc4e15a03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fa530dc9-2e68-4ee9-b219-dcdfc4e15a03""}
2025-08-01 03:20:23.766 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d89b950e-cadb-4b20-8e70-b4b93acbe4f4] received connection request
2025-08-01 03:20:23.767 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:20:23.812 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d89b950e-cadb-4b20-8e70-b4b93acbe4f4] socks forwarding established
2025-08-01 03:20:23.862 [info] [command][fa530dc9-2e68-4ee9-b219-dcdfc4e15a03] Process exited with code 0
2025-08-01 03:20:23.862 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d89b950e-cadb-4b20-8e70-b4b93acbe4f4] socks connection closed
2025-08-01 03:20:23.863 [info] [command][fa530dc9-2e68-4ee9-b219-dcdfc4e15a03] Socket close event received
2025-08-01 03:20:23.903 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50033 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:21:23.870 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:21:23.872 [info] [command][914af498-3df9-48e0-b2cf-cf173542e12b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""914af498-3df9-48e0-b2cf-cf173542e12b""}
2025-08-01 03:21:23.872 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][225ab206-5ee6-464e-a411-41e62461835e] received connection request
2025-08-01 03:21:23.873 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:21:23.917 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][225ab206-5ee6-464e-a411-41e62461835e] socks forwarding established
2025-08-01 03:21:23.962 [info] [command][914af498-3df9-48e0-b2cf-cf173542e12b] Process exited with code 0
2025-08-01 03:21:23.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][225ab206-5ee6-464e-a411-41e62461835e] socks connection closed
2025-08-01 03:21:23.963 [info] [command][914af498-3df9-48e0-b2cf-cf173542e12b] Socket close event received
2025-08-01 03:21:24.005 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50048 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:22:23.963 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:22:23.964 [info] [command][d69dfca7-ac5b-4532-a287-f6dd496a8c9d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d69dfca7-ac5b-4532-a287-f6dd496a8c9d""}
2025-08-01 03:22:23.965 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][df59aa07-5870-4d1b-aaab-413486a9b427] received connection request
2025-08-01 03:22:23.965 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:22:24.008 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][df59aa07-5870-4d1b-aaab-413486a9b427] socks forwarding established
2025-08-01 03:22:24.054 [info] [command][d69dfca7-ac5b-4532-a287-f6dd496a8c9d] Process exited with code 0
2025-08-01 03:22:24.054 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][df59aa07-5870-4d1b-aaab-413486a9b427] socks connection closed
2025-08-01 03:22:24.055 [info] [command][d69dfca7-ac5b-4532-a287-f6dd496a8c9d] Socket close event received
2025-08-01 03:22:24.095 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50076 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:23:24.059 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:23:24.060 [info] [command][faa7646e-2445-4132-801c-4ae26a0c36ca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""faa7646e-2445-4132-801c-4ae26a0c36ca""}
2025-08-01 03:23:24.061 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][439561bf-dfd7-4121-b09f-268dbb9441b3] received connection request
2025-08-01 03:23:24.062 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:23:24.107 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][439561bf-dfd7-4121-b09f-268dbb9441b3] socks forwarding established
2025-08-01 03:23:24.153 [info] [command][faa7646e-2445-4132-801c-4ae26a0c36ca] Process exited with code 0
2025-08-01 03:23:24.153 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][439561bf-dfd7-4121-b09f-268dbb9441b3] socks connection closed
2025-08-01 03:23:24.153 [info] [command][faa7646e-2445-4132-801c-4ae26a0c36ca] Socket close event received
2025-08-01 03:23:24.192 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50105 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:24:24.154 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:24:24.155 [info] [command][8b1f5831-b2c5-4d1e-b1a1-73300a9a226d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8b1f5831-b2c5-4d1e-b1a1-73300a9a226d""}
2025-08-01 03:24:24.156 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3a7f3f56-4509-4323-86a8-aaa07b961a09] received connection request
2025-08-01 03:24:24.157 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:24:24.199 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3a7f3f56-4509-4323-86a8-aaa07b961a09] socks forwarding established
2025-08-01 03:24:24.245 [info] [command][8b1f5831-b2c5-4d1e-b1a1-73300a9a226d] Process exited with code 0
2025-08-01 03:24:24.245 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3a7f3f56-4509-4323-86a8-aaa07b961a09] socks connection closed
2025-08-01 03:24:24.245 [info] [command][8b1f5831-b2c5-4d1e-b1a1-73300a9a226d] Socket close event received
2025-08-01 03:24:24.286 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50120 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:25:24.252 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:25:24.253 [info] [command][ef0d2e61-25cc-4dbb-9e4b-7afc789b0409] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ef0d2e61-25cc-4dbb-9e4b-7afc789b0409""}
2025-08-01 03:25:24.254 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][79c6d4dd-b75a-41f7-ba7d-fb1dae00c3d3] received connection request
2025-08-01 03:25:24.254 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:25:24.300 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][79c6d4dd-b75a-41f7-ba7d-fb1dae00c3d3] socks forwarding established
2025-08-01 03:25:24.345 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][79c6d4dd-b75a-41f7-ba7d-fb1dae00c3d3] socks connection closed
2025-08-01 03:25:24.346 [info] [command][ef0d2e61-25cc-4dbb-9e4b-7afc789b0409] Process exited with code 0
2025-08-01 03:25:24.346 [info] [command][ef0d2e61-25cc-4dbb-9e4b-7afc789b0409] Socket close event received
2025-08-01 03:25:24.386 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50134 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:26:24.355 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:26:24.357 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2f77b4b1-6f2f-4e3f-b8ad-13cb7995d3c8] received connection request
2025-08-01 03:26:24.358 [info] [command][d02e5358-9e28-41de-8f62-61fc94499a92] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d02e5358-9e28-41de-8f62-61fc94499a92""}
2025-08-01 03:26:24.358 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:26:24.404 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2f77b4b1-6f2f-4e3f-b8ad-13cb7995d3c8] socks forwarding established
2025-08-01 03:26:24.449 [info] [command][d02e5358-9e28-41de-8f62-61fc94499a92] Process exited with code 0
2025-08-01 03:26:24.450 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2f77b4b1-6f2f-4e3f-b8ad-13cb7995d3c8] socks connection closed
2025-08-01 03:26:24.450 [info] [command][d02e5358-9e28-41de-8f62-61fc94499a92] Socket close event received
2025-08-01 03:26:24.488 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50146 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:27:24.460 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:27:24.461 [info] [command][e5d8ea40-7895-4dd6-a74f-17ee28a332e3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e5d8ea40-7895-4dd6-a74f-17ee28a332e3""}
2025-08-01 03:27:24.462 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e00cc08a-3032-489a-9106-75ff152944c5] received connection request
2025-08-01 03:27:24.463 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:27:24.524 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e00cc08a-3032-489a-9106-75ff152944c5] socks forwarding established
2025-08-01 03:27:24.738 [info] [command][e5d8ea40-7895-4dd6-a74f-17ee28a332e3] Process exited with code 0
2025-08-01 03:27:24.738 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e00cc08a-3032-489a-9106-75ff152944c5] socks connection closed
2025-08-01 03:27:24.738 [info] [command][e5d8ea40-7895-4dd6-a74f-17ee28a332e3] Socket close event received
2025-08-01 03:27:24.790 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50178 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:28:24.749 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:28:24.750 [info] [command][65dab6f8-6556-4d13-8034-155659337ec5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""65dab6f8-6556-4d13-8034-155659337ec5""}
2025-08-01 03:28:24.751 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4d957ab2-2e28-41b6-aa35-b19740244005] received connection request
2025-08-01 03:28:24.751 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:28:24.791 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d957ab2-2e28-41b6-aa35-b19740244005] socks forwarding established
2025-08-01 03:28:24.834 [info] [command][65dab6f8-6556-4d13-8034-155659337ec5] Process exited with code 0
2025-08-01 03:28:24.835 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4d957ab2-2e28-41b6-aa35-b19740244005] socks connection closed
2025-08-01 03:28:24.835 [info] [command][65dab6f8-6556-4d13-8034-155659337ec5] Socket close event received
2025-08-01 03:28:24.874 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50207 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:29:24.839 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:29:24.840 [info] [command][a94e1da8-5b02-4edc-868d-74335ea68494] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a94e1da8-5b02-4edc-868d-74335ea68494""}
2025-08-01 03:29:24.841 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4bff9f95-1be5-4d00-b1ed-a143e0983084] received connection request
2025-08-01 03:29:24.841 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:29:24.886 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bff9f95-1be5-4d00-b1ed-a143e0983084] socks forwarding established
2025-08-01 03:29:24.997 [info] [command][a94e1da8-5b02-4edc-868d-74335ea68494] Process exited with code 0
2025-08-01 03:29:24.997 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bff9f95-1be5-4d00-b1ed-a143e0983084] socks connection closed
2025-08-01 03:29:24.997 [info] [command][a94e1da8-5b02-4edc-868d-74335ea68494] Socket close event received
2025-08-01 03:29:25.037 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50230 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:30:25.007 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:30:25.008 [info] [command][3563c7ac-dce7-468b-8302-57d6a68b5436] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3563c7ac-dce7-468b-8302-57d6a68b5436""}
2025-08-01 03:30:25.009 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][aa13f3c2-10d5-405a-9a84-f80347ebb05f] received connection request
2025-08-01 03:30:25.010 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:30:25.051 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][aa13f3c2-10d5-405a-9a84-f80347ebb05f] socks forwarding established
2025-08-01 03:30:25.096 [info] [command][3563c7ac-dce7-468b-8302-57d6a68b5436] Process exited with code 0
2025-08-01 03:30:25.096 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][aa13f3c2-10d5-405a-9a84-f80347ebb05f] socks connection closed
2025-08-01 03:30:25.096 [info] [command][3563c7ac-dce7-468b-8302-57d6a68b5436] Socket close event received
2025-08-01 03:30:25.138 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50246 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:31:25.100 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:31:25.101 [info] [command][a295537c-734d-4cbc-87c8-bbada92be04a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a295537c-734d-4cbc-87c8-bbada92be04a""}
2025-08-01 03:31:25.102 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f7b8794b-869e-4799-96f8-0bcae77323fd] received connection request
2025-08-01 03:31:25.102 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:31:25.166 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f7b8794b-869e-4799-96f8-0bcae77323fd] socks forwarding established
2025-08-01 03:31:25.231 [info] [command][a295537c-734d-4cbc-87c8-bbada92be04a] Process exited with code 0
2025-08-01 03:31:25.231 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f7b8794b-869e-4799-96f8-0bcae77323fd] socks connection closed
2025-08-01 03:31:25.231 [info] [command][a295537c-734d-4cbc-87c8-bbada92be04a] Socket close event received
2025-08-01 03:31:25.275 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50261 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:32:25.237 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:32:25.238 [info] [command][25b1169f-f178-4e69-80cc-d0eabca35fe5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""25b1169f-f178-4e69-80cc-d0eabca35fe5""}
2025-08-01 03:32:25.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][89d1248e-f046-4298-aaea-a1331da1aa35] received connection request
2025-08-01 03:32:25.240 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:32:25.285 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89d1248e-f046-4298-aaea-a1331da1aa35] socks forwarding established
2025-08-01 03:32:25.331 [info] [command][25b1169f-f178-4e69-80cc-d0eabca35fe5] Process exited with code 0
2025-08-01 03:32:25.331 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][89d1248e-f046-4298-aaea-a1331da1aa35] socks connection closed
2025-08-01 03:32:25.331 [info] [command][25b1169f-f178-4e69-80cc-d0eabca35fe5] Socket close event received
2025-08-01 03:32:25.371 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50289 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:33:25.332 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:33:25.334 [info] [command][499b4d15-7498-44c2-961e-bdf8e08cb3c6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""499b4d15-7498-44c2-961e-bdf8e08cb3c6""}
2025-08-01 03:33:25.334 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][73f8913b-8496-49e2-9f92-d5667f1021da] received connection request
2025-08-01 03:33:25.335 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:33:25.387 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][73f8913b-8496-49e2-9f92-d5667f1021da] socks forwarding established
2025-08-01 03:33:25.432 [info] [command][499b4d15-7498-44c2-961e-bdf8e08cb3c6] Process exited with code 0
2025-08-01 03:33:25.432 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][73f8913b-8496-49e2-9f92-d5667f1021da] socks connection closed
2025-08-01 03:33:25.432 [info] [command][499b4d15-7498-44c2-961e-bdf8e08cb3c6] Socket close event received
2025-08-01 03:33:25.471 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50326 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:34:25.440 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:34:25.441 [info] [command][b919bc93-542b-4ef7-8f78-aac362f0bee4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b919bc93-542b-4ef7-8f78-aac362f0bee4""}
2025-08-01 03:34:25.441 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][771cc333-6c85-432e-8a63-419f01ee4b76] received connection request
2025-08-01 03:34:25.441 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:34:25.490 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][771cc333-6c85-432e-8a63-419f01ee4b76] socks forwarding established
2025-08-01 03:34:25.548 [info] [command][b919bc93-542b-4ef7-8f78-aac362f0bee4] Process exited with code 0
2025-08-01 03:34:25.548 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][771cc333-6c85-432e-8a63-419f01ee4b76] socks connection closed
2025-08-01 03:34:25.548 [info] [command][b919bc93-542b-4ef7-8f78-aac362f0bee4] Socket close event received
2025-08-01 03:34:25.589 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50348 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:35:25.558 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:35:25.559 [info] [command][a8a669bc-ee2e-4d75-ba34-b4e219afe5b5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a8a669bc-ee2e-4d75-ba34-b4e219afe5b5""}
2025-08-01 03:35:25.560 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ad7e4a83-9ea5-4376-bebf-6b202da47321] received connection request
2025-08-01 03:35:25.560 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:35:25.602 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad7e4a83-9ea5-4376-bebf-6b202da47321] socks forwarding established
2025-08-01 03:35:25.647 [info] [command][a8a669bc-ee2e-4d75-ba34-b4e219afe5b5] Process exited with code 0
2025-08-01 03:35:25.647 [info] [command][a8a669bc-ee2e-4d75-ba34-b4e219afe5b5] Socket close event received
2025-08-01 03:35:25.648 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad7e4a83-9ea5-4376-bebf-6b202da47321] socks connection closed
2025-08-01 03:35:25.687 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50369 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:36:25.651 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:36:25.652 [info] [command][4ea598fb-ac4c-4375-835f-e1924d44c856] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4ea598fb-ac4c-4375-835f-e1924d44c856""}
2025-08-01 03:36:25.653 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][14b5a2c2-74d9-4f25-b6bd-930a0e5053c5] received connection request
2025-08-01 03:36:25.653 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:36:25.903 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14b5a2c2-74d9-4f25-b6bd-930a0e5053c5] socks forwarding established
2025-08-01 03:36:25.948 [info] [command][4ea598fb-ac4c-4375-835f-e1924d44c856] Process exited with code 0
2025-08-01 03:36:25.948 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][14b5a2c2-74d9-4f25-b6bd-930a0e5053c5] socks connection closed
2025-08-01 03:36:25.948 [info] [command][4ea598fb-ac4c-4375-835f-e1924d44c856] Socket close event received
2025-08-01 03:36:25.989 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50383 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:37:25.955 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:37:25.957 [info] [command][d8f1c1f7-a38f-469f-ab87-335de430bb6e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d8f1c1f7-a38f-469f-ab87-335de430bb6e""}
2025-08-01 03:37:25.958 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ad96f5e6-bdfa-4d43-ae53-202c7ea880c9] received connection request
2025-08-01 03:37:25.958 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:37:26.002 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad96f5e6-bdfa-4d43-ae53-202c7ea880c9] socks forwarding established
2025-08-01 03:37:26.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad96f5e6-bdfa-4d43-ae53-202c7ea880c9] socks connection closed
2025-08-01 03:37:26.049 [info] [command][d8f1c1f7-a38f-469f-ab87-335de430bb6e] Process exited with code 0
2025-08-01 03:37:26.049 [info] [command][d8f1c1f7-a38f-469f-ab87-335de430bb6e] Socket close event received
2025-08-01 03:37:26.091 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50414 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:38:26.056 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:38:26.057 [info] [command][04d65a56-8e36-4661-8065-1a6c626a7079] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""04d65a56-8e36-4661-8065-1a6c626a7079""}
2025-08-01 03:38:26.057 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1bdf6cad-8ae2-4cd1-a867-ff7a23cea726] received connection request
2025-08-01 03:38:26.058 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:38:26.105 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1bdf6cad-8ae2-4cd1-a867-ff7a23cea726] socks forwarding established
2025-08-01 03:38:26.150 [info] [command][04d65a56-8e36-4661-8065-1a6c626a7079] Process exited with code 0
2025-08-01 03:38:26.150 [info] [command][04d65a56-8e36-4661-8065-1a6c626a7079] Socket close event received
2025-08-01 03:38:26.151 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1bdf6cad-8ae2-4cd1-a867-ff7a23cea726] socks connection closed
2025-08-01 03:38:26.190 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50445 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:39:26.153 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:39:26.155 [info] [command][d53dcbbb-d5dc-41c2-bf50-701d05089558] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d53dcbbb-d5dc-41c2-bf50-701d05089558""}
2025-08-01 03:39:26.156 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ff132034-57d1-47bc-a373-91271067009b] received connection request
2025-08-01 03:39:26.156 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:39:26.198 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff132034-57d1-47bc-a373-91271067009b] socks forwarding established
2025-08-01 03:39:26.242 [info] [command][d53dcbbb-d5dc-41c2-bf50-701d05089558] Process exited with code 0
2025-08-01 03:39:26.243 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ff132034-57d1-47bc-a373-91271067009b] socks connection closed
2025-08-01 03:39:26.243 [info] [command][d53dcbbb-d5dc-41c2-bf50-701d05089558] Socket close event received
2025-08-01 03:39:26.281 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50468 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:40:26.243 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:40:26.245 [info] [command][9c47dc67-635d-40d2-8d46-61987a66cf63] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9c47dc67-635d-40d2-8d46-61987a66cf63""}
2025-08-01 03:40:26.246 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1ee7c181-9b01-4cdd-bb3d-8e80673d5349] received connection request
2025-08-01 03:40:26.246 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:40:26.287 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1ee7c181-9b01-4cdd-bb3d-8e80673d5349] socks forwarding established
2025-08-01 03:40:26.331 [info] [command][9c47dc67-635d-40d2-8d46-61987a66cf63] Process exited with code 0
2025-08-01 03:40:26.332 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1ee7c181-9b01-4cdd-bb3d-8e80673d5349] socks connection closed
2025-08-01 03:40:26.332 [info] [command][9c47dc67-635d-40d2-8d46-61987a66cf63] Socket close event received
2025-08-01 03:40:26.372 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50490 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:41:26.341 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:41:26.344 [info] [command][cce9ff6f-53a6-4914-9fb3-46f00735501b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cce9ff6f-53a6-4914-9fb3-46f00735501b""}
2025-08-01 03:41:26.344 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ed628185-b82b-4292-b338-3661bbd4ee8d] received connection request
2025-08-01 03:41:26.345 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:41:26.588 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ed628185-b82b-4292-b338-3661bbd4ee8d] socks forwarding established
2025-08-01 03:41:26.632 [info] [command][cce9ff6f-53a6-4914-9fb3-46f00735501b] Process exited with code 0
2025-08-01 03:41:26.632 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ed628185-b82b-4292-b338-3661bbd4ee8d] socks connection closed
2025-08-01 03:41:26.632 [info] [command][cce9ff6f-53a6-4914-9fb3-46f00735501b] Socket close event received
2025-08-01 03:41:26.673 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50505 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:42:26.635 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:42:26.637 [info] [command][75c28c73-77ed-4659-aff8-892274c63a4a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""75c28c73-77ed-4659-aff8-892274c63a4a""}
2025-08-01 03:42:26.637 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ddcd671e-6683-464d-9915-9ac615e6290e] received connection request
2025-08-01 03:42:26.638 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:42:26.685 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ddcd671e-6683-464d-9915-9ac615e6290e] socks forwarding established
2025-08-01 03:42:26.730 [info] [command][75c28c73-77ed-4659-aff8-892274c63a4a] Process exited with code 0
2025-08-01 03:42:26.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ddcd671e-6683-464d-9915-9ac615e6290e] socks connection closed
2025-08-01 03:42:26.731 [info] [command][75c28c73-77ed-4659-aff8-892274c63a4a] Socket close event received
2025-08-01 03:42:26.770 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50535 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:43:26.731 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:43:26.733 [info] [command][978f7457-53dd-4816-9bf8-dd3ee45cf249] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""978f7457-53dd-4816-9bf8-dd3ee45cf249""}
2025-08-01 03:43:26.734 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b6616a42-33e0-4300-9094-25a3c8bc4be6] received connection request
2025-08-01 03:43:26.735 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:43:26.775 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b6616a42-33e0-4300-9094-25a3c8bc4be6] socks forwarding established
2025-08-01 03:43:26.828 [info] [command][978f7457-53dd-4816-9bf8-dd3ee45cf249] Process exited with code 0
2025-08-01 03:43:26.829 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b6616a42-33e0-4300-9094-25a3c8bc4be6] socks connection closed
2025-08-01 03:43:26.829 [info] [command][978f7457-53dd-4816-9bf8-dd3ee45cf249] Socket close event received
2025-08-01 03:43:26.869 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50563 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:44:26.832 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:44:26.834 [info] [command][70015f89-6c19-4ab0-8bca-fce5db4eefaf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""70015f89-6c19-4ab0-8bca-fce5db4eefaf""}
2025-08-01 03:44:26.835 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][978e8acc-467b-48da-b9d8-8c47e5511b7b] received connection request
2025-08-01 03:44:26.835 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:44:26.880 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][978e8acc-467b-48da-b9d8-8c47e5511b7b] socks forwarding established
2025-08-01 03:44:26.924 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][978e8acc-467b-48da-b9d8-8c47e5511b7b] socks connection closed
2025-08-01 03:44:26.925 [info] [command][70015f89-6c19-4ab0-8bca-fce5db4eefaf] Process exited with code 0
2025-08-01 03:44:26.925 [info] [command][70015f89-6c19-4ab0-8bca-fce5db4eefaf] Socket close event received
2025-08-01 03:44:26.964 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50579 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:45:26.929 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:45:26.930 [info] [command][86733edf-cb26-4060-bbbd-b206d123343e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""86733edf-cb26-4060-bbbd-b206d123343e""}
2025-08-01 03:45:26.930 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7ec67dc7-d7e7-4b17-86d3-dc8674918256] received connection request
2025-08-01 03:45:26.931 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:45:27.030 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7ec67dc7-d7e7-4b17-86d3-dc8674918256] socks forwarding established
2025-08-01 03:45:27.075 [info] [command][86733edf-cb26-4060-bbbd-b206d123343e] Process exited with code 0
2025-08-01 03:45:27.076 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7ec67dc7-d7e7-4b17-86d3-dc8674918256] socks connection closed
2025-08-01 03:45:27.076 [info] [command][86733edf-cb26-4060-bbbd-b206d123343e] Socket close event received
2025-08-01 03:45:27.118 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50591 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:46:27.083 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:46:27.084 [info] [command][0545ac93-8278-463d-b704-201c7b984506] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0545ac93-8278-463d-b704-201c7b984506""}
2025-08-01 03:46:27.085 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2719eed4-bbef-4e47-901d-0ec47800f521] received connection request
2025-08-01 03:46:27.085 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:46:27.129 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2719eed4-bbef-4e47-901d-0ec47800f521] socks forwarding established
2025-08-01 03:46:27.174 [info] [command][0545ac93-8278-463d-b704-201c7b984506] Process exited with code 0
2025-08-01 03:46:27.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2719eed4-bbef-4e47-901d-0ec47800f521] socks connection closed
2025-08-01 03:46:27.174 [info] [command][0545ac93-8278-463d-b704-201c7b984506] Socket close event received
2025-08-01 03:46:27.215 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50611 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:47:27.180 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:47:27.182 [info] [command][153f3519-8e87-4a8e-ad20-eb939eb69275] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""153f3519-8e87-4a8e-ad20-eb939eb69275""}
2025-08-01 03:47:27.183 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4884eb63-d297-436d-8ac5-37e288132c0d] received connection request
2025-08-01 03:47:27.184 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:47:27.227 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4884eb63-d297-436d-8ac5-37e288132c0d] socks forwarding established
2025-08-01 03:47:27.271 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4884eb63-d297-436d-8ac5-37e288132c0d] socks connection closed
2025-08-01 03:47:27.271 [info] [command][153f3519-8e87-4a8e-ad20-eb939eb69275] Process exited with code 0
2025-08-01 03:47:27.271 [info] [command][153f3519-8e87-4a8e-ad20-eb939eb69275] Socket close event received
2025-08-01 03:47:27.312 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50641 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:48:27.276 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:48:27.278 [info] [command][49d762bf-8214-4b59-9129-6796f4bf56c2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""49d762bf-8214-4b59-9129-6796f4bf56c2""}
2025-08-01 03:48:27.278 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3f69cce-4496-42a7-aa15-f989170847a2] received connection request
2025-08-01 03:48:27.279 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:48:27.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3f69cce-4496-42a7-aa15-f989170847a2] socks forwarding established
2025-08-01 03:48:27.369 [info] [command][49d762bf-8214-4b59-9129-6796f4bf56c2] Process exited with code 0
2025-08-01 03:48:27.369 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3f69cce-4496-42a7-aa15-f989170847a2] socks connection closed
2025-08-01 03:48:27.369 [info] [command][49d762bf-8214-4b59-9129-6796f4bf56c2] Socket close event received
2025-08-01 03:48:27.415 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50668 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:49:27.379 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:49:27.381 [info] [command][8655e200-8e2b-4ae2-b703-741ee191a638] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8655e200-8e2b-4ae2-b703-741ee191a638""}
2025-08-01 03:49:27.381 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a048d36b-6527-4707-865c-d8c40e9ef181] received connection request
2025-08-01 03:49:27.382 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:49:27.430 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a048d36b-6527-4707-865c-d8c40e9ef181] socks forwarding established
2025-08-01 03:49:27.472 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a048d36b-6527-4707-865c-d8c40e9ef181] socks connection closed
2025-08-01 03:49:27.473 [info] [command][8655e200-8e2b-4ae2-b703-741ee191a638] Process exited with code 0
2025-08-01 03:49:27.473 [info] [command][8655e200-8e2b-4ae2-b703-741ee191a638] Socket close event received
2025-08-01 03:49:27.515 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50694 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:50:27.474 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:50:27.476 [info] [command][61f20b9d-4ad0-4747-95d4-e1adfe161aad] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""61f20b9d-4ad0-4747-95d4-e1adfe161aad""}
2025-08-01 03:50:27.477 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][019c2760-3ab8-4db7-b547-1d88089ffdf8] received connection request
2025-08-01 03:50:27.478 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:50:27.606 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][019c2760-3ab8-4db7-b547-1d88089ffdf8] socks forwarding established
2025-08-01 03:50:27.652 [info] [command][61f20b9d-4ad0-4747-95d4-e1adfe161aad] Process exited with code 0
2025-08-01 03:50:27.652 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][019c2760-3ab8-4db7-b547-1d88089ffdf8] socks connection closed
2025-08-01 03:50:27.653 [info] [command][61f20b9d-4ad0-4747-95d4-e1adfe161aad] Socket close event received
2025-08-01 03:50:27.693 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50708 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:51:27.658 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:51:27.659 [info] [command][b0dde1ad-a1a8-4100-ab59-82fc8fd78014] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b0dde1ad-a1a8-4100-ab59-82fc8fd78014""}
2025-08-01 03:51:27.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b86a6e4e-293c-4c9e-8e05-b4369bda4e87] received connection request
2025-08-01 03:51:27.660 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:51:27.708 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b86a6e4e-293c-4c9e-8e05-b4369bda4e87] socks forwarding established
2025-08-01 03:51:27.754 [info] [command][b0dde1ad-a1a8-4100-ab59-82fc8fd78014] Process exited with code 0
2025-08-01 03:51:27.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b86a6e4e-293c-4c9e-8e05-b4369bda4e87] socks connection closed
2025-08-01 03:51:27.754 [info] [command][b0dde1ad-a1a8-4100-ab59-82fc8fd78014] Socket close event received
2025-08-01 03:51:27.796 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50729 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:52:27.758 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:52:27.760 [info] [command][a7c6f000-5c84-4b03-a617-4e10a33c8e89] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a7c6f000-5c84-4b03-a617-4e10a33c8e89""}
2025-08-01 03:52:27.761 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][32937120-6508-49af-a76f-bd2af962f8b4] received connection request
2025-08-01 03:52:27.761 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:52:27.803 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][32937120-6508-49af-a76f-bd2af962f8b4] socks forwarding established
2025-08-01 03:52:27.848 [info] [command][a7c6f000-5c84-4b03-a617-4e10a33c8e89] Process exited with code 0
2025-08-01 03:52:27.849 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][32937120-6508-49af-a76f-bd2af962f8b4] socks connection closed
2025-08-01 03:52:27.849 [info] [command][a7c6f000-5c84-4b03-a617-4e10a33c8e89] Socket close event received
2025-08-01 03:52:27.889 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50763 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:53:27.856 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:53:27.858 [info] [command][4e02f613-9c58-496a-8ea9-cb3aca830a8e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4e02f613-9c58-496a-8ea9-cb3aca830a8e""}
2025-08-01 03:53:27.858 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cbd0b29e-af52-4ad7-a018-98a594c99407] received connection request
2025-08-01 03:53:27.859 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:53:27.903 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cbd0b29e-af52-4ad7-a018-98a594c99407] socks forwarding established
2025-08-01 03:53:27.947 [info] [command][4e02f613-9c58-496a-8ea9-cb3aca830a8e] Process exited with code 0
2025-08-01 03:53:27.948 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cbd0b29e-af52-4ad7-a018-98a594c99407] socks connection closed
2025-08-01 03:53:27.948 [info] [command][4e02f613-9c58-496a-8ea9-cb3aca830a8e] Socket close event received
2025-08-01 03:53:27.987 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50795 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:54:27.957 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:54:27.959 [info] [command][4a17af4c-67d4-4d14-a3ec-97a054143e0d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4a17af4c-67d4-4d14-a3ec-97a054143e0d""}
2025-08-01 03:54:27.960 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][07e861bd-ee93-4f3b-beb2-2474967165fb] received connection request
2025-08-01 03:54:27.961 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:54:28.002 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07e861bd-ee93-4f3b-beb2-2474967165fb] socks forwarding established
2025-08-01 03:54:28.046 [info] [command][4a17af4c-67d4-4d14-a3ec-97a054143e0d] Process exited with code 0
2025-08-01 03:54:28.046 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][07e861bd-ee93-4f3b-beb2-2474967165fb] socks connection closed
2025-08-01 03:54:28.047 [info] [command][4a17af4c-67d4-4d14-a3ec-97a054143e0d] Socket close event received
2025-08-01 03:54:28.086 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50808 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:55:28.051 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:55:28.053 [info] [command][15b6f8ec-c930-4258-a3df-3d7af4f06f8f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""15b6f8ec-c930-4258-a3df-3d7af4f06f8f""}
2025-08-01 03:55:28.053 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5d77d7ff-97a6-44db-9f05-3136e1ff5925] received connection request
2025-08-01 03:55:28.054 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:55:28.129 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d77d7ff-97a6-44db-9f05-3136e1ff5925] socks forwarding established
2025-08-01 03:55:28.175 [info] [command][15b6f8ec-c930-4258-a3df-3d7af4f06f8f] Process exited with code 0
2025-08-01 03:55:28.175 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d77d7ff-97a6-44db-9f05-3136e1ff5925] socks connection closed
2025-08-01 03:55:28.175 [info] [command][15b6f8ec-c930-4258-a3df-3d7af4f06f8f] Socket close event received
2025-08-01 03:55:28.215 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50820 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:56:28.182 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:56:28.184 [info] [command][ec75fe1a-8104-4442-b5aa-f924e01c8928] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ec75fe1a-8104-4442-b5aa-f924e01c8928""}
2025-08-01 03:56:28.185 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ccf1fda2-7404-487c-8451-be06987320f9] received connection request
2025-08-01 03:56:28.186 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:56:28.235 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ccf1fda2-7404-487c-8451-be06987320f9] socks forwarding established
2025-08-01 03:56:28.279 [info] [command][ec75fe1a-8104-4442-b5aa-f924e01c8928] Process exited with code 0
2025-08-01 03:56:28.279 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ccf1fda2-7404-487c-8451-be06987320f9] socks connection closed
2025-08-01 03:56:28.279 [info] [command][ec75fe1a-8104-4442-b5aa-f924e01c8928] Socket close event received
2025-08-01 03:56:28.319 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50835 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:57:28.290 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:57:28.291 [info] [command][77d5da5f-93be-446a-bbaa-2a9b449b3855] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""77d5da5f-93be-446a-bbaa-2a9b449b3855""}
2025-08-01 03:57:28.291 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][97973a3d-45e9-4275-9ce2-b5d7c5f723a3] received connection request
2025-08-01 03:57:28.292 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:57:28.343 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97973a3d-45e9-4275-9ce2-b5d7c5f723a3] socks forwarding established
2025-08-01 03:57:28.387 [info] [command][77d5da5f-93be-446a-bbaa-2a9b449b3855] Process exited with code 0
2025-08-01 03:57:28.388 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97973a3d-45e9-4275-9ce2-b5d7c5f723a3] socks connection closed
2025-08-01 03:57:28.388 [info] [command][77d5da5f-93be-446a-bbaa-2a9b449b3855] Socket close event received
2025-08-01 03:57:28.428 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50864 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:58:28.398 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:58:28.400 [info] [command][9a25eecc-93bd-4dc2-a2e5-0e21317ae0fd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9a25eecc-93bd-4dc2-a2e5-0e21317ae0fd""}
2025-08-01 03:58:28.401 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d1b33cb6-47a0-4f8e-bdf8-bfce9b7e150c] received connection request
2025-08-01 03:58:28.401 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:58:28.443 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d1b33cb6-47a0-4f8e-bdf8-bfce9b7e150c] socks forwarding established
2025-08-01 03:58:28.489 [info] [command][9a25eecc-93bd-4dc2-a2e5-0e21317ae0fd] Process exited with code 0
2025-08-01 03:58:28.489 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d1b33cb6-47a0-4f8e-bdf8-bfce9b7e150c] socks connection closed
2025-08-01 03:58:28.489 [info] [command][9a25eecc-93bd-4dc2-a2e5-0e21317ae0fd] Socket close event received
2025-08-01 03:58:28.540 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50893 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 03:59:28.499 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 03:59:28.499 [info] [command][725877b6-5e3d-4acb-9463-77abcc7623e5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""725877b6-5e3d-4acb-9463-77abcc7623e5""}
2025-08-01 03:59:28.500 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6587eafc-31d2-4874-a8e0-6d158768fd27] received connection request
2025-08-01 03:59:28.500 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 03:59:28.542 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6587eafc-31d2-4874-a8e0-6d158768fd27] socks forwarding established
2025-08-01 03:59:28.588 [info] [command][725877b6-5e3d-4acb-9463-77abcc7623e5] Process exited with code 0
2025-08-01 03:59:28.588 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6587eafc-31d2-4874-a8e0-6d158768fd27] socks connection closed
2025-08-01 03:59:28.588 [info] [command][725877b6-5e3d-4acb-9463-77abcc7623e5] Socket close event received
2025-08-01 03:59:28.628 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50911 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:00:28.598 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:00:28.600 [info] [command][d723a0ff-9d1a-407a-9ef5-7c66a2340e8a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d723a0ff-9d1a-407a-9ef5-7c66a2340e8a""}
2025-08-01 04:00:28.600 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a04a8e5e-acf5-415b-b7a2-591f7e284e8e] received connection request
2025-08-01 04:00:28.601 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:00:28.655 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a04a8e5e-acf5-415b-b7a2-591f7e284e8e] socks forwarding established
2025-08-01 04:00:28.962 [info] [command][d723a0ff-9d1a-407a-9ef5-7c66a2340e8a] Process exited with code 0
2025-08-01 04:00:28.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a04a8e5e-acf5-415b-b7a2-591f7e284e8e] socks connection closed
2025-08-01 04:00:28.963 [info] [command][d723a0ff-9d1a-407a-9ef5-7c66a2340e8a] Socket close event received
2025-08-01 04:00:29.005 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50926 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:01:28.967 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:01:28.968 [info] [command][8f58683e-82b4-48a7-9ae4-e360874c3ae4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8f58683e-82b4-48a7-9ae4-e360874c3ae4""}
2025-08-01 04:01:28.969 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][76063a56-c37e-4723-8b7c-0a10c72c6fd5] received connection request
2025-08-01 04:01:28.970 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:01:29.012 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76063a56-c37e-4723-8b7c-0a10c72c6fd5] socks forwarding established
2025-08-01 04:01:29.057 [info] [command][8f58683e-82b4-48a7-9ae4-e360874c3ae4] Process exited with code 0
2025-08-01 04:01:29.057 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76063a56-c37e-4723-8b7c-0a10c72c6fd5] socks connection closed
2025-08-01 04:01:29.058 [info] [command][8f58683e-82b4-48a7-9ae4-e360874c3ae4] Socket close event received
2025-08-01 04:01:29.096 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50939 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:02:29.060 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:02:29.062 [info] [command][53e70b90-b09f-487a-879a-8b7de28b106b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""53e70b90-b09f-487a-879a-8b7de28b106b""}
2025-08-01 04:02:29.063 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d3d1865f-a5cd-4386-bc1e-06528bc99a3e] received connection request
2025-08-01 04:02:29.064 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:02:29.110 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3d1865f-a5cd-4386-bc1e-06528bc99a3e] socks forwarding established
2025-08-01 04:02:29.154 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d3d1865f-a5cd-4386-bc1e-06528bc99a3e] socks connection closed
2025-08-01 04:02:29.155 [info] [command][53e70b90-b09f-487a-879a-8b7de28b106b] Process exited with code 0
2025-08-01 04:02:29.155 [info] [command][53e70b90-b09f-487a-879a-8b7de28b106b] Socket close event received
2025-08-01 04:02:29.193 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50966 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:03:29.162 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:03:29.164 [info] [command][ec118c5b-4200-4331-a6ea-9e32798b9783] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ec118c5b-4200-4331-a6ea-9e32798b9783""}
2025-08-01 04:03:29.165 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][34cbaa51-e175-4b4f-91e2-91156040009e] received connection request
2025-08-01 04:03:29.166 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:03:29.209 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][34cbaa51-e175-4b4f-91e2-91156040009e] socks forwarding established
2025-08-01 04:03:29.255 [info] [command][ec118c5b-4200-4331-a6ea-9e32798b9783] Process exited with code 0
2025-08-01 04:03:29.256 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][34cbaa51-e175-4b4f-91e2-91156040009e] socks connection closed
2025-08-01 04:03:29.256 [info] [command][ec118c5b-4200-4331-a6ea-9e32798b9783] Socket close event received
2025-08-01 04:03:29.295 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 50994 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:04:29.261 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:04:29.263 [info] [command][b1d8e1b7-6e92-4903-bc35-330b6800af00] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b1d8e1b7-6e92-4903-bc35-330b6800af00""}
2025-08-01 04:04:29.264 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d463ca8a-88ed-4b69-9669-d6466ba4dcc4] received connection request
2025-08-01 04:04:29.264 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:04:29.384 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d463ca8a-88ed-4b69-9669-d6466ba4dcc4] socks forwarding established
2025-08-01 04:04:29.430 [info] [command][b1d8e1b7-6e92-4903-bc35-330b6800af00] Process exited with code 0
2025-08-01 04:04:29.430 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d463ca8a-88ed-4b69-9669-d6466ba4dcc4] socks connection closed
2025-08-01 04:04:29.430 [info] [command][b1d8e1b7-6e92-4903-bc35-330b6800af00] Socket close event received
2025-08-01 04:04:29.470 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51010 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:05:29.432 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:05:29.433 [info] [command][504b9753-4ef5-421e-938a-1c9dd6ff5c11] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""504b9753-4ef5-421e-938a-1c9dd6ff5c11""}
2025-08-01 04:05:29.434 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6a56237a-249e-4053-a98a-12fdf977a641] received connection request
2025-08-01 04:05:29.434 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:05:29.476 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6a56237a-249e-4053-a98a-12fdf977a641] socks forwarding established
2025-08-01 04:05:29.521 [info] [command][504b9753-4ef5-421e-938a-1c9dd6ff5c11] Process exited with code 0
2025-08-01 04:05:29.522 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6a56237a-249e-4053-a98a-12fdf977a641] socks connection closed
2025-08-01 04:05:29.522 [info] [command][504b9753-4ef5-421e-938a-1c9dd6ff5c11] Socket close event received
2025-08-01 04:05:29.561 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51024 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:06:29.532 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:06:29.534 [info] [command][7fa98792-45b2-42ce-843f-278cb4ce48bc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7fa98792-45b2-42ce-843f-278cb4ce48bc""}
2025-08-01 04:06:29.534 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6236a5c2-518c-4e4b-bfab-359235a20b8c] received connection request
2025-08-01 04:06:29.535 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:06:29.580 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6236a5c2-518c-4e4b-bfab-359235a20b8c] socks forwarding established
2025-08-01 04:06:29.626 [info] [command][7fa98792-45b2-42ce-843f-278cb4ce48bc] Process exited with code 0
2025-08-01 04:06:29.626 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6236a5c2-518c-4e4b-bfab-359235a20b8c] socks connection closed
2025-08-01 04:06:29.626 [info] [command][7fa98792-45b2-42ce-843f-278cb4ce48bc] Socket close event received
2025-08-01 04:06:29.666 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51045 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:07:29.636 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:07:29.638 [info] [command][d61e2e8e-3323-4fd7-b3c5-b62f1b2d9cf0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d61e2e8e-3323-4fd7-b3c5-b62f1b2d9cf0""}
2025-08-01 04:07:29.638 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][78b36074-0c96-4051-8878-7be62623ed26] received connection request
2025-08-01 04:07:29.639 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:07:29.680 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78b36074-0c96-4051-8878-7be62623ed26] socks forwarding established
2025-08-01 04:07:29.724 [info] [command][d61e2e8e-3323-4fd7-b3c5-b62f1b2d9cf0] Process exited with code 0
2025-08-01 04:07:29.724 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78b36074-0c96-4051-8878-7be62623ed26] socks connection closed
2025-08-01 04:07:29.725 [info] [command][d61e2e8e-3323-4fd7-b3c5-b62f1b2d9cf0] Socket close event received
2025-08-01 04:07:29.769 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51077 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:08:29.729 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:08:29.729 [info] [command][57c477d9-6011-48ba-9b19-df69548a177c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""57c477d9-6011-48ba-9b19-df69548a177c""}
2025-08-01 04:08:29.730 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][48fafd98-d3dd-4ee4-8b5e-c2c7d16d1fb8] received connection request
2025-08-01 04:08:29.730 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:08:29.778 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48fafd98-d3dd-4ee4-8b5e-c2c7d16d1fb8] socks forwarding established
2025-08-01 04:08:29.822 [info] [command][57c477d9-6011-48ba-9b19-df69548a177c] Process exited with code 0
2025-08-01 04:08:29.822 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][48fafd98-d3dd-4ee4-8b5e-c2c7d16d1fb8] socks connection closed
2025-08-01 04:08:29.823 [info] [command][57c477d9-6011-48ba-9b19-df69548a177c] Socket close event received
2025-08-01 04:08:29.868 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51108 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:09:29.828 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:09:29.830 [info] [command][4a0cb86a-99ab-45e9-a66c-9faf915d4b06] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4a0cb86a-99ab-45e9-a66c-9faf915d4b06""}
2025-08-01 04:09:29.830 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][85fc6780-1708-424d-b1bd-2b4f68c6b3aa] received connection request
2025-08-01 04:09:29.831 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:09:29.888 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85fc6780-1708-424d-b1bd-2b4f68c6b3aa] socks forwarding established
2025-08-01 04:09:29.945 [info] [command][4a0cb86a-99ab-45e9-a66c-9faf915d4b06] Process exited with code 0
2025-08-01 04:09:29.946 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85fc6780-1708-424d-b1bd-2b4f68c6b3aa] socks connection closed
2025-08-01 04:09:29.946 [info] [command][4a0cb86a-99ab-45e9-a66c-9faf915d4b06] Socket close event received
2025-08-01 04:09:29.994 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51128 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:10:29.951 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:10:29.953 [info] [command][84893b88-580a-4bb1-91f4-4acd8bf4d877] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""84893b88-580a-4bb1-91f4-4acd8bf4d877""}
2025-08-01 04:10:29.953 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][107ca9b8-cea7-4f0f-8f40-b60888866f66] received connection request
2025-08-01 04:10:29.954 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:10:29.998 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][107ca9b8-cea7-4f0f-8f40-b60888866f66] socks forwarding established
2025-08-01 04:10:30.044 [info] [command][84893b88-580a-4bb1-91f4-4acd8bf4d877] Process exited with code 0
2025-08-01 04:10:30.044 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][107ca9b8-cea7-4f0f-8f40-b60888866f66] socks connection closed
2025-08-01 04:10:30.045 [info] [command][84893b88-580a-4bb1-91f4-4acd8bf4d877] Socket close event received
2025-08-01 04:10:30.085 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51150 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:11:30.051 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:11:30.052 [info] [command][4873af0f-6623-4a57-b9cd-2381463e9b74] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4873af0f-6623-4a57-b9cd-2381463e9b74""}
2025-08-01 04:11:30.053 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3cfd054c-197b-4fbd-99bc-4c91f5b39a67] received connection request
2025-08-01 04:11:30.053 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:11:30.101 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3cfd054c-197b-4fbd-99bc-4c91f5b39a67] socks forwarding established
2025-08-01 04:11:30.152 [info] [command][4873af0f-6623-4a57-b9cd-2381463e9b74] Process exited with code 0
2025-08-01 04:11:30.153 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3cfd054c-197b-4fbd-99bc-4c91f5b39a67] socks connection closed
2025-08-01 04:11:30.153 [info] [command][4873af0f-6623-4a57-b9cd-2381463e9b74] Socket close event received
2025-08-01 04:11:30.192 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51165 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:12:30.157 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:12:30.159 [info] [command][a31bfce2-fa31-421c-80fa-1379e9bb7280] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a31bfce2-fa31-421c-80fa-1379e9bb7280""}
2025-08-01 04:12:30.160 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ac35650c-a4a5-47cd-8387-316d6e973de4] received connection request
2025-08-01 04:12:30.161 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:12:30.207 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac35650c-a4a5-47cd-8387-316d6e973de4] socks forwarding established
2025-08-01 04:12:30.256 [info] [command][a31bfce2-fa31-421c-80fa-1379e9bb7280] Process exited with code 0
2025-08-01 04:12:30.256 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac35650c-a4a5-47cd-8387-316d6e973de4] socks connection closed
2025-08-01 04:12:30.256 [info] [command][a31bfce2-fa31-421c-80fa-1379e9bb7280] Socket close event received
2025-08-01 04:12:30.297 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51195 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:13:30.263 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:13:30.265 [info] [command][d0029739-7826-4630-9da4-2376760bb360] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d0029739-7826-4630-9da4-2376760bb360""}
2025-08-01 04:13:30.266 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bc49ac49-c8c6-4175-b813-60551b58f41c] received connection request
2025-08-01 04:13:30.266 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:13:30.322 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bc49ac49-c8c6-4175-b813-60551b58f41c] socks forwarding established
2025-08-01 04:13:30.368 [info] [command][d0029739-7826-4630-9da4-2376760bb360] Process exited with code 0
2025-08-01 04:13:30.369 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bc49ac49-c8c6-4175-b813-60551b58f41c] socks connection closed
2025-08-01 04:13:30.369 [info] [command][d0029739-7826-4630-9da4-2376760bb360] Socket close event received
2025-08-01 04:13:30.408 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51228 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:14:30.376 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:14:30.377 [info] [command][4c4d9380-6e19-4dd6-9fe2-2bad0b7b5959] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4c4d9380-6e19-4dd6-9fe2-2bad0b7b5959""}
2025-08-01 04:14:30.378 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e5ab24da-387c-4fdd-b2f6-07fd95a22c77] received connection request
2025-08-01 04:14:30.378 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:14:30.425 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e5ab24da-387c-4fdd-b2f6-07fd95a22c77] socks forwarding established
2025-08-01 04:14:30.475 [info] [command][4c4d9380-6e19-4dd6-9fe2-2bad0b7b5959] Process exited with code 0
2025-08-01 04:14:30.476 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e5ab24da-387c-4fdd-b2f6-07fd95a22c77] socks connection closed
2025-08-01 04:14:30.476 [info] [command][4c4d9380-6e19-4dd6-9fe2-2bad0b7b5959] Socket close event received
2025-08-01 04:14:30.529 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51240 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:15:30.482 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:15:30.484 [info] [command][ba8083b3-ff7c-431a-b0e8-a65d0aa61560] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ba8083b3-ff7c-431a-b0e8-a65d0aa61560""}
2025-08-01 04:15:30.485 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d01015f3-e64d-4f4c-85a7-bcf833915643] received connection request
2025-08-01 04:15:30.485 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:15:30.536 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d01015f3-e64d-4f4c-85a7-bcf833915643] socks forwarding established
2025-08-01 04:15:30.801 [info] [command][ba8083b3-ff7c-431a-b0e8-a65d0aa61560] Process exited with code 0
2025-08-01 04:15:30.801 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d01015f3-e64d-4f4c-85a7-bcf833915643] socks connection closed
2025-08-01 04:15:30.801 [info] [command][ba8083b3-ff7c-431a-b0e8-a65d0aa61560] Socket close event received
2025-08-01 04:15:30.861 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51253 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:16:30.806 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:16:30.807 [info] [command][731d21bf-6f16-4cd6-8e1b-28c77fb26b7f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""731d21bf-6f16-4cd6-8e1b-28c77fb26b7f""}
2025-08-01 04:16:30.808 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][65a23aae-0807-4f0b-941f-4c9c2888e3a6] received connection request
2025-08-01 04:16:30.808 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:16:30.856 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][65a23aae-0807-4f0b-941f-4c9c2888e3a6] socks forwarding established
2025-08-01 04:16:30.902 [info] [command][731d21bf-6f16-4cd6-8e1b-28c77fb26b7f] Process exited with code 0
2025-08-01 04:16:30.902 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][65a23aae-0807-4f0b-941f-4c9c2888e3a6] socks connection closed
2025-08-01 04:16:30.902 [info] [command][731d21bf-6f16-4cd6-8e1b-28c77fb26b7f] Socket close event received
2025-08-01 04:16:30.942 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51265 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:17:30.912 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:17:30.913 [info] [command][d5919421-dfbb-4ece-a77b-eb0b91e40438] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d5919421-dfbb-4ece-a77b-eb0b91e40438""}
2025-08-01 04:17:30.913 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ce722eaa-0078-4146-b228-78b4d65e7cbb] received connection request
2025-08-01 04:17:30.914 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:17:30.957 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ce722eaa-0078-4146-b228-78b4d65e7cbb] socks forwarding established
2025-08-01 04:17:31.001 [info] [command][d5919421-dfbb-4ece-a77b-eb0b91e40438] Process exited with code 0
2025-08-01 04:17:31.001 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ce722eaa-0078-4146-b228-78b4d65e7cbb] socks connection closed
2025-08-01 04:17:31.001 [info] [command][d5919421-dfbb-4ece-a77b-eb0b91e40438] Socket close event received
2025-08-01 04:17:31.042 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51302 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:18:31.009 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:18:31.010 [info] [command][b7172637-722b-4004-a93a-39aa3944640a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7172637-722b-4004-a93a-39aa3944640a""}
2025-08-01 04:18:31.011 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][360a683a-dfee-4800-875f-62511e5d7a5d] received connection request
2025-08-01 04:18:31.012 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:18:31.125 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][360a683a-dfee-4800-875f-62511e5d7a5d] socks forwarding established
2025-08-01 04:18:31.175 [info] [command][b7172637-722b-4004-a93a-39aa3944640a] Process exited with code 0
2025-08-01 04:18:31.175 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][360a683a-dfee-4800-875f-62511e5d7a5d] socks connection closed
2025-08-01 04:18:31.175 [info] [command][b7172637-722b-4004-a93a-39aa3944640a] Socket close event received
2025-08-01 04:18:31.218 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51336 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:19:31.176 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:19:31.178 [info] [command][a0b4e43f-44d6-4b42-840c-6dda2cededa1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a0b4e43f-44d6-4b42-840c-6dda2cededa1""}
2025-08-01 04:19:31.179 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3c8904f-4c5e-425a-8b11-04c46d3eeab1] received connection request
2025-08-01 04:19:31.179 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:19:31.222 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3c8904f-4c5e-425a-8b11-04c46d3eeab1] socks forwarding established
2025-08-01 04:19:31.268 [info] [command][a0b4e43f-44d6-4b42-840c-6dda2cededa1] Process exited with code 0
2025-08-01 04:19:31.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3c8904f-4c5e-425a-8b11-04c46d3eeab1] socks connection closed
2025-08-01 04:19:31.268 [info] [command][a0b4e43f-44d6-4b42-840c-6dda2cededa1] Socket close event received
2025-08-01 04:19:31.307 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51353 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:20:31.278 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:20:31.280 [info] [command][339c301a-efe9-4414-8a22-0f31053105e5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""339c301a-efe9-4414-8a22-0f31053105e5""}
2025-08-01 04:20:31.281 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0f631d53-04e2-42d5-803e-ff776b1bc30c] received connection request
2025-08-01 04:20:31.282 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:20:31.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0f631d53-04e2-42d5-803e-ff776b1bc30c] socks forwarding established
2025-08-01 04:20:31.370 [info] [command][339c301a-efe9-4414-8a22-0f31053105e5] Process exited with code 0
2025-08-01 04:20:31.370 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0f631d53-04e2-42d5-803e-ff776b1bc30c] socks connection closed
2025-08-01 04:20:31.370 [info] [command][339c301a-efe9-4414-8a22-0f31053105e5] Socket close event received
2025-08-01 04:20:31.421 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51377 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:21:31.376 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:21:31.378 [info] [command][890f3a70-373a-411f-9354-a727bdc4c3ac] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""890f3a70-373a-411f-9354-a727bdc4c3ac""}
2025-08-01 04:21:31.379 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a8713748-a676-4486-adff-0b2345e6612a] received connection request
2025-08-01 04:21:31.379 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:21:31.421 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8713748-a676-4486-adff-0b2345e6612a] socks forwarding established
2025-08-01 04:21:31.465 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8713748-a676-4486-adff-0b2345e6612a] socks connection closed
2025-08-01 04:21:31.465 [info] [command][890f3a70-373a-411f-9354-a727bdc4c3ac] Process exited with code 0
2025-08-01 04:21:31.465 [info] [command][890f3a70-373a-411f-9354-a727bdc4c3ac] Socket close event received
2025-08-01 04:21:31.505 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51413 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:22:31.466 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:22:31.467 [info] [command][dbf58dc5-34e5-4b79-8a15-ef470c962295] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dbf58dc5-34e5-4b79-8a15-ef470c962295""}
2025-08-01 04:22:31.468 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][efd3fb87-7680-4abb-8214-2f252a8c34ba] received connection request
2025-08-01 04:22:31.468 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:22:31.511 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efd3fb87-7680-4abb-8214-2f252a8c34ba] socks forwarding established
2025-08-01 04:22:31.556 [info] [command][dbf58dc5-34e5-4b79-8a15-ef470c962295] Process exited with code 0
2025-08-01 04:22:31.556 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efd3fb87-7680-4abb-8214-2f252a8c34ba] socks connection closed
2025-08-01 04:22:31.556 [info] [command][dbf58dc5-34e5-4b79-8a15-ef470c962295] Socket close event received
2025-08-01 04:22:31.595 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51441 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:23:31.564 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:23:31.566 [info] [command][71ed59d4-106c-4067-b1a4-bf656545b0c5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""71ed59d4-106c-4067-b1a4-bf656545b0c5""}
2025-08-01 04:23:31.566 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a0a5b945-ed8e-4fe3-8af6-c4c6ef225d69] received connection request
2025-08-01 04:23:31.567 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:23:31.696 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a0a5b945-ed8e-4fe3-8af6-c4c6ef225d69] socks forwarding established
2025-08-01 04:23:31.743 [info] [command][71ed59d4-106c-4067-b1a4-bf656545b0c5] Process exited with code 0
2025-08-01 04:23:31.743 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a0a5b945-ed8e-4fe3-8af6-c4c6ef225d69] socks connection closed
2025-08-01 04:23:31.743 [info] [command][71ed59d4-106c-4067-b1a4-bf656545b0c5] Socket close event received
2025-08-01 04:23:31.783 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51475 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:24:31.752 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:24:31.753 [info] [command][f0f089df-8f16-4ff3-bd46-f7e7dcf5c7cb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f0f089df-8f16-4ff3-bd46-f7e7dcf5c7cb""}
2025-08-01 04:24:31.754 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7964c304-dbe1-498e-965c-871a72f205ce] received connection request
2025-08-01 04:24:31.754 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:24:31.799 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7964c304-dbe1-498e-965c-871a72f205ce] socks forwarding established
2025-08-01 04:24:31.842 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7964c304-dbe1-498e-965c-871a72f205ce] socks connection closed
2025-08-01 04:24:31.842 [info] [command][f0f089df-8f16-4ff3-bd46-f7e7dcf5c7cb] Process exited with code 0
2025-08-01 04:24:31.843 [info] [command][f0f089df-8f16-4ff3-bd46-f7e7dcf5c7cb] Socket close event received
2025-08-01 04:24:31.882 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51490 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:25:31.852 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:25:31.853 [info] [command][0d56df83-787b-4a22-8722-0e0d4d265032] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0d56df83-787b-4a22-8722-0e0d4d265032""}
2025-08-01 04:25:31.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a78c999f-be09-44a9-a3bf-f7c3068b2d26] received connection request
2025-08-01 04:25:31.853 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:25:31.896 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a78c999f-be09-44a9-a3bf-f7c3068b2d26] socks forwarding established
2025-08-01 04:25:31.942 [info] [command][0d56df83-787b-4a22-8722-0e0d4d265032] Process exited with code 0
2025-08-01 04:25:31.942 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a78c999f-be09-44a9-a3bf-f7c3068b2d26] socks connection closed
2025-08-01 04:25:31.943 [info] [command][0d56df83-787b-4a22-8722-0e0d4d265032] Socket close event received
2025-08-01 04:25:31.982 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51510 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:26:31.946 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:26:31.947 [info] [command][ae917204-c645-4f19-8ce5-e9be9d74543c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ae917204-c645-4f19-8ce5-e9be9d74543c""}
2025-08-01 04:26:31.948 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][51a1c750-a608-41d1-91bb-ec622f70079c] received connection request
2025-08-01 04:26:31.948 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:26:31.992 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][51a1c750-a608-41d1-91bb-ec622f70079c] socks forwarding established
2025-08-01 04:26:32.593 [info] [command][ae917204-c645-4f19-8ce5-e9be9d74543c] Process exited with code 0
2025-08-01 04:26:32.593 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][51a1c750-a608-41d1-91bb-ec622f70079c] socks connection closed
2025-08-01 04:26:32.593 [info] [command][ae917204-c645-4f19-8ce5-e9be9d74543c] Socket close event received
2025-08-01 04:26:32.651 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51533 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:27:32.598 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:27:32.599 [info] [command][fc8676fe-a41e-4a47-afb2-684d7984e8f1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fc8676fe-a41e-4a47-afb2-684d7984e8f1""}
2025-08-01 04:27:32.600 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0577f646-f220-4de0-a2fa-591e5e3be06d] received connection request
2025-08-01 04:27:32.600 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:27:32.642 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0577f646-f220-4de0-a2fa-591e5e3be06d] socks forwarding established
2025-08-01 04:27:32.687 [info] [command][fc8676fe-a41e-4a47-afb2-684d7984e8f1] Process exited with code 0
2025-08-01 04:27:32.688 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0577f646-f220-4de0-a2fa-591e5e3be06d] socks connection closed
2025-08-01 04:27:32.688 [info] [command][fc8676fe-a41e-4a47-afb2-684d7984e8f1] Socket close event received
2025-08-01 04:27:32.741 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51564 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:28:32.695 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:28:33.608 [info] [command][8b471468-a95e-424b-9c52-72077b979304] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8b471468-a95e-424b-9c52-72077b979304""}
2025-08-01 04:28:33.609 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b854ca65-5af9-45e1-bb5e-13c7589f6b6b] received connection request
2025-08-01 04:28:33.609 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:28:33.881 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b854ca65-5af9-45e1-bb5e-13c7589f6b6b] socks forwarding established
2025-08-01 04:28:33.956 [info] [command][8b471468-a95e-424b-9c52-72077b979304] Process exited with code 0
2025-08-01 04:28:33.956 [info] [command][8b471468-a95e-424b-9c52-72077b979304] Socket close event received
2025-08-01 04:28:34.006 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51597 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:28:34.006 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b854ca65-5af9-45e1-bb5e-13c7589f6b6b] socks connection closed
2025-08-01 04:29:33.957 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:29:33.958 [info] [command][e27b2a7e-069a-4ba2-a24f-2f49f0ec2fc8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e27b2a7e-069a-4ba2-a24f-2f49f0ec2fc8""}
2025-08-01 04:29:33.958 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ebad2e5d-4198-442c-aea1-8a606ce73bd9] received connection request
2025-08-01 04:29:33.959 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:29:34.005 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ebad2e5d-4198-442c-aea1-8a606ce73bd9] socks forwarding established
2025-08-01 04:29:34.050 [info] [command][e27b2a7e-069a-4ba2-a24f-2f49f0ec2fc8] Process exited with code 0
2025-08-01 04:29:34.050 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ebad2e5d-4198-442c-aea1-8a606ce73bd9] socks connection closed
2025-08-01 04:29:34.050 [info] [command][e27b2a7e-069a-4ba2-a24f-2f49f0ec2fc8] Socket close event received
2025-08-01 04:29:34.090 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51622 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:30:34.059 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:30:34.059 [info] [command][d567c2b0-6c6c-4354-994c-a6b7a80fd2a8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d567c2b0-6c6c-4354-994c-a6b7a80fd2a8""}
2025-08-01 04:30:34.060 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][90fd6a67-4c62-4138-8e53-97811d5eae37] received connection request
2025-08-01 04:30:34.060 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:30:34.177 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][90fd6a67-4c62-4138-8e53-97811d5eae37] socks forwarding established
2025-08-01 04:30:34.222 [info] [command][d567c2b0-6c6c-4354-994c-a6b7a80fd2a8] Process exited with code 0
2025-08-01 04:30:34.222 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][90fd6a67-4c62-4138-8e53-97811d5eae37] socks connection closed
2025-08-01 04:30:34.222 [info] [command][d567c2b0-6c6c-4354-994c-a6b7a80fd2a8] Socket close event received
2025-08-01 04:30:34.266 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51644 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:31:34.227 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:31:34.229 [info] [command][cd1ccd46-0c58-4581-a758-a6d6857ae1ba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cd1ccd46-0c58-4581-a758-a6d6857ae1ba""}
2025-08-01 04:31:34.229 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bbc2d0f7-e07a-461d-abd6-a376a4bc2d77] received connection request
2025-08-01 04:31:34.230 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:31:34.278 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bbc2d0f7-e07a-461d-abd6-a376a4bc2d77] socks forwarding established
2025-08-01 04:31:34.322 [info] [command][cd1ccd46-0c58-4581-a758-a6d6857ae1ba] Process exited with code 0
2025-08-01 04:31:34.323 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bbc2d0f7-e07a-461d-abd6-a376a4bc2d77] socks connection closed
2025-08-01 04:31:34.323 [info] [command][cd1ccd46-0c58-4581-a758-a6d6857ae1ba] Socket close event received
2025-08-01 04:31:34.363 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51669 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:32:34.329 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:32:34.330 [info] [command][e92b06b5-86b3-40cc-b6b0-a87f3e582431] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e92b06b5-86b3-40cc-b6b0-a87f3e582431""}
2025-08-01 04:32:34.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][49439971-967b-4335-93bb-ff7fa71f069b] received connection request
2025-08-01 04:32:34.331 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:32:34.436 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][49439971-967b-4335-93bb-ff7fa71f069b] socks forwarding established
2025-08-01 04:32:34.481 [info] [command][e92b06b5-86b3-40cc-b6b0-a87f3e582431] Process exited with code 0
2025-08-01 04:32:34.481 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][49439971-967b-4335-93bb-ff7fa71f069b] socks connection closed
2025-08-01 04:32:34.481 [info] [command][e92b06b5-86b3-40cc-b6b0-a87f3e582431] Socket close event received
2025-08-01 04:32:34.550 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51698 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:33:34.491 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:33:34.492 [info] [command][fb206d6d-2977-4bc6-9c11-0049d665e0f0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fb206d6d-2977-4bc6-9c11-0049d665e0f0""}
2025-08-01 04:33:34.493 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ac4500f9-caa0-4a0c-8dff-4ba50849b9d5] received connection request
2025-08-01 04:33:34.493 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:33:34.537 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac4500f9-caa0-4a0c-8dff-4ba50849b9d5] socks forwarding established
2025-08-01 04:33:34.581 [info] [command][fb206d6d-2977-4bc6-9c11-0049d665e0f0] Process exited with code 0
2025-08-01 04:33:34.582 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ac4500f9-caa0-4a0c-8dff-4ba50849b9d5] socks connection closed
2025-08-01 04:33:34.582 [info] [command][fb206d6d-2977-4bc6-9c11-0049d665e0f0] Socket close event received
2025-08-01 04:33:34.622 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51736 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:34:34.585 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:34:34.585 [info] [command][7ed047a3-bf7c-4dd5-86eb-8b0091011286] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7ed047a3-bf7c-4dd5-86eb-8b0091011286""}
2025-08-01 04:34:34.586 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e14930e7-8d43-40a0-b76b-ff74008cb685] received connection request
2025-08-01 04:34:34.586 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:34:34.631 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e14930e7-8d43-40a0-b76b-ff74008cb685] socks forwarding established
2025-08-01 04:34:34.676 [info] [command][7ed047a3-bf7c-4dd5-86eb-8b0091011286] Process exited with code 0
2025-08-01 04:34:34.676 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e14930e7-8d43-40a0-b76b-ff74008cb685] socks connection closed
2025-08-01 04:34:34.676 [info] [command][7ed047a3-bf7c-4dd5-86eb-8b0091011286] Socket close event received
2025-08-01 04:34:34.716 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51754 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:35:34.683 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:35:34.684 [info] [command][91124d33-2d00-450b-a790-534ef7de4cbb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""91124d33-2d00-450b-a790-534ef7de4cbb""}
2025-08-01 04:35:34.685 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e9af1ffc-3f5b-4942-9ba0-24d2c0cb9bbe] received connection request
2025-08-01 04:35:34.685 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:35:34.727 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9af1ffc-3f5b-4942-9ba0-24d2c0cb9bbe] socks forwarding established
2025-08-01 04:35:34.774 [info] [command][91124d33-2d00-450b-a790-534ef7de4cbb] Process exited with code 0
2025-08-01 04:35:34.774 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9af1ffc-3f5b-4942-9ba0-24d2c0cb9bbe] socks connection closed
2025-08-01 04:35:34.774 [info] [command][91124d33-2d00-450b-a790-534ef7de4cbb] Socket close event received
2025-08-01 04:35:34.814 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51781 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:36:34.777 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:36:34.778 [info] [command][40dfad60-6863-4356-bcc1-589fcca1ed8b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""40dfad60-6863-4356-bcc1-589fcca1ed8b""}
2025-08-01 04:36:34.780 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8e7fa154-ccd5-48c0-932a-a72fd8cbf0ef] received connection request
2025-08-01 04:36:34.780 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:36:34.821 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8e7fa154-ccd5-48c0-932a-a72fd8cbf0ef] socks forwarding established
2025-08-01 04:36:34.866 [info] [command][40dfad60-6863-4356-bcc1-589fcca1ed8b] Process exited with code 0
2025-08-01 04:36:34.866 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8e7fa154-ccd5-48c0-932a-a72fd8cbf0ef] socks connection closed
2025-08-01 04:36:34.866 [info] [command][40dfad60-6863-4356-bcc1-589fcca1ed8b] Socket close event received
2025-08-01 04:36:34.907 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51799 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:37:34.868 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:37:34.870 [info] [command][5d2992f2-73b0-484c-8751-5ae8450898f3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5d2992f2-73b0-484c-8751-5ae8450898f3""}
2025-08-01 04:37:34.871 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][02a470b7-bf7e-40a7-bc89-241c5fc9d045] received connection request
2025-08-01 04:37:34.872 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:37:34.917 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][02a470b7-bf7e-40a7-bc89-241c5fc9d045] socks forwarding established
2025-08-01 04:37:34.961 [info] [command][5d2992f2-73b0-484c-8751-5ae8450898f3] Process exited with code 0
2025-08-01 04:37:34.962 [info] [command][5d2992f2-73b0-484c-8751-5ae8450898f3] Socket close event received
2025-08-01 04:37:34.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][02a470b7-bf7e-40a7-bc89-241c5fc9d045] socks connection closed
2025-08-01 04:37:35.075 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51834 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:38:34.971 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:38:34.972 [info] [command][cb0a17da-3f60-4c7d-845f-452011356a2a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cb0a17da-3f60-4c7d-845f-452011356a2a""}
2025-08-01 04:38:34.973 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c04b7ffa-1573-4193-bbca-fdd136f506ab] received connection request
2025-08-01 04:38:34.973 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:38:35.023 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c04b7ffa-1573-4193-bbca-fdd136f506ab] socks forwarding established
2025-08-01 04:38:35.066 [info] [command][cb0a17da-3f60-4c7d-845f-452011356a2a] Process exited with code 0
2025-08-01 04:38:35.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c04b7ffa-1573-4193-bbca-fdd136f506ab] socks connection closed
2025-08-01 04:38:35.067 [info] [command][cb0a17da-3f60-4c7d-845f-452011356a2a] Socket close event received
2025-08-01 04:38:35.110 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51870 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:39:35.077 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:39:35.079 [info] [command][247c7c0f-df5b-42d5-ae81-f64c801397b6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""247c7c0f-df5b-42d5-ae81-f64c801397b6""}
2025-08-01 04:39:35.080 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0c89c208-091a-486b-afde-627f9b260e9b] received connection request
2025-08-01 04:39:35.080 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:39:35.127 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c89c208-091a-486b-afde-627f9b260e9b] socks forwarding established
2025-08-01 04:39:35.171 [info] [command][247c7c0f-df5b-42d5-ae81-f64c801397b6] Process exited with code 0
2025-08-01 04:39:35.171 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c89c208-091a-486b-afde-627f9b260e9b] socks connection closed
2025-08-01 04:39:35.171 [info] [command][247c7c0f-df5b-42d5-ae81-f64c801397b6] Socket close event received
2025-08-01 04:39:35.210 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51909 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:40:35.172 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:40:35.174 [info] [command][56780472-ede0-49d3-9be1-ee5b5ba1ec1b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""56780472-ede0-49d3-9be1-ee5b5ba1ec1b""}
2025-08-01 04:40:35.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8d76f2af-2c05-46ae-bbf6-875143383e44] received connection request
2025-08-01 04:40:35.175 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:40:35.221 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8d76f2af-2c05-46ae-bbf6-875143383e44] socks forwarding established
2025-08-01 04:40:35.266 [info] [command][56780472-ede0-49d3-9be1-ee5b5ba1ec1b] Process exited with code 0
2025-08-01 04:40:35.266 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8d76f2af-2c05-46ae-bbf6-875143383e44] socks connection closed
2025-08-01 04:40:35.266 [info] [command][56780472-ede0-49d3-9be1-ee5b5ba1ec1b] Socket close event received
2025-08-01 04:40:35.307 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51944 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:41:35.276 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:41:35.278 [info] [command][312383bf-dc0d-4180-99e5-b67cfd55a994] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""312383bf-dc0d-4180-99e5-b67cfd55a994""}
2025-08-01 04:41:35.279 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8cf35544-cf11-4bbe-a58b-2d4f8e216280] received connection request
2025-08-01 04:41:35.280 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:41:35.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8cf35544-cf11-4bbe-a58b-2d4f8e216280] socks forwarding established
2025-08-01 04:41:35.368 [info] [command][312383bf-dc0d-4180-99e5-b67cfd55a994] Process exited with code 0
2025-08-01 04:41:35.368 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8cf35544-cf11-4bbe-a58b-2d4f8e216280] socks connection closed
2025-08-01 04:41:35.368 [info] [command][312383bf-dc0d-4180-99e5-b67cfd55a994] Socket close event received
2025-08-01 04:41:35.407 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51964 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:42:35.371 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:42:35.372 [info] [command][b0850278-e2e2-4946-9191-3362342f98fc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b0850278-e2e2-4946-9191-3362342f98fc""}
2025-08-01 04:42:35.373 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c65ddad0-ad5e-49ba-be7c-7123cb627559] received connection request
2025-08-01 04:42:35.374 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:42:35.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c65ddad0-ad5e-49ba-be7c-7123cb627559] socks forwarding established
2025-08-01 04:42:35.467 [info] [command][b0850278-e2e2-4946-9191-3362342f98fc] Process exited with code 0
2025-08-01 04:42:35.467 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c65ddad0-ad5e-49ba-be7c-7123cb627559] socks connection closed
2025-08-01 04:42:35.467 [info] [command][b0850278-e2e2-4946-9191-3362342f98fc] Socket close event received
2025-08-01 04:42:35.526 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 51996 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:43:35.477 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:43:35.479 [info] [command][ce237072-9aaf-4608-8f34-7773dcce2fae] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ce237072-9aaf-4608-8f34-7773dcce2fae""}
2025-08-01 04:43:35.480 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][588d3f80-a747-4531-8185-3905e428c556] received connection request
2025-08-01 04:43:35.481 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:43:35.527 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][588d3f80-a747-4531-8185-3905e428c556] socks forwarding established
2025-08-01 04:43:35.572 [info] [command][ce237072-9aaf-4608-8f34-7773dcce2fae] Process exited with code 0
2025-08-01 04:43:35.572 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][588d3f80-a747-4531-8185-3905e428c556] socks connection closed
2025-08-01 04:43:35.573 [info] [command][ce237072-9aaf-4608-8f34-7773dcce2fae] Socket close event received
2025-08-01 04:43:35.612 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52029 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:44:35.575 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:44:35.576 [info] [command][4c15df02-12c5-48f4-99a2-2d388e1ebd3b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4c15df02-12c5-48f4-99a2-2d388e1ebd3b""}
2025-08-01 04:44:35.576 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][663f819e-0299-4677-8535-cbca5f569eb0] received connection request
2025-08-01 04:44:35.577 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:44:35.617 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][663f819e-0299-4677-8535-cbca5f569eb0] socks forwarding established
2025-08-01 04:44:35.663 [info] [command][4c15df02-12c5-48f4-99a2-2d388e1ebd3b] Process exited with code 0
2025-08-01 04:44:35.663 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][663f819e-0299-4677-8535-cbca5f569eb0] socks connection closed
2025-08-01 04:44:35.663 [info] [command][4c15df02-12c5-48f4-99a2-2d388e1ebd3b] Socket close event received
2025-08-01 04:44:35.708 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52043 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:45:35.670 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:45:35.673 [info] [command][351ce7fd-d1a5-4f41-a0d3-5aa9808d1ef0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""351ce7fd-d1a5-4f41-a0d3-5aa9808d1ef0""}
2025-08-01 04:45:35.673 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][35860032-4e58-408f-920e-e101ea66d2a9] received connection request
2025-08-01 04:45:35.674 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:45:35.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35860032-4e58-408f-920e-e101ea66d2a9] socks forwarding established
2025-08-01 04:45:35.763 [info] [command][351ce7fd-d1a5-4f41-a0d3-5aa9808d1ef0] Process exited with code 0
2025-08-01 04:45:35.764 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35860032-4e58-408f-920e-e101ea66d2a9] socks connection closed
2025-08-01 04:45:35.764 [info] [command][351ce7fd-d1a5-4f41-a0d3-5aa9808d1ef0] Socket close event received
2025-08-01 04:45:35.803 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52058 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:46:35.774 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:46:35.776 [info] [command][f0a0aa70-4568-4026-abca-548d3d57342c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f0a0aa70-4568-4026-abca-548d3d57342c""}
2025-08-01 04:46:35.777 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8ee6917c-2631-4ab5-9803-9a73ce65e46f] received connection request
2025-08-01 04:46:35.777 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:46:35.826 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ee6917c-2631-4ab5-9803-9a73ce65e46f] socks forwarding established
2025-08-01 04:46:35.873 [info] [command][f0a0aa70-4568-4026-abca-548d3d57342c] Process exited with code 0
2025-08-01 04:46:35.873 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8ee6917c-2631-4ab5-9803-9a73ce65e46f] socks connection closed
2025-08-01 04:46:35.874 [info] [command][f0a0aa70-4568-4026-abca-548d3d57342c] Socket close event received
2025-08-01 04:46:35.915 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52073 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:47:35.884 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:47:35.888 [info] [command][19668027-bc4e-455f-b93a-52a3e25a6235] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""19668027-bc4e-455f-b93a-52a3e25a6235""}
2025-08-01 04:47:35.888 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9995f2f7-614a-4efb-8efc-77d9914ffeed] received connection request
2025-08-01 04:47:35.889 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:47:35.931 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9995f2f7-614a-4efb-8efc-77d9914ffeed] socks forwarding established
2025-08-01 04:47:35.973 [info] [command][19668027-bc4e-455f-b93a-52a3e25a6235] Process exited with code 0
2025-08-01 04:47:35.973 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9995f2f7-614a-4efb-8efc-77d9914ffeed] socks connection closed
2025-08-01 04:47:35.974 [info] [command][19668027-bc4e-455f-b93a-52a3e25a6235] Socket close event received
2025-08-01 04:47:36.014 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52102 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:48:35.983 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:48:35.984 [info] [command][fc7e1b06-fcd1-4ec5-80a9-1949bda3f24d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fc7e1b06-fcd1-4ec5-80a9-1949bda3f24d""}
2025-08-01 04:48:35.984 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9fa486e8-898d-458a-9c8e-542f91cce995] received connection request
2025-08-01 04:48:35.985 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:48:36.032 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9fa486e8-898d-458a-9c8e-542f91cce995] socks forwarding established
2025-08-01 04:48:36.078 [info] [command][fc7e1b06-fcd1-4ec5-80a9-1949bda3f24d] Process exited with code 0
2025-08-01 04:48:36.078 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9fa486e8-898d-458a-9c8e-542f91cce995] socks connection closed
2025-08-01 04:48:36.078 [info] [command][fc7e1b06-fcd1-4ec5-80a9-1949bda3f24d] Socket close event received
2025-08-01 04:48:36.119 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52135 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:49:36.081 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:49:36.083 [info] [command][8510be3a-9450-41ac-a81d-62e6466395b1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8510be3a-9450-41ac-a81d-62e6466395b1""}
2025-08-01 04:49:36.083 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b0acc4b8-e90c-4181-a63b-54471056858c] received connection request
2025-08-01 04:49:36.084 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:49:36.126 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b0acc4b8-e90c-4181-a63b-54471056858c] socks forwarding established
2025-08-01 04:49:36.171 [info] [command][8510be3a-9450-41ac-a81d-62e6466395b1] Process exited with code 0
2025-08-01 04:49:36.171 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b0acc4b8-e90c-4181-a63b-54471056858c] socks connection closed
2025-08-01 04:49:36.171 [info] [command][8510be3a-9450-41ac-a81d-62e6466395b1] Socket close event received
2025-08-01 04:49:36.209 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52154 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:50:36.175 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:50:36.177 [info] [command][15b754c2-ea1e-4495-b4c5-e7df9ac197d9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""15b754c2-ea1e-4495-b4c5-e7df9ac197d9""}
2025-08-01 04:50:36.178 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][470e05fc-fddf-4330-b285-c87f33c42f95] received connection request
2025-08-01 04:50:36.179 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:50:36.223 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][470e05fc-fddf-4330-b285-c87f33c42f95] socks forwarding established
2025-08-01 04:50:36.268 [info] [command][15b754c2-ea1e-4495-b4c5-e7df9ac197d9] Process exited with code 0
2025-08-01 04:50:36.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][470e05fc-fddf-4330-b285-c87f33c42f95] socks connection closed
2025-08-01 04:50:36.269 [info] [command][15b754c2-ea1e-4495-b4c5-e7df9ac197d9] Socket close event received
2025-08-01 04:50:36.310 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52185 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:51:36.273 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:51:36.275 [info] [command][91bc4744-b350-45ce-bc71-b1cc72f2b5b3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""91bc4744-b350-45ce-bc71-b1cc72f2b5b3""}
2025-08-01 04:51:36.275 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f5cb0a0d-6903-4527-8de1-f659a64d3f97] received connection request
2025-08-01 04:51:36.276 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:51:36.337 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f5cb0a0d-6903-4527-8de1-f659a64d3f97] socks forwarding established
2025-08-01 04:51:36.404 [info] [command][91bc4744-b350-45ce-bc71-b1cc72f2b5b3] Process exited with code 0
2025-08-01 04:51:36.405 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f5cb0a0d-6903-4527-8de1-f659a64d3f97] socks connection closed
2025-08-01 04:51:36.405 [info] [command][91bc4744-b350-45ce-bc71-b1cc72f2b5b3] Socket close event received
2025-08-01 04:51:36.448 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52201 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:52:36.415 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:52:36.417 [info] [command][6f71e77c-e778-40bf-9d90-8d46a9bbf0b6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6f71e77c-e778-40bf-9d90-8d46a9bbf0b6""}
2025-08-01 04:52:36.417 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][711e2a83-0e68-43bb-9a57-9e001a5ba98a] received connection request
2025-08-01 04:52:36.418 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:52:36.462 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][711e2a83-0e68-43bb-9a57-9e001a5ba98a] socks forwarding established
2025-08-01 04:52:36.507 [info] [command][6f71e77c-e778-40bf-9d90-8d46a9bbf0b6] Process exited with code 0
2025-08-01 04:52:36.507 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][711e2a83-0e68-43bb-9a57-9e001a5ba98a] socks connection closed
2025-08-01 04:52:36.507 [info] [command][6f71e77c-e778-40bf-9d90-8d46a9bbf0b6] Socket close event received
2025-08-01 04:52:36.547 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52231 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:53:36.512 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:53:36.514 [info] [command][64639167-68fb-40aa-a42d-a5d500dcf6cf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""64639167-68fb-40aa-a42d-a5d500dcf6cf""}
2025-08-01 04:53:36.514 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][78087c6b-d937-4bef-be24-1e02ecce8499] received connection request
2025-08-01 04:53:36.514 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:53:36.559 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78087c6b-d937-4bef-be24-1e02ecce8499] socks forwarding established
2025-08-01 04:53:36.607 [info] [command][64639167-68fb-40aa-a42d-a5d500dcf6cf] Process exited with code 0
2025-08-01 04:53:36.607 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][78087c6b-d937-4bef-be24-1e02ecce8499] socks connection closed
2025-08-01 04:53:36.608 [info] [command][64639167-68fb-40aa-a42d-a5d500dcf6cf] Socket close event received
2025-08-01 04:53:36.648 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52266 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:54:36.609 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:54:36.609 [info] [command][eb345ce7-d9f5-41ea-af00-50e8ed1fb248] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eb345ce7-d9f5-41ea-af00-50e8ed1fb248""}
2025-08-01 04:54:36.610 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9df1397e-d1d1-45fa-8195-68d5995c55e8] received connection request
2025-08-01 04:54:36.610 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:54:36.652 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9df1397e-d1d1-45fa-8195-68d5995c55e8] socks forwarding established
2025-08-01 04:54:36.698 [info] [command][eb345ce7-d9f5-41ea-af00-50e8ed1fb248] Process exited with code 0
2025-08-01 04:54:36.698 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9df1397e-d1d1-45fa-8195-68d5995c55e8] socks connection closed
2025-08-01 04:54:36.698 [info] [command][eb345ce7-d9f5-41ea-af00-50e8ed1fb248] Socket close event received
2025-08-01 04:54:36.741 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52290 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:55:36.708 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:55:36.710 [info] [command][6f8e43d1-6522-47b0-8ec8-f5da9555d90c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6f8e43d1-6522-47b0-8ec8-f5da9555d90c""}
2025-08-01 04:55:36.711 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][94ab9e64-998e-48d3-981e-fd035656f92e] received connection request
2025-08-01 04:55:36.712 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:55:36.756 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][94ab9e64-998e-48d3-981e-fd035656f92e] socks forwarding established
2025-08-01 04:55:36.895 [info] [command][6f8e43d1-6522-47b0-8ec8-f5da9555d90c] Process exited with code 0
2025-08-01 04:55:36.895 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][94ab9e64-998e-48d3-981e-fd035656f92e] socks connection closed
2025-08-01 04:55:36.895 [info] [command][6f8e43d1-6522-47b0-8ec8-f5da9555d90c] Socket close event received
2025-08-01 04:55:36.934 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52305 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:56:36.897 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:56:36.898 [info] [command][67ff9617-4e07-40da-8d87-1439f616e206] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""67ff9617-4e07-40da-8d87-1439f616e206""}
2025-08-01 04:56:36.899 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][811c8691-8673-4f31-aa3f-a479a003d02b] received connection request
2025-08-01 04:56:36.900 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:56:37.137 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][811c8691-8673-4f31-aa3f-a479a003d02b] socks forwarding established
2025-08-01 04:56:37.180 [info] [command][67ff9617-4e07-40da-8d87-1439f616e206] Process exited with code 0
2025-08-01 04:56:37.180 [info] [command][67ff9617-4e07-40da-8d87-1439f616e206] Socket close event received
2025-08-01 04:56:37.180 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][811c8691-8673-4f31-aa3f-a479a003d02b] socks connection closed
2025-08-01 04:56:37.221 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52322 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:57:37.188 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:57:37.190 [info] [command][384da10a-bf0d-43de-a453-4133fc826cf0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""384da10a-bf0d-43de-a453-4133fc826cf0""}
2025-08-01 04:57:37.191 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d786cde1-741b-4c75-a27f-4ad18a74de73] received connection request
2025-08-01 04:57:37.192 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:57:37.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d786cde1-741b-4c75-a27f-4ad18a74de73] socks forwarding established
2025-08-01 04:57:37.282 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d786cde1-741b-4c75-a27f-4ad18a74de73] socks connection closed
2025-08-01 04:57:37.283 [info] [command][384da10a-bf0d-43de-a453-4133fc826cf0] Process exited with code 0
2025-08-01 04:57:37.283 [info] [command][384da10a-bf0d-43de-a453-4133fc826cf0] Socket close event received
2025-08-01 04:57:37.322 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52350 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:58:37.290 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:58:37.292 [info] [command][b373b7b6-b97e-4bc0-8af2-d7e956a3addc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b373b7b6-b97e-4bc0-8af2-d7e956a3addc""}
2025-08-01 04:58:37.292 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6052775b-3fbd-4d94-a109-fd514d7cf108] received connection request
2025-08-01 04:58:37.293 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:58:37.343 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6052775b-3fbd-4d94-a109-fd514d7cf108] socks forwarding established
2025-08-01 04:58:37.388 [info] [command][b373b7b6-b97e-4bc0-8af2-d7e956a3addc] Process exited with code 0
2025-08-01 04:58:37.388 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6052775b-3fbd-4d94-a109-fd514d7cf108] socks connection closed
2025-08-01 04:58:37.388 [info] [command][b373b7b6-b97e-4bc0-8af2-d7e956a3addc] Socket close event received
2025-08-01 04:58:37.429 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52387 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 04:59:37.398 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 04:59:37.400 [info] [command][625a7c6a-9893-4f2f-82cc-a9b352057c7b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""625a7c6a-9893-4f2f-82cc-a9b352057c7b""}
2025-08-01 04:59:37.400 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][58ab5b22-7ab8-49e3-a9de-64090e5609b8] received connection request
2025-08-01 04:59:37.401 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 04:59:37.401 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 04:59:37.443 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58ab5b22-7ab8-49e3-a9de-64090e5609b8] socks forwarding established
2025-08-01 04:59:37.488 [info] [command][625a7c6a-9893-4f2f-82cc-a9b352057c7b] Process exited with code 0
2025-08-01 04:59:37.488 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58ab5b22-7ab8-49e3-a9de-64090e5609b8] socks connection closed
2025-08-01 04:59:37.489 [info] [command][625a7c6a-9893-4f2f-82cc-a9b352057c7b] Socket close event received
2025-08-01 04:59:37.526 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52424 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:00:37.490 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:00:37.491 [info] [command][549f6657-1482-4f8f-b948-436770ee210b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""549f6657-1482-4f8f-b948-436770ee210b""}
2025-08-01 05:00:37.492 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][83560d48-e959-48fa-83f0-2762f00b7b8c] received connection request
2025-08-01 05:00:37.493 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:00:37.614 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][83560d48-e959-48fa-83f0-2762f00b7b8c] socks forwarding established
2025-08-01 05:00:37.659 [info] [command][549f6657-1482-4f8f-b948-436770ee210b] Process exited with code 0
2025-08-01 05:00:37.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][83560d48-e959-48fa-83f0-2762f00b7b8c] socks connection closed
2025-08-01 05:00:37.660 [info] [command][549f6657-1482-4f8f-b948-436770ee210b] Socket close event received
2025-08-01 05:00:37.699 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52435 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:01:37.669 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:01:37.672 [info] [command][5c4fdb94-2842-4d0e-8847-686c7b2f0d29] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5c4fdb94-2842-4d0e-8847-686c7b2f0d29""}
2025-08-01 05:01:37.673 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][423d7240-e273-4fc3-8ab3-df490d89f5a9] received connection request
2025-08-01 05:01:37.673 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:01:37.715 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][423d7240-e273-4fc3-8ab3-df490d89f5a9] socks forwarding established
2025-08-01 05:01:37.845 [info] [command][5c4fdb94-2842-4d0e-8847-686c7b2f0d29] Process exited with code 0
2025-08-01 05:01:37.845 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][423d7240-e273-4fc3-8ab3-df490d89f5a9] socks connection closed
2025-08-01 05:01:37.845 [info] [command][5c4fdb94-2842-4d0e-8847-686c7b2f0d29] Socket close event received
2025-08-01 05:01:37.912 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52449 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:02:37.855 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:02:37.857 [info] [command][8ed45284-7d14-4a7a-82af-fbeeb2943a41] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8ed45284-7d14-4a7a-82af-fbeeb2943a41""}
2025-08-01 05:02:37.858 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b73d4f36-bbac-4bac-a4d2-3cb8d19ddf35] received connection request
2025-08-01 05:02:37.859 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:02:37.901 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b73d4f36-bbac-4bac-a4d2-3cb8d19ddf35] socks forwarding established
2025-08-01 05:02:37.945 [info] [command][8ed45284-7d14-4a7a-82af-fbeeb2943a41] Process exited with code 0
2025-08-01 05:02:37.946 [info] [command][8ed45284-7d14-4a7a-82af-fbeeb2943a41] Socket close event received
2025-08-01 05:02:37.946 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b73d4f36-bbac-4bac-a4d2-3cb8d19ddf35] socks connection closed
2025-08-01 05:02:37.986 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52475 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:03:37.949 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:03:37.951 [info] [command][06df7cb5-c38a-4673-8aff-f04c6ee10b83] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""06df7cb5-c38a-4673-8aff-f04c6ee10b83""}
2025-08-01 05:03:37.951 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][43d0dc52-b721-4c46-b5dd-ae641eacdeac] received connection request
2025-08-01 05:03:37.952 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:03:37.993 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][43d0dc52-b721-4c46-b5dd-ae641eacdeac] socks forwarding established
2025-08-01 05:03:38.036 [info] [command][06df7cb5-c38a-4673-8aff-f04c6ee10b83] Process exited with code 0
2025-08-01 05:03:38.036 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][43d0dc52-b721-4c46-b5dd-ae641eacdeac] socks connection closed
2025-08-01 05:03:38.036 [info] [command][06df7cb5-c38a-4673-8aff-f04c6ee10b83] Socket close event received
2025-08-01 05:03:38.076 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52503 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:04:38.046 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:04:38.047 [info] [command][9b6a55f0-4adf-4309-9a2d-082e0455825c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9b6a55f0-4adf-4309-9a2d-082e0455825c""}
2025-08-01 05:04:38.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3544434b-0ae2-4a42-8314-854e213e651f] received connection request
2025-08-01 05:04:38.048 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:04:38.090 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3544434b-0ae2-4a42-8314-854e213e651f] socks forwarding established
2025-08-01 05:04:38.140 [info] [command][9b6a55f0-4adf-4309-9a2d-082e0455825c] Process exited with code 0
2025-08-01 05:04:38.140 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3544434b-0ae2-4a42-8314-854e213e651f] socks connection closed
2025-08-01 05:04:38.140 [info] [command][9b6a55f0-4adf-4309-9a2d-082e0455825c] Socket close event received
2025-08-01 05:04:38.179 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52518 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:05:38.144 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:05:38.146 [info] [command][e38d5f29-000c-4929-bb1d-6ff0d0f001b0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e38d5f29-000c-4929-bb1d-6ff0d0f001b0""}
2025-08-01 05:05:38.147 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e4004446-a36e-448d-acd2-d4b116a0e766] received connection request
2025-08-01 05:05:38.147 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:05:38.194 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4004446-a36e-448d-acd2-d4b116a0e766] socks forwarding established
2025-08-01 05:05:38.240 [info] [command][e38d5f29-000c-4929-bb1d-6ff0d0f001b0] Process exited with code 0
2025-08-01 05:05:38.240 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4004446-a36e-448d-acd2-d4b116a0e766] socks connection closed
2025-08-01 05:05:38.240 [info] [command][e38d5f29-000c-4929-bb1d-6ff0d0f001b0] Socket close event received
2025-08-01 05:05:38.301 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52528 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:06:38.250 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:06:38.252 [info] [command][cf624e41-1db3-4c20-bfd3-92faa3a2c38f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cf624e41-1db3-4c20-bfd3-92faa3a2c38f""}
2025-08-01 05:06:38.252 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e278f3c3-1083-4359-90a0-9f482aa3efab] received connection request
2025-08-01 05:06:38.253 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:06:38.298 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e278f3c3-1083-4359-90a0-9f482aa3efab] socks forwarding established
2025-08-01 05:06:38.343 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e278f3c3-1083-4359-90a0-9f482aa3efab] socks connection closed
2025-08-01 05:06:38.344 [info] [command][cf624e41-1db3-4c20-bfd3-92faa3a2c38f] Process exited with code 0
2025-08-01 05:06:38.344 [info] [command][cf624e41-1db3-4c20-bfd3-92faa3a2c38f] Socket close event received
2025-08-01 05:06:38.387 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52546 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:07:38.348 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:07:38.350 [info] [command][6763ea22-31c1-4c56-b7d0-5412a2380574] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6763ea22-31c1-4c56-b7d0-5412a2380574""}
2025-08-01 05:07:38.350 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d05537d9-d482-4c9a-930b-df11cb5e0df5] received connection request
2025-08-01 05:07:38.351 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:07:38.398 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d05537d9-d482-4c9a-930b-df11cb5e0df5] socks forwarding established
2025-08-01 05:07:38.443 [info] [command][6763ea22-31c1-4c56-b7d0-5412a2380574] Process exited with code 0
2025-08-01 05:07:38.443 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d05537d9-d482-4c9a-930b-df11cb5e0df5] socks connection closed
2025-08-01 05:07:38.444 [info] [command][6763ea22-31c1-4c56-b7d0-5412a2380574] Socket close event received
2025-08-01 05:07:38.485 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52579 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:08:38.452 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:08:38.454 [info] [command][13815b9a-5a83-44c5-9fba-e763ff511717] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""13815b9a-5a83-44c5-9fba-e763ff511717""}
2025-08-01 05:08:38.455 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6141434a-63b6-455a-a17b-4fa0b08965ea] received connection request
2025-08-01 05:08:38.455 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:08:38.508 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6141434a-63b6-455a-a17b-4fa0b08965ea] socks forwarding established
2025-08-01 05:08:38.552 [info] [command][13815b9a-5a83-44c5-9fba-e763ff511717] Process exited with code 0
2025-08-01 05:08:38.552 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6141434a-63b6-455a-a17b-4fa0b08965ea] socks connection closed
2025-08-01 05:08:38.552 [info] [command][13815b9a-5a83-44c5-9fba-e763ff511717] Socket close event received
2025-08-01 05:08:38.591 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52615 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:09:38.562 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:09:38.563 [info] [command][e29deaba-6beb-4df2-b0c0-95ce677e36a1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e29deaba-6beb-4df2-b0c0-95ce677e36a1""}
2025-08-01 05:09:38.564 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0fca61fa-2ed7-4245-a247-83f255b8c0e4] received connection request
2025-08-01 05:09:38.565 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:09:38.613 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0fca61fa-2ed7-4245-a247-83f255b8c0e4] socks forwarding established
2025-08-01 05:09:38.657 [info] [command][e29deaba-6beb-4df2-b0c0-95ce677e36a1] Process exited with code 0
2025-08-01 05:09:38.657 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0fca61fa-2ed7-4245-a247-83f255b8c0e4] socks connection closed
2025-08-01 05:09:38.657 [info] [command][e29deaba-6beb-4df2-b0c0-95ce677e36a1] Socket close event received
2025-08-01 05:09:38.695 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52631 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:10:38.667 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:10:38.669 [info] [command][40cb3b19-147a-42d7-81e1-d9a7036e1a56] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""40cb3b19-147a-42d7-81e1-d9a7036e1a56""}
2025-08-01 05:10:38.670 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6807f352-af88-4a41-82a3-4dc16539401f] received connection request
2025-08-01 05:10:38.670 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:10:38.773 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6807f352-af88-4a41-82a3-4dc16539401f] socks forwarding established
2025-08-01 05:10:38.819 [info] [command][40cb3b19-147a-42d7-81e1-d9a7036e1a56] Process exited with code 0
2025-08-01 05:10:38.820 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6807f352-af88-4a41-82a3-4dc16539401f] socks connection closed
2025-08-01 05:10:38.820 [info] [command][40cb3b19-147a-42d7-81e1-d9a7036e1a56] Socket close event received
2025-08-01 05:10:38.884 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52646 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:11:38.821 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:11:38.823 [info] [command][a06cc21c-596b-4947-9990-544150e9f3c1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a06cc21c-596b-4947-9990-544150e9f3c1""}
2025-08-01 05:11:38.823 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1716ddac-d2cc-43c5-9a5a-b8966349d320] received connection request
2025-08-01 05:11:38.824 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:11:38.870 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1716ddac-d2cc-43c5-9a5a-b8966349d320] socks forwarding established
2025-08-01 05:11:38.916 [info] [command][a06cc21c-596b-4947-9990-544150e9f3c1] Process exited with code 0
2025-08-01 05:11:38.916 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1716ddac-d2cc-43c5-9a5a-b8966349d320] socks connection closed
2025-08-01 05:11:38.916 [info] [command][a06cc21c-596b-4947-9990-544150e9f3c1] Socket close event received
2025-08-01 05:11:38.956 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52666 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:12:38.917 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:12:38.918 [info] [command][34a21c62-82d4-4e09-98e4-00a3fb7ced49] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""34a21c62-82d4-4e09-98e4-00a3fb7ced49""}
2025-08-01 05:12:38.919 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][622320f5-20be-4117-b4d2-3024caf5f364] received connection request
2025-08-01 05:12:38.919 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:12:38.968 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][622320f5-20be-4117-b4d2-3024caf5f364] socks forwarding established
2025-08-01 05:12:39.013 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][622320f5-20be-4117-b4d2-3024caf5f364] socks connection closed
2025-08-01 05:12:39.014 [info] [command][34a21c62-82d4-4e09-98e4-00a3fb7ced49] Process exited with code 0
2025-08-01 05:12:39.014 [info] [command][34a21c62-82d4-4e09-98e4-00a3fb7ced49] Socket close event received
2025-08-01 05:12:39.054 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52694 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:13:39.025 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:13:39.026 [info] [command][38ff3dc4-b518-4ef3-95ca-059874da87df] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""38ff3dc4-b518-4ef3-95ca-059874da87df""}
2025-08-01 05:13:39.027 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8cdb0c9d-9b03-4768-973d-65787e7b56c8] received connection request
2025-08-01 05:13:39.028 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:13:39.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8cdb0c9d-9b03-4768-973d-65787e7b56c8] socks forwarding established
2025-08-01 05:13:39.111 [info] [command][38ff3dc4-b518-4ef3-95ca-059874da87df] Process exited with code 0
2025-08-01 05:13:39.111 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8cdb0c9d-9b03-4768-973d-65787e7b56c8] socks connection closed
2025-08-01 05:13:39.111 [info] [command][38ff3dc4-b518-4ef3-95ca-059874da87df] Socket close event received
2025-08-01 05:13:39.151 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52739 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:14:39.118 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:14:39.119 [info] [command][be235913-6e0e-4632-838c-1f72bcf67783] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""be235913-6e0e-4632-838c-1f72bcf67783""}
2025-08-01 05:14:39.120 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e60852a0-9c30-4e57-af36-ae99c33c63d8] received connection request
2025-08-01 05:14:39.120 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:14:39.201 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e60852a0-9c30-4e57-af36-ae99c33c63d8] socks forwarding established
2025-08-01 05:14:39.248 [info] [command][be235913-6e0e-4632-838c-1f72bcf67783] Process exited with code 0
2025-08-01 05:14:39.248 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e60852a0-9c30-4e57-af36-ae99c33c63d8] socks connection closed
2025-08-01 05:14:39.248 [info] [command][be235913-6e0e-4632-838c-1f72bcf67783] Socket close event received
2025-08-01 05:14:39.290 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52754 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:15:39.258 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:15:39.260 [info] [command][c904c16d-c545-4894-9084-841bce26b98a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c904c16d-c545-4894-9084-841bce26b98a""}
2025-08-01 05:15:39.261 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][13824036-c953-4ebf-80c2-b83e53b8b6b8] received connection request
2025-08-01 05:15:39.261 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:15:39.434 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13824036-c953-4ebf-80c2-b83e53b8b6b8] socks forwarding established
2025-08-01 05:15:39.480 [info] [command][c904c16d-c545-4894-9084-841bce26b98a] Process exited with code 0
2025-08-01 05:15:39.480 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13824036-c953-4ebf-80c2-b83e53b8b6b8] socks connection closed
2025-08-01 05:15:39.480 [info] [command][c904c16d-c545-4894-9084-841bce26b98a] Socket close event received
2025-08-01 05:15:39.540 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52768 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:16:39.481 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:16:39.482 [info] [command][c65ccf6b-3f73-41c3-9269-d028652b351e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c65ccf6b-3f73-41c3-9269-d028652b351e""}
2025-08-01 05:16:39.483 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f3f64b6b-7169-484a-aade-397a7472b8a5] received connection request
2025-08-01 05:16:39.484 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:16:39.613 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f3f64b6b-7169-484a-aade-397a7472b8a5] socks forwarding established
2025-08-01 05:16:39.659 [info] [command][c65ccf6b-3f73-41c3-9269-d028652b351e] Process exited with code 0
2025-08-01 05:16:39.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f3f64b6b-7169-484a-aade-397a7472b8a5] socks connection closed
2025-08-01 05:16:39.660 [info] [command][c65ccf6b-3f73-41c3-9269-d028652b351e] Socket close event received
2025-08-01 05:16:39.701 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52784 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:17:39.671 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:17:39.673 [info] [command][6b8ec798-7c72-4a1c-826a-1aa50080c892] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6b8ec798-7c72-4a1c-826a-1aa50080c892""}
2025-08-01 05:17:39.673 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f61185e2-775d-4d97-86fc-5ca985fecdba] received connection request
2025-08-01 05:17:39.674 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:17:39.726 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f61185e2-775d-4d97-86fc-5ca985fecdba] socks forwarding established
2025-08-01 05:17:39.771 [info] [command][6b8ec798-7c72-4a1c-826a-1aa50080c892] Process exited with code 0
2025-08-01 05:17:39.771 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f61185e2-775d-4d97-86fc-5ca985fecdba] socks connection closed
2025-08-01 05:17:39.771 [info] [command][6b8ec798-7c72-4a1c-826a-1aa50080c892] Socket close event received
2025-08-01 05:17:39.811 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52819 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:18:39.779 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:18:39.781 [info] [command][e8c9ed72-3282-4bbb-a33e-494813bdeae8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e8c9ed72-3282-4bbb-a33e-494813bdeae8""}
2025-08-01 05:18:39.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ebfb7d59-4433-4e50-8276-6e86f7dfbd77] received connection request
2025-08-01 05:18:39.783 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:18:39.829 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ebfb7d59-4433-4e50-8276-6e86f7dfbd77] socks forwarding established
2025-08-01 05:18:39.874 [info] [command][e8c9ed72-3282-4bbb-a33e-494813bdeae8] Process exited with code 0
2025-08-01 05:18:39.874 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ebfb7d59-4433-4e50-8276-6e86f7dfbd77] socks connection closed
2025-08-01 05:18:39.874 [info] [command][e8c9ed72-3282-4bbb-a33e-494813bdeae8] Socket close event received
2025-08-01 05:18:39.918 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52847 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:19:39.885 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:19:39.887 [info] [command][b3679eaa-30c6-4c90-bd64-0532993661d8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b3679eaa-30c6-4c90-bd64-0532993661d8""}
2025-08-01 05:19:39.888 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d6e7d7cd-a5c4-48f1-91e9-eb5f005e70a6] received connection request
2025-08-01 05:19:39.889 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:19:39.943 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d6e7d7cd-a5c4-48f1-91e9-eb5f005e70a6] socks forwarding established
2025-08-01 05:19:39.994 [info] [command][b3679eaa-30c6-4c90-bd64-0532993661d8] Process exited with code 0
2025-08-01 05:19:39.994 [info] [command][b3679eaa-30c6-4c90-bd64-0532993661d8] Socket close event received
2025-08-01 05:19:40.033 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d6e7d7cd-a5c4-48f1-91e9-eb5f005e70a6] socks connection closed
2025-08-01 05:19:40.034 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52870 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:20:39.996 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:20:39.998 [info] [command][89b2eeaa-c5db-4bcf-84a2-43fa25f51de5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""89b2eeaa-c5db-4bcf-84a2-43fa25f51de5""}
2025-08-01 05:20:39.999 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1c961a73-0406-42bf-8cf7-353584921fb9] received connection request
2025-08-01 05:20:39.999 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:20:40.039 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c961a73-0406-42bf-8cf7-353584921fb9] socks forwarding established
2025-08-01 05:20:40.083 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c961a73-0406-42bf-8cf7-353584921fb9] socks connection closed
2025-08-01 05:20:40.083 [info] [command][89b2eeaa-c5db-4bcf-84a2-43fa25f51de5] Process exited with code 0
2025-08-01 05:20:40.083 [info] [command][89b2eeaa-c5db-4bcf-84a2-43fa25f51de5] Socket close event received
2025-08-01 05:20:40.124 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52899 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:21:40.088 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:21:40.089 [info] [command][26adaee6-4e45-4282-a9ba-d82ea83752a9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""26adaee6-4e45-4282-a9ba-d82ea83752a9""}
2025-08-01 05:21:40.089 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c6c27386-4e0a-4234-839e-e9c57c515d2b] received connection request
2025-08-01 05:21:40.090 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:21:40.135 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c6c27386-4e0a-4234-839e-e9c57c515d2b] socks forwarding established
2025-08-01 05:21:40.179 [info] [command][26adaee6-4e45-4282-a9ba-d82ea83752a9] Process exited with code 0
2025-08-01 05:21:40.180 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c6c27386-4e0a-4234-839e-e9c57c515d2b] socks connection closed
2025-08-01 05:21:40.180 [info] [command][26adaee6-4e45-4282-a9ba-d82ea83752a9] Socket close event received
2025-08-01 05:21:40.221 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52934 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:22:40.186 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:22:40.187 [info] [command][aaaaa437-ba80-42d4-bab7-519d1a3b6a21] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""aaaaa437-ba80-42d4-bab7-519d1a3b6a21""}
2025-08-01 05:22:40.188 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bc052247-f68f-42e9-97f2-da9409527051] received connection request
2025-08-01 05:22:40.189 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:22:40.233 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bc052247-f68f-42e9-97f2-da9409527051] socks forwarding established
2025-08-01 05:22:40.279 [info] [command][aaaaa437-ba80-42d4-bab7-519d1a3b6a21] Process exited with code 0
2025-08-01 05:22:40.279 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bc052247-f68f-42e9-97f2-da9409527051] socks connection closed
2025-08-01 05:22:40.279 [info] [command][aaaaa437-ba80-42d4-bab7-519d1a3b6a21] Socket close event received
2025-08-01 05:22:40.321 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52963 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:23:40.289 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:23:40.290 [info] [command][355c5a20-c838-4b98-8b29-8a7753dba2c2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""355c5a20-c838-4b98-8b29-8a7753dba2c2""}
2025-08-01 05:23:40.291 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][db3bbd37-5811-4b7e-920d-f7ee30c04b96] received connection request
2025-08-01 05:23:40.292 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:23:40.334 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db3bbd37-5811-4b7e-920d-f7ee30c04b96] socks forwarding established
2025-08-01 05:23:40.379 [info] [command][355c5a20-c838-4b98-8b29-8a7753dba2c2] Process exited with code 0
2025-08-01 05:23:40.380 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db3bbd37-5811-4b7e-920d-f7ee30c04b96] socks connection closed
2025-08-01 05:23:40.380 [info] [command][355c5a20-c838-4b98-8b29-8a7753dba2c2] Socket close event received
2025-08-01 05:23:40.420 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 52994 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:24:40.389 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:24:40.391 [info] [command][0775c49d-0faa-4a53-8c6b-c485ea6c705b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0775c49d-0faa-4a53-8c6b-c485ea6c705b""}
2025-08-01 05:24:40.391 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e05f21e9-69da-4e6e-b312-dd6bb7bf65bb] received connection request
2025-08-01 05:24:40.392 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:24:40.487 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e05f21e9-69da-4e6e-b312-dd6bb7bf65bb] socks forwarding established
2025-08-01 05:24:40.535 [info] [command][0775c49d-0faa-4a53-8c6b-c485ea6c705b] Process exited with code 0
2025-08-01 05:24:40.535 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e05f21e9-69da-4e6e-b312-dd6bb7bf65bb] socks connection closed
2025-08-01 05:24:40.535 [info] [command][0775c49d-0faa-4a53-8c6b-c485ea6c705b] Socket close event received
2025-08-01 05:24:40.578 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53005 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:25:40.544 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:25:40.545 [info] [command][bd0d35c1-7a6f-4629-9ed8-4e4491e1d0b1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bd0d35c1-7a6f-4629-9ed8-4e4491e1d0b1""}
2025-08-01 05:25:40.546 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][75fef7c3-d690-4974-a6ad-e66511efc3d3] received connection request
2025-08-01 05:25:40.546 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:25:40.590 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][75fef7c3-d690-4974-a6ad-e66511efc3d3] socks forwarding established
2025-08-01 05:25:40.636 [info] [command][bd0d35c1-7a6f-4629-9ed8-4e4491e1d0b1] Process exited with code 0
2025-08-01 05:25:40.636 [info] [command][bd0d35c1-7a6f-4629-9ed8-4e4491e1d0b1] Socket close event received
2025-08-01 05:25:40.637 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][75fef7c3-d690-4974-a6ad-e66511efc3d3] socks connection closed
2025-08-01 05:25:40.680 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53021 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:26:40.639 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:26:40.640 [info] [command][96099374-192b-47e6-9504-51b36112ef5b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""96099374-192b-47e6-9504-51b36112ef5b""}
2025-08-01 05:26:40.641 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0f197b6a-65f7-4c1f-a18c-143746061fb8] received connection request
2025-08-01 05:26:40.642 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:26:40.683 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0f197b6a-65f7-4c1f-a18c-143746061fb8] socks forwarding established
2025-08-01 05:26:40.727 [info] [command][96099374-192b-47e6-9504-51b36112ef5b] Process exited with code 0
2025-08-01 05:26:40.728 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0f197b6a-65f7-4c1f-a18c-143746061fb8] socks connection closed
2025-08-01 05:26:40.728 [info] [command][96099374-192b-47e6-9504-51b36112ef5b] Socket close event received
2025-08-01 05:26:40.768 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53041 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:27:40.730 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:27:40.731 [info] [command][7b88c9e6-586f-4be3-9864-819b935d7f1e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7b88c9e6-586f-4be3-9864-819b935d7f1e""}
2025-08-01 05:27:40.732 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ea076719-1025-4dc3-a7f3-967aacb1bc88] received connection request
2025-08-01 05:27:40.733 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:27:40.776 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ea076719-1025-4dc3-a7f3-967aacb1bc88] socks forwarding established
2025-08-01 05:27:40.823 [info] [command][7b88c9e6-586f-4be3-9864-819b935d7f1e] Process exited with code 0
2025-08-01 05:27:40.823 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ea076719-1025-4dc3-a7f3-967aacb1bc88] socks connection closed
2025-08-01 05:27:40.823 [info] [command][7b88c9e6-586f-4be3-9864-819b935d7f1e] Socket close event received
2025-08-01 05:27:40.863 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53072 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:28:40.826 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:28:40.827 [info] [command][0446011e-6d4d-4b80-8078-97b5c3f1dfd4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0446011e-6d4d-4b80-8078-97b5c3f1dfd4""}
2025-08-01 05:28:40.828 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6c48d54c-e8b6-43ad-adee-0645c8593379] received connection request
2025-08-01 05:28:40.828 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:28:40.877 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c48d54c-e8b6-43ad-adee-0645c8593379] socks forwarding established
2025-08-01 05:28:40.922 [info] [command][0446011e-6d4d-4b80-8078-97b5c3f1dfd4] Process exited with code 0
2025-08-01 05:28:40.922 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c48d54c-e8b6-43ad-adee-0645c8593379] socks connection closed
2025-08-01 05:28:40.922 [info] [command][0446011e-6d4d-4b80-8078-97b5c3f1dfd4] Socket close event received
2025-08-01 05:28:40.963 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53105 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:29:40.924 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:29:40.925 [info] [command][866ef53f-8d9e-43c8-9d84-fa199ca7d782] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""866ef53f-8d9e-43c8-9d84-fa199ca7d782""}
2025-08-01 05:29:40.925 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c35dfa40-4687-4c59-87d7-ddf7c241e4b4] received connection request
2025-08-01 05:29:40.925 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:29:40.969 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c35dfa40-4687-4c59-87d7-ddf7c241e4b4] socks forwarding established
2025-08-01 05:29:41.015 [info] [command][866ef53f-8d9e-43c8-9d84-fa199ca7d782] Process exited with code 0
2025-08-01 05:29:41.016 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c35dfa40-4687-4c59-87d7-ddf7c241e4b4] socks connection closed
2025-08-01 05:29:41.016 [info] [command][866ef53f-8d9e-43c8-9d84-fa199ca7d782] Socket close event received
2025-08-01 05:29:41.087 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53120 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:30:41.026 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:30:41.026 [info] [command][94835c6d-2c46-428d-86c4-1fc5a83c6a5e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""94835c6d-2c46-428d-86c4-1fc5a83c6a5e""}
2025-08-01 05:30:41.027 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][93efc972-bd7d-4653-8226-5be4dc9d5511] received connection request
2025-08-01 05:30:41.027 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:30:41.068 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93efc972-bd7d-4653-8226-5be4dc9d5511] socks forwarding established
2025-08-01 05:30:41.113 [info] [command][94835c6d-2c46-428d-86c4-1fc5a83c6a5e] Process exited with code 0
2025-08-01 05:30:41.113 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93efc972-bd7d-4653-8226-5be4dc9d5511] socks connection closed
2025-08-01 05:30:41.114 [info] [command][94835c6d-2c46-428d-86c4-1fc5a83c6a5e] Socket close event received
2025-08-01 05:30:41.153 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53139 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:31:41.116 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:31:41.117 [info] [command][1d68d735-9e31-485a-916c-7e745ea72979] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1d68d735-9e31-485a-916c-7e745ea72979""}
2025-08-01 05:31:41.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e412d6b0-afba-4470-9669-53790edd5101] received connection request
2025-08-01 05:31:41.119 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:31:41.229 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e412d6b0-afba-4470-9669-53790edd5101] socks forwarding established
2025-08-01 05:31:41.275 [info] [command][1d68d735-9e31-485a-916c-7e745ea72979] Process exited with code 0
2025-08-01 05:31:41.275 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e412d6b0-afba-4470-9669-53790edd5101] socks connection closed
2025-08-01 05:31:41.275 [info] [command][1d68d735-9e31-485a-916c-7e745ea72979] Socket close event received
2025-08-01 05:31:41.316 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53159 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:32:41.275 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:32:41.277 [info] [command][a05dcd40-3f71-43ba-a28a-fb735e581341] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a05dcd40-3f71-43ba-a28a-fb735e581341""}
2025-08-01 05:32:41.277 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5eddbfd4-2f72-4644-9802-ad9ff8c6eba4] received connection request
2025-08-01 05:32:41.278 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:32:41.327 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5eddbfd4-2f72-4644-9802-ad9ff8c6eba4] socks forwarding established
2025-08-01 05:32:41.374 [info] [command][a05dcd40-3f71-43ba-a28a-fb735e581341] Process exited with code 0
2025-08-01 05:32:41.374 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5eddbfd4-2f72-4644-9802-ad9ff8c6eba4] socks connection closed
2025-08-01 05:32:41.374 [info] [command][a05dcd40-3f71-43ba-a28a-fb735e581341] Socket close event received
2025-08-01 05:32:41.414 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53188 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:33:41.382 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:33:41.383 [info] [command][b8738cd7-9920-4204-8943-ca7126e493e8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b8738cd7-9920-4204-8943-ca7126e493e8""}
2025-08-01 05:33:41.383 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][651f45a1-887d-47f8-ac6f-723791cc2e07] received connection request
2025-08-01 05:33:41.384 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:33:41.425 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][651f45a1-887d-47f8-ac6f-723791cc2e07] socks forwarding established
2025-08-01 05:33:41.471 [info] [command][b8738cd7-9920-4204-8943-ca7126e493e8] Process exited with code 0
2025-08-01 05:33:41.471 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][651f45a1-887d-47f8-ac6f-723791cc2e07] socks connection closed
2025-08-01 05:33:41.471 [info] [command][b8738cd7-9920-4204-8943-ca7126e493e8] Socket close event received
2025-08-01 05:33:41.513 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53222 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:34:41.471 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:34:41.473 [info] [command][65bf9c41-4a45-401f-aec0-5d0b58860f07] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""65bf9c41-4a45-401f-aec0-5d0b58860f07""}
2025-08-01 05:34:41.474 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a2618d53-cca2-48dd-bad9-27e6ef077b6a] received connection request
2025-08-01 05:34:41.474 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:34:41.519 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a2618d53-cca2-48dd-bad9-27e6ef077b6a] socks forwarding established
2025-08-01 05:34:41.567 [info] [command][65bf9c41-4a45-401f-aec0-5d0b58860f07] Process exited with code 0
2025-08-01 05:34:41.567 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a2618d53-cca2-48dd-bad9-27e6ef077b6a] socks connection closed
2025-08-01 05:34:41.567 [info] [command][65bf9c41-4a45-401f-aec0-5d0b58860f07] Socket close event received
2025-08-01 05:34:41.624 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53238 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:35:41.575 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:35:41.576 [info] [command][836fd0cb-c7aa-4209-9da3-c7c8648a3d61] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""836fd0cb-c7aa-4209-9da3-c7c8648a3d61""}
2025-08-01 05:35:41.577 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][15994b84-0173-45c3-8dfe-49cde2a69d2a] received connection request
2025-08-01 05:35:41.578 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:35:41.626 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][15994b84-0173-45c3-8dfe-49cde2a69d2a] socks forwarding established
2025-08-01 05:35:41.671 [info] [command][836fd0cb-c7aa-4209-9da3-c7c8648a3d61] Process exited with code 0
2025-08-01 05:35:41.671 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][15994b84-0173-45c3-8dfe-49cde2a69d2a] socks connection closed
2025-08-01 05:35:41.671 [info] [command][836fd0cb-c7aa-4209-9da3-c7c8648a3d61] Socket close event received
2025-08-01 05:35:41.713 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53253 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:36:41.681 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:36:41.682 [info] [command][1adbcf7d-7e10-4b1c-bc7f-56ff54d5a69e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1adbcf7d-7e10-4b1c-bc7f-56ff54d5a69e""}
2025-08-01 05:36:41.683 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][277787f8-0263-4d13-9bf5-9161e0433e8f] received connection request
2025-08-01 05:36:41.684 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:36:41.725 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][277787f8-0263-4d13-9bf5-9161e0433e8f] socks forwarding established
2025-08-01 05:36:41.769 [info] [command][1adbcf7d-7e10-4b1c-bc7f-56ff54d5a69e] Process exited with code 0
2025-08-01 05:36:41.770 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][277787f8-0263-4d13-9bf5-9161e0433e8f] socks connection closed
2025-08-01 05:36:41.770 [info] [command][1adbcf7d-7e10-4b1c-bc7f-56ff54d5a69e] Socket close event received
2025-08-01 05:36:41.810 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53267 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:37:41.771 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:37:41.772 [info] [command][6252fd74-17eb-4d46-a9d1-97aa37031448] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6252fd74-17eb-4d46-a9d1-97aa37031448""}
2025-08-01 05:37:41.773 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e3333382-bcd3-4db8-ae07-3f37748162e0] received connection request
2025-08-01 05:37:41.773 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:37:41.814 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e3333382-bcd3-4db8-ae07-3f37748162e0] socks forwarding established
2025-08-01 05:37:41.862 [info] [command][6252fd74-17eb-4d46-a9d1-97aa37031448] Process exited with code 0
2025-08-01 05:37:41.862 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e3333382-bcd3-4db8-ae07-3f37748162e0] socks connection closed
2025-08-01 05:37:41.862 [info] [command][6252fd74-17eb-4d46-a9d1-97aa37031448] Socket close event received
2025-08-01 05:37:41.903 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53308 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:38:41.869 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:38:41.870 [info] [command][ff3f19b8-74ed-49da-bbb4-f00c705a8690] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ff3f19b8-74ed-49da-bbb4-f00c705a8690""}
2025-08-01 05:38:41.871 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][239e45f6-0318-4b62-84bc-f02fe3360153] received connection request
2025-08-01 05:38:41.872 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:38:42.160 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][239e45f6-0318-4b62-84bc-f02fe3360153] socks forwarding established
2025-08-01 05:38:42.205 [info] [command][ff3f19b8-74ed-49da-bbb4-f00c705a8690] Process exited with code 0
2025-08-01 05:38:42.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][239e45f6-0318-4b62-84bc-f02fe3360153] socks connection closed
2025-08-01 05:38:42.205 [info] [command][ff3f19b8-74ed-49da-bbb4-f00c705a8690] Socket close event received
2025-08-01 05:38:42.244 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53342 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:39:42.214 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:39:42.216 [info] [command][31836cec-d85b-4b13-a4e4-150a40409ef0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""31836cec-d85b-4b13-a4e4-150a40409ef0""}
2025-08-01 05:39:42.217 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][60342d8c-f1c0-4e4f-8ef4-07a87145cb76] received connection request
2025-08-01 05:39:42.217 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:39:42.260 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][60342d8c-f1c0-4e4f-8ef4-07a87145cb76] socks forwarding established
2025-08-01 05:39:42.306 [info] [command][31836cec-d85b-4b13-a4e4-150a40409ef0] Process exited with code 0
2025-08-01 05:39:42.306 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][60342d8c-f1c0-4e4f-8ef4-07a87145cb76] socks connection closed
2025-08-01 05:39:42.306 [info] [command][31836cec-d85b-4b13-a4e4-150a40409ef0] Socket close event received
2025-08-01 05:39:42.345 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53360 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:40:42.312 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:40:42.314 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][93250b5b-294c-4d16-b1fa-a04728d80842] received connection request
2025-08-01 05:40:42.314 [info] [command][b92cd621-3744-46a4-b0df-6216490fbe65] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b92cd621-3744-46a4-b0df-6216490fbe65""}
2025-08-01 05:40:42.314 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:40:42.358 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93250b5b-294c-4d16-b1fa-a04728d80842] socks forwarding established
2025-08-01 05:40:42.402 [info] [command][b92cd621-3744-46a4-b0df-6216490fbe65] Process exited with code 0
2025-08-01 05:40:42.403 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][93250b5b-294c-4d16-b1fa-a04728d80842] socks connection closed
2025-08-01 05:40:42.403 [info] [command][b92cd621-3744-46a4-b0df-6216490fbe65] Socket close event received
2025-08-01 05:40:42.443 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53379 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:41:42.409 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:41:42.411 [info] [command][eefd447a-3470-48a7-acde-8d4a9e2e3a2c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eefd447a-3470-48a7-acde-8d4a9e2e3a2c""}
2025-08-01 05:41:42.411 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2624f6c0-9e08-4cbf-927c-51d2d2b3bb70] received connection request
2025-08-01 05:41:42.412 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:41:42.456 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2624f6c0-9e08-4cbf-927c-51d2d2b3bb70] socks forwarding established
2025-08-01 05:41:42.502 [info] [command][eefd447a-3470-48a7-acde-8d4a9e2e3a2c] Process exited with code 0
2025-08-01 05:41:42.502 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2624f6c0-9e08-4cbf-927c-51d2d2b3bb70] socks connection closed
2025-08-01 05:41:42.503 [info] [command][eefd447a-3470-48a7-acde-8d4a9e2e3a2c] Socket close event received
2025-08-01 05:41:42.544 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53394 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:42:42.507 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:42:42.509 [info] [command][e9a620b5-0e04-476b-bba9-2b8051bf12f8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e9a620b5-0e04-476b-bba9-2b8051bf12f8""}
2025-08-01 05:42:42.510 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1deaff5a-209a-48b4-b3f2-74628e4fd1ad] received connection request
2025-08-01 05:42:42.511 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:42:42.641 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1deaff5a-209a-48b4-b3f2-74628e4fd1ad] socks forwarding established
2025-08-01 05:42:42.689 [info] [command][e9a620b5-0e04-476b-bba9-2b8051bf12f8] Process exited with code 0
2025-08-01 05:42:42.690 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1deaff5a-209a-48b4-b3f2-74628e4fd1ad] socks connection closed
2025-08-01 05:42:42.690 [info] [command][e9a620b5-0e04-476b-bba9-2b8051bf12f8] Socket close event received
2025-08-01 05:42:42.733 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53423 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:43:42.700 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:43:42.702 [info] [command][93cc47a9-b055-46d8-a945-97b9885c5cf7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""93cc47a9-b055-46d8-a945-97b9885c5cf7""}
2025-08-01 05:43:42.703 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][47044761-7152-4923-be87-99a128f68d33] received connection request
2025-08-01 05:43:42.703 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:43:43.000 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47044761-7152-4923-be87-99a128f68d33] socks forwarding established
2025-08-01 05:43:43.043 [info] [command][93cc47a9-b055-46d8-a945-97b9885c5cf7] Process exited with code 0
2025-08-01 05:43:43.044 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][47044761-7152-4923-be87-99a128f68d33] socks connection closed
2025-08-01 05:43:43.044 [info] [command][93cc47a9-b055-46d8-a945-97b9885c5cf7] Socket close event received
2025-08-01 05:43:43.084 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53453 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:44:43.048 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:44:43.050 [info] [command][32a52be4-8a9d-4d9a-93ff-88609a475ed0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""32a52be4-8a9d-4d9a-93ff-88609a475ed0""}
2025-08-01 05:44:43.050 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][13f3163d-3c13-453a-b5d6-a008165cb6bc] received connection request
2025-08-01 05:44:43.051 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:44:43.097 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13f3163d-3c13-453a-b5d6-a008165cb6bc] socks forwarding established
2025-08-01 05:44:43.143 [info] [command][32a52be4-8a9d-4d9a-93ff-88609a475ed0] Process exited with code 0
2025-08-01 05:44:43.143 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13f3163d-3c13-453a-b5d6-a008165cb6bc] socks connection closed
2025-08-01 05:44:43.143 [info] [command][32a52be4-8a9d-4d9a-93ff-88609a475ed0] Socket close event received
2025-08-01 05:44:43.182 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53465 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:45:43.153 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:45:43.155 [info] [command][d815c876-e796-4b13-9d01-e56144d93d93] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d815c876-e796-4b13-9d01-e56144d93d93""}
2025-08-01 05:45:43.156 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3b1c5a6e-4661-4bfe-be7a-7d127070a25a] received connection request
2025-08-01 05:45:43.157 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:45:43.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3b1c5a6e-4661-4bfe-be7a-7d127070a25a] socks forwarding established
2025-08-01 05:45:43.252 [info] [command][d815c876-e796-4b13-9d01-e56144d93d93] Process exited with code 0
2025-08-01 05:45:43.253 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3b1c5a6e-4661-4bfe-be7a-7d127070a25a] socks connection closed
2025-08-01 05:45:43.253 [info] [command][d815c876-e796-4b13-9d01-e56144d93d93] Socket close event received
2025-08-01 05:45:43.292 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53480 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:46:43.262 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:46:43.264 [info] [command][d92eecc5-b787-4aef-afbe-65b3115ca38f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d92eecc5-b787-4aef-afbe-65b3115ca38f""}
2025-08-01 05:46:43.265 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c1355141-4857-40b6-ab2d-af6fbf43db86] received connection request
2025-08-01 05:46:43.265 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:46:43.324 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1355141-4857-40b6-ab2d-af6fbf43db86] socks forwarding established
2025-08-01 05:46:43.371 [info] [command][d92eecc5-b787-4aef-afbe-65b3115ca38f] Process exited with code 0
2025-08-01 05:46:43.371 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c1355141-4857-40b6-ab2d-af6fbf43db86] socks connection closed
2025-08-01 05:46:43.371 [info] [command][d92eecc5-b787-4aef-afbe-65b3115ca38f] Socket close event received
2025-08-01 05:46:43.416 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53493 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:47:43.375 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:47:43.377 [info] [command][1a5fd5a9-183c-408e-bdfb-cfc03cf80da1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1a5fd5a9-183c-408e-bdfb-cfc03cf80da1""}
2025-08-01 05:47:43.377 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][99b48dc2-ae70-4984-b54a-c94e5e3c73be] received connection request
2025-08-01 05:47:43.378 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:47:43.442 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][99b48dc2-ae70-4984-b54a-c94e5e3c73be] socks forwarding established
2025-08-01 05:47:43.488 [info] [command][1a5fd5a9-183c-408e-bdfb-cfc03cf80da1] Process exited with code 0
2025-08-01 05:47:43.488 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][99b48dc2-ae70-4984-b54a-c94e5e3c73be] socks connection closed
2025-08-01 05:47:43.488 [info] [command][1a5fd5a9-183c-408e-bdfb-cfc03cf80da1] Socket close event received
2025-08-01 05:47:43.530 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53546 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:48:43.493 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:48:43.495 [info] [command][8b0e02c9-0e5e-4c0b-a0bd-39970a49b59d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8b0e02c9-0e5e-4c0b-a0bd-39970a49b59d""}
2025-08-01 05:48:43.496 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6632e2a4-ee8a-49f4-b0b2-18b159c09b49] received connection request
2025-08-01 05:48:43.497 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:48:43.541 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6632e2a4-ee8a-49f4-b0b2-18b159c09b49] socks forwarding established
2025-08-01 05:48:43.586 [info] [command][8b0e02c9-0e5e-4c0b-a0bd-39970a49b59d] Process exited with code 0
2025-08-01 05:48:43.586 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6632e2a4-ee8a-49f4-b0b2-18b159c09b49] socks connection closed
2025-08-01 05:48:43.587 [info] [command][8b0e02c9-0e5e-4c0b-a0bd-39970a49b59d] Socket close event received
2025-08-01 05:48:43.630 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53583 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:49:43.591 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:49:43.593 [info] [command][bb0f893b-e527-4d43-b828-270fe6541b35] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bb0f893b-e527-4d43-b828-270fe6541b35""}
2025-08-01 05:49:43.594 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d869068c-ae78-4352-8750-24143f59e460] received connection request
2025-08-01 05:49:43.595 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:49:43.635 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d869068c-ae78-4352-8750-24143f59e460] socks forwarding established
2025-08-01 05:49:43.682 [info] [command][bb0f893b-e527-4d43-b828-270fe6541b35] Process exited with code 0
2025-08-01 05:49:43.682 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d869068c-ae78-4352-8750-24143f59e460] socks connection closed
2025-08-01 05:49:43.682 [info] [command][bb0f893b-e527-4d43-b828-270fe6541b35] Socket close event received
2025-08-01 05:49:43.722 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53600 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:50:43.687 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:50:43.688 [info] [command][e02c5171-28c2-40e0-8301-4f8eeddb2d73] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e02c5171-28c2-40e0-8301-4f8eeddb2d73""}
2025-08-01 05:50:43.689 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][156dddd6-627d-4876-8d13-afffce79d372] received connection request
2025-08-01 05:50:43.689 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:50:43.730 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][156dddd6-627d-4876-8d13-afffce79d372] socks forwarding established
2025-08-01 05:50:43.773 [info] [command][e02c5171-28c2-40e0-8301-4f8eeddb2d73] Process exited with code 0
2025-08-01 05:50:43.774 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][156dddd6-627d-4876-8d13-afffce79d372] socks connection closed
2025-08-01 05:50:43.774 [info] [command][e02c5171-28c2-40e0-8301-4f8eeddb2d73] Socket close event received
2025-08-01 05:50:43.813 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53625 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:51:43.779 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:51:43.781 [info] [command][d31d420e-f8c3-4e41-983d-f8aa93566a8e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d31d420e-f8c3-4e41-983d-f8aa93566a8e""}
2025-08-01 05:51:43.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][80f2155b-e7a8-4118-be44-0a5639b06f11] received connection request
2025-08-01 05:51:43.783 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:51:43.827 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][80f2155b-e7a8-4118-be44-0a5639b06f11] socks forwarding established
2025-08-01 05:51:43.871 [info] [command][d31d420e-f8c3-4e41-983d-f8aa93566a8e] Process exited with code 0
2025-08-01 05:51:43.871 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][80f2155b-e7a8-4118-be44-0a5639b06f11] socks connection closed
2025-08-01 05:51:43.871 [info] [command][d31d420e-f8c3-4e41-983d-f8aa93566a8e] Socket close event received
2025-08-01 05:51:43.911 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53638 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:52:43.871 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:52:43.873 [info] [command][4314dccd-2ab8-4ff0-9d7b-b1a97e876eca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4314dccd-2ab8-4ff0-9d7b-b1a97e876eca""}
2025-08-01 05:52:43.873 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3f67713e-a5d5-4631-9f17-cfd90c30e376] received connection request
2025-08-01 05:52:43.873 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:52:43.971 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3f67713e-a5d5-4631-9f17-cfd90c30e376] socks forwarding established
2025-08-01 05:52:44.016 [info] [command][4314dccd-2ab8-4ff0-9d7b-b1a97e876eca] Process exited with code 0
2025-08-01 05:52:44.016 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3f67713e-a5d5-4631-9f17-cfd90c30e376] socks connection closed
2025-08-01 05:52:44.017 [info] [command][4314dccd-2ab8-4ff0-9d7b-b1a97e876eca] Socket close event received
2025-08-01 05:52:44.055 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53669 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:53:44.018 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:53:44.019 [info] [command][a861a98d-5181-4fce-9edf-4305521c0eef] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a861a98d-5181-4fce-9edf-4305521c0eef""}
2025-08-01 05:53:44.020 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][601ac920-b2af-4c2d-904f-68be46d8ab72] received connection request
2025-08-01 05:53:44.020 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:53:44.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][601ac920-b2af-4c2d-904f-68be46d8ab72] socks forwarding established
2025-08-01 05:53:44.113 [info] [command][a861a98d-5181-4fce-9edf-4305521c0eef] Process exited with code 0
2025-08-01 05:53:44.113 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][601ac920-b2af-4c2d-904f-68be46d8ab72] socks connection closed
2025-08-01 05:53:44.113 [info] [command][a861a98d-5181-4fce-9edf-4305521c0eef] Socket close event received
2025-08-01 05:53:44.154 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53705 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:54:44.114 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:54:44.115 [info] [command][76659113-4519-4b9e-b9a0-b72ff9b7235c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""76659113-4519-4b9e-b9a0-b72ff9b7235c""}
2025-08-01 05:54:44.115 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][55868b94-9dd2-465f-bbcc-a5e088bf228b] received connection request
2025-08-01 05:54:44.116 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:54:44.157 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][55868b94-9dd2-465f-bbcc-a5e088bf228b] socks forwarding established
2025-08-01 05:54:44.201 [info] [command][76659113-4519-4b9e-b9a0-b72ff9b7235c] Process exited with code 0
2025-08-01 05:54:44.201 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][55868b94-9dd2-465f-bbcc-a5e088bf228b] socks connection closed
2025-08-01 05:54:44.201 [info] [command][76659113-4519-4b9e-b9a0-b72ff9b7235c] Socket close event received
2025-08-01 05:54:44.239 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53716 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:55:44.203 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:55:44.205 [info] [command][7ea03b2e-3369-4c3e-b1c4-7777a61ebe21] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7ea03b2e-3369-4c3e-b1c4-7777a61ebe21""}
2025-08-01 05:55:44.205 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c4e641eb-03e7-48ba-8793-c44853e1571d] received connection request
2025-08-01 05:55:44.206 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:55:44.251 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4e641eb-03e7-48ba-8793-c44853e1571d] socks forwarding established
2025-08-01 05:55:44.297 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c4e641eb-03e7-48ba-8793-c44853e1571d] socks connection closed
2025-08-01 05:55:44.298 [info] [command][7ea03b2e-3369-4c3e-b1c4-7777a61ebe21] Process exited with code 0
2025-08-01 05:55:44.298 [info] [command][7ea03b2e-3369-4c3e-b1c4-7777a61ebe21] Socket close event received
2025-08-01 05:55:44.337 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53731 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:56:44.302 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:56:44.304 [info] [command][6b329ad6-ffad-46f0-953a-acfb0d8b89be] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6b329ad6-ffad-46f0-953a-acfb0d8b89be""}
2025-08-01 05:56:44.305 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e545f502-e3f5-4d63-a552-99c2872440fb] received connection request
2025-08-01 05:56:44.305 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:56:44.347 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e545f502-e3f5-4d63-a552-99c2872440fb] socks forwarding established
2025-08-01 05:56:44.391 [info] [command][6b329ad6-ffad-46f0-953a-acfb0d8b89be] Process exited with code 0
2025-08-01 05:56:44.391 [info] [command][6b329ad6-ffad-46f0-953a-acfb0d8b89be] Socket close event received
2025-08-01 05:56:44.391 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e545f502-e3f5-4d63-a552-99c2872440fb] socks connection closed
2025-08-01 05:56:44.431 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53750 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:57:44.397 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:57:44.400 [info] [command][faf3d7d4-0205-430d-a5b7-85649e7bfc76] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""faf3d7d4-0205-430d-a5b7-85649e7bfc76""}
2025-08-01 05:57:44.400 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][fd1cbe80-e05e-4ace-b0e7-882bc5afedb1] received connection request
2025-08-01 05:57:44.401 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:57:44.643 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fd1cbe80-e05e-4ace-b0e7-882bc5afedb1] socks forwarding established
2025-08-01 05:57:44.688 [info] [command][faf3d7d4-0205-430d-a5b7-85649e7bfc76] Process exited with code 0
2025-08-01 05:57:44.688 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][fd1cbe80-e05e-4ace-b0e7-882bc5afedb1] socks connection closed
2025-08-01 05:57:44.689 [info] [command][faf3d7d4-0205-430d-a5b7-85649e7bfc76] Socket close event received
2025-08-01 05:57:44.727 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53780 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:58:44.689 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:58:44.691 [info] [command][87b39c8c-c337-490e-af50-1e908ffc6b89] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""87b39c8c-c337-490e-af50-1e908ffc6b89""}
2025-08-01 05:58:44.691 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c986b801-3367-4ea9-a475-6e9de026a71f] received connection request
2025-08-01 05:58:44.692 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:58:44.837 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c986b801-3367-4ea9-a475-6e9de026a71f] socks forwarding established
2025-08-01 05:58:44.979 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c986b801-3367-4ea9-a475-6e9de026a71f] socks connection closed
2025-08-01 05:58:44.979 [info] [command][87b39c8c-c337-490e-af50-1e908ffc6b89] Process exited with code 0
2025-08-01 05:58:44.979 [info] [command][87b39c8c-c337-490e-af50-1e908ffc6b89] Socket close event received
2025-08-01 05:58:45.018 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53821 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 05:59:44.982 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 05:59:44.984 [info] [command][c494c377-9cf9-41c7-b734-3eb7bc421a2c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c494c377-9cf9-41c7-b734-3eb7bc421a2c""}
2025-08-01 05:59:44.984 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1d99aedc-0a75-433d-933c-d12d82a43e61] received connection request
2025-08-01 05:59:44.985 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 05:59:45.029 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d99aedc-0a75-433d-933c-d12d82a43e61] socks forwarding established
2025-08-01 05:59:45.072 [info] [command][c494c377-9cf9-41c7-b734-3eb7bc421a2c] Process exited with code 0
2025-08-01 05:59:45.072 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d99aedc-0a75-433d-933c-d12d82a43e61] socks connection closed
2025-08-01 05:59:45.073 [info] [command][c494c377-9cf9-41c7-b734-3eb7bc421a2c] Socket close event received
2025-08-01 05:59:45.112 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53843 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:00:45.082 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:00:45.084 [info] [command][a2e84f92-2df8-40dd-84ea-fcb293a8fb56] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a2e84f92-2df8-40dd-84ea-fcb293a8fb56""}
2025-08-01 06:00:45.085 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][aee64aa9-094a-4822-bb1f-5c06cc4b38dd] received connection request
2025-08-01 06:00:45.086 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:00:45.129 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][aee64aa9-094a-4822-bb1f-5c06cc4b38dd] socks forwarding established
2025-08-01 06:00:45.174 [info] [command][a2e84f92-2df8-40dd-84ea-fcb293a8fb56] Process exited with code 0
2025-08-01 06:00:45.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][aee64aa9-094a-4822-bb1f-5c06cc4b38dd] socks connection closed
2025-08-01 06:00:45.174 [info] [command][a2e84f92-2df8-40dd-84ea-fcb293a8fb56] Socket close event received
2025-08-01 06:00:45.214 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53857 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:01:45.178 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:01:45.181 [info] [command][19f0e66c-ee84-4831-96e6-4a8d38458234] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""19f0e66c-ee84-4831-96e6-4a8d38458234""}
2025-08-01 06:01:45.182 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e277cc05-937e-455b-9a71-3319cd73c623] received connection request
2025-08-01 06:01:45.182 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:01:45.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e277cc05-937e-455b-9a71-3319cd73c623] socks forwarding established
2025-08-01 06:01:45.271 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e277cc05-937e-455b-9a71-3319cd73c623] socks connection closed
2025-08-01 06:01:45.272 [info] [command][19f0e66c-ee84-4831-96e6-4a8d38458234] Process exited with code 0
2025-08-01 06:01:45.272 [info] [command][19f0e66c-ee84-4831-96e6-4a8d38458234] Socket close event received
2025-08-01 06:01:45.313 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53874 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:02:45.274 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:02:45.276 [info] [command][d1a6bcde-6d7a-4113-bc41-977cab16b6f7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d1a6bcde-6d7a-4113-bc41-977cab16b6f7""}
2025-08-01 06:02:45.277 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][023bfa26-a334-4a0d-a3b0-530388e4fe94] received connection request
2025-08-01 06:02:45.277 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:02:45.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][023bfa26-a334-4a0d-a3b0-530388e4fe94] socks forwarding established
2025-08-01 06:02:45.365 [info] [command][d1a6bcde-6d7a-4113-bc41-977cab16b6f7] Process exited with code 0
2025-08-01 06:02:45.365 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][023bfa26-a334-4a0d-a3b0-530388e4fe94] socks connection closed
2025-08-01 06:02:45.365 [info] [command][d1a6bcde-6d7a-4113-bc41-977cab16b6f7] Socket close event received
2025-08-01 06:02:45.404 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53910 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:03:45.370 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:03:45.371 [info] [command][9acd16fe-fdd9-423f-81ac-97ea64133d58] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9acd16fe-fdd9-423f-81ac-97ea64133d58""}
2025-08-01 06:03:45.372 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][26bd19ad-ceb4-4113-a127-7ab7b9efbf99] received connection request
2025-08-01 06:03:45.372 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:03:45.414 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][26bd19ad-ceb4-4113-a127-7ab7b9efbf99] socks forwarding established
2025-08-01 06:03:45.459 [info] [command][9acd16fe-fdd9-423f-81ac-97ea64133d58] Process exited with code 0
2025-08-01 06:03:45.460 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][26bd19ad-ceb4-4113-a127-7ab7b9efbf99] socks connection closed
2025-08-01 06:03:45.460 [info] [command][9acd16fe-fdd9-423f-81ac-97ea64133d58] Socket close event received
2025-08-01 06:03:45.499 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53944 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:04:45.460 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:04:45.462 [info] [command][d3c3a199-3977-4a42-9468-6a2c0c702dd2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d3c3a199-3977-4a42-9468-6a2c0c702dd2""}
2025-08-01 06:04:45.462 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][86f40cf3-8ea4-4b5c-a099-f03f3431bff9] received connection request
2025-08-01 06:04:45.463 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:04:45.506 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][86f40cf3-8ea4-4b5c-a099-f03f3431bff9] socks forwarding established
2025-08-01 06:04:45.551 [info] [command][d3c3a199-3977-4a42-9468-6a2c0c702dd2] Process exited with code 0
2025-08-01 06:04:45.551 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][86f40cf3-8ea4-4b5c-a099-f03f3431bff9] socks connection closed
2025-08-01 06:04:45.551 [info] [command][d3c3a199-3977-4a42-9468-6a2c0c702dd2] Socket close event received
2025-08-01 06:04:45.596 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53968 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:05:45.554 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:05:45.556 [info] [command][097c61ac-6537-4c52-8af7-2870a91cffa2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""097c61ac-6537-4c52-8af7-2870a91cffa2""}
2025-08-01 06:05:45.556 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2ec62cc5-b6e1-4aaf-a2b5-52203efccf71] received connection request
2025-08-01 06:05:45.556 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:05:45.604 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2ec62cc5-b6e1-4aaf-a2b5-52203efccf71] socks forwarding established
2025-08-01 06:05:45.651 [info] [command][097c61ac-6537-4c52-8af7-2870a91cffa2] Process exited with code 0
2025-08-01 06:05:45.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2ec62cc5-b6e1-4aaf-a2b5-52203efccf71] socks connection closed
2025-08-01 06:05:45.651 [info] [command][097c61ac-6537-4c52-8af7-2870a91cffa2] Socket close event received
2025-08-01 06:05:45.691 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 53980 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:06:45.661 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:06:45.662 [info] [command][95babdb2-c4d3-4a1a-aa5f-55a4b26313d0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""95babdb2-c4d3-4a1a-aa5f-55a4b26313d0""}
2025-08-01 06:06:45.663 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0860914b-f083-4148-bec5-3f79c941847f] received connection request
2025-08-01 06:06:45.663 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:06:45.828 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0860914b-f083-4148-bec5-3f79c941847f] socks forwarding established
2025-08-01 06:06:45.998 [info] [command][95babdb2-c4d3-4a1a-aa5f-55a4b26313d0] Process exited with code 0
2025-08-01 06:06:45.998 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0860914b-f083-4148-bec5-3f79c941847f] socks connection closed
2025-08-01 06:06:45.998 [info] [command][95babdb2-c4d3-4a1a-aa5f-55a4b26313d0] Socket close event received
2025-08-01 06:06:46.038 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54000 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:07:46.002 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:07:46.004 [info] [command][85e4f911-a887-4b4a-896a-63fb05cd2b77] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""85e4f911-a887-4b4a-896a-63fb05cd2b77""}
2025-08-01 06:07:46.005 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a27aaea8-88fc-467b-ae65-0524aba7d545] received connection request
2025-08-01 06:07:46.006 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:07:46.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a27aaea8-88fc-467b-ae65-0524aba7d545] socks forwarding established
2025-08-01 06:07:46.095 [info] [command][85e4f911-a887-4b4a-896a-63fb05cd2b77] Process exited with code 0
2025-08-01 06:07:46.096 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a27aaea8-88fc-467b-ae65-0524aba7d545] socks connection closed
2025-08-01 06:07:46.096 [info] [command][85e4f911-a887-4b4a-896a-63fb05cd2b77] Socket close event received
2025-08-01 06:07:46.143 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54028 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:08:46.099 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:08:46.101 [info] [command][8425c4c3-397b-406c-8d49-0633e228f97f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8425c4c3-397b-406c-8d49-0633e228f97f""}
2025-08-01 06:08:46.102 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][934d0426-75e9-43fb-8870-5ee68dac274e] received connection request
2025-08-01 06:08:46.102 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:08:46.143 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][934d0426-75e9-43fb-8870-5ee68dac274e] socks forwarding established
2025-08-01 06:08:46.190 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][934d0426-75e9-43fb-8870-5ee68dac274e] socks connection closed
2025-08-01 06:08:46.191 [info] [command][8425c4c3-397b-406c-8d49-0633e228f97f] Process exited with code 0
2025-08-01 06:08:46.191 [info] [command][8425c4c3-397b-406c-8d49-0633e228f97f] Socket close event received
2025-08-01 06:08:46.230 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54061 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:09:46.198 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:09:46.201 [info] [command][11f02c75-6b1c-48cc-abc6-f596244e4d85] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""11f02c75-6b1c-48cc-abc6-f596244e4d85""}
2025-08-01 06:09:46.201 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][22ab3003-4401-4c84-8669-e36e9ab53345] received connection request
2025-08-01 06:09:46.202 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:09:46.247 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22ab3003-4401-4c84-8669-e36e9ab53345] socks forwarding established
2025-08-01 06:09:46.290 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][22ab3003-4401-4c84-8669-e36e9ab53345] socks connection closed
2025-08-01 06:09:46.291 [info] [command][11f02c75-6b1c-48cc-abc6-f596244e4d85] Process exited with code 0
2025-08-01 06:09:46.291 [info] [command][11f02c75-6b1c-48cc-abc6-f596244e4d85] Socket close event received
2025-08-01 06:09:46.333 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54081 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:10:46.301 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:10:46.303 [info] [command][9df572ab-fadf-4e06-95de-998843f3a276] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9df572ab-fadf-4e06-95de-998843f3a276""}
2025-08-01 06:10:46.303 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5125b296-e533-46a8-8188-a3585292b370] received connection request
2025-08-01 06:10:46.304 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 06:10:46.304 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:10:46.354 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5125b296-e533-46a8-8188-a3585292b370] socks forwarding established
2025-08-01 06:10:46.408 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5125b296-e533-46a8-8188-a3585292b370] socks connection closed
2025-08-01 06:10:46.409 [info] [command][9df572ab-fadf-4e06-95de-998843f3a276] Process exited with code 0
2025-08-01 06:10:46.409 [info] [command][9df572ab-fadf-4e06-95de-998843f3a276] Socket close event received
2025-08-01 06:10:46.450 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54101 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:11:46.410 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:11:46.412 [info] [command][ccd4787b-1db3-4185-8683-9be418bb2176] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ccd4787b-1db3-4185-8683-9be418bb2176""}
2025-08-01 06:11:46.413 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b6a70595-f417-49e0-82d3-8a4e978b9b34] received connection request
2025-08-01 06:11:46.413 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:11:46.515 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b6a70595-f417-49e0-82d3-8a4e978b9b34] socks forwarding established
2025-08-01 06:11:46.560 [info] [command][ccd4787b-1db3-4185-8683-9be418bb2176] Process exited with code 0
2025-08-01 06:11:46.561 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b6a70595-f417-49e0-82d3-8a4e978b9b34] socks connection closed
2025-08-01 06:11:46.561 [info] [command][ccd4787b-1db3-4185-8683-9be418bb2176] Socket close event received
2025-08-01 06:11:46.601 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54121 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:12:46.567 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:12:46.569 [info] [command][341f562c-b7a6-4586-8428-259bc6fbca3d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""341f562c-b7a6-4586-8428-259bc6fbca3d""}
2025-08-01 06:12:46.569 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cfbab773-7562-4829-ac02-64313595bc02] received connection request
2025-08-01 06:12:46.570 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:12:46.612 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfbab773-7562-4829-ac02-64313595bc02] socks forwarding established
2025-08-01 06:12:46.658 [info] [command][341f562c-b7a6-4586-8428-259bc6fbca3d] Process exited with code 0
2025-08-01 06:12:46.658 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cfbab773-7562-4829-ac02-64313595bc02] socks connection closed
2025-08-01 06:12:46.658 [info] [command][341f562c-b7a6-4586-8428-259bc6fbca3d] Socket close event received
2025-08-01 06:12:46.697 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54164 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:13:46.663 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:13:46.665 [info] [command][1f537347-98e0-481a-afc1-7e53e5d61acc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1f537347-98e0-481a-afc1-7e53e5d61acc""}
2025-08-01 06:13:46.665 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cdb86d7c-0ead-4b2d-9579-4f998c7a6593] received connection request
2025-08-01 06:13:46.665 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:13:46.712 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cdb86d7c-0ead-4b2d-9579-4f998c7a6593] socks forwarding established
2025-08-01 06:13:46.756 [info] [command][1f537347-98e0-481a-afc1-7e53e5d61acc] Process exited with code 0
2025-08-01 06:13:46.756 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cdb86d7c-0ead-4b2d-9579-4f998c7a6593] socks connection closed
2025-08-01 06:13:46.756 [info] [command][1f537347-98e0-481a-afc1-7e53e5d61acc] Socket close event received
2025-08-01 06:13:46.796 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54194 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:14:46.757 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:14:46.759 [info] [command][5ebb0a1a-dbf1-41ac-baa5-d73de9e3df39] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5ebb0a1a-dbf1-41ac-baa5-d73de9e3df39""}
2025-08-01 06:14:46.759 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][047ac021-c31e-48f4-93fa-ff40e8c87150] received connection request
2025-08-01 06:14:46.760 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:14:46.826 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][047ac021-c31e-48f4-93fa-ff40e8c87150] socks forwarding established
2025-08-01 06:14:46.878 [info] [command][5ebb0a1a-dbf1-41ac-baa5-d73de9e3df39] Process exited with code 0
2025-08-01 06:14:46.878 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][047ac021-c31e-48f4-93fa-ff40e8c87150] socks connection closed
2025-08-01 06:14:46.878 [info] [command][5ebb0a1a-dbf1-41ac-baa5-d73de9e3df39] Socket close event received
2025-08-01 06:14:46.936 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54210 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:15:46.882 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:15:46.884 [info] [command][51edf100-e67a-4609-9037-f9996468b5c8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""51edf100-e67a-4609-9037-f9996468b5c8""}
2025-08-01 06:15:46.884 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9516b55f-398a-4e8d-92a3-82e3c4479787] received connection request
2025-08-01 06:15:46.884 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:15:46.934 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9516b55f-398a-4e8d-92a3-82e3c4479787] socks forwarding established
2025-08-01 06:15:47.061 [info] [command][51edf100-e67a-4609-9037-f9996468b5c8] Process exited with code 0
2025-08-01 06:15:47.062 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9516b55f-398a-4e8d-92a3-82e3c4479787] socks connection closed
2025-08-01 06:15:47.062 [info] [command][51edf100-e67a-4609-9037-f9996468b5c8] Socket close event received
2025-08-01 06:15:47.108 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54229 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:16:47.072 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:16:47.073 [info] [command][b58929c4-468b-4f20-bb42-d4e56619abd3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b58929c4-468b-4f20-bb42-d4e56619abd3""}
2025-08-01 06:16:47.074 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b380cb72-f979-4a11-b597-a6f2946bd14e] received connection request
2025-08-01 06:16:47.074 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:16:47.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b380cb72-f979-4a11-b597-a6f2946bd14e] socks forwarding established
2025-08-01 06:16:47.165 [info] [command][b58929c4-468b-4f20-bb42-d4e56619abd3] Process exited with code 0
2025-08-01 06:16:47.166 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b380cb72-f979-4a11-b597-a6f2946bd14e] socks connection closed
2025-08-01 06:16:47.166 [info] [command][b58929c4-468b-4f20-bb42-d4e56619abd3] Socket close event received
2025-08-01 06:16:47.214 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54242 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:17:47.174 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:17:47.176 [info] [command][cc4d2947-099c-48ff-8105-6f5f874ba28a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cc4d2947-099c-48ff-8105-6f5f874ba28a""}
2025-08-01 06:17:47.176 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][908ec96f-3326-40fe-a89d-e345e2d483ec] received connection request
2025-08-01 06:17:47.176 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:17:47.222 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][908ec96f-3326-40fe-a89d-e345e2d483ec] socks forwarding established
2025-08-01 06:17:47.267 [info] [command][cc4d2947-099c-48ff-8105-6f5f874ba28a] Process exited with code 0
2025-08-01 06:17:47.267 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][908ec96f-3326-40fe-a89d-e345e2d483ec] socks connection closed
2025-08-01 06:17:47.267 [info] [command][cc4d2947-099c-48ff-8105-6f5f874ba28a] Socket close event received
2025-08-01 06:17:47.324 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54268 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:18:47.277 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:18:47.279 [info] [command][a17c5cf3-f23b-4f19-8c34-e0f922400438] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a17c5cf3-f23b-4f19-8c34-e0f922400438""}
2025-08-01 06:18:47.280 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7a490d49-2df4-42fc-a2ce-08869fe2fc7d] received connection request
2025-08-01 06:18:47.280 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:18:47.322 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7a490d49-2df4-42fc-a2ce-08869fe2fc7d] socks forwarding established
2025-08-01 06:18:47.368 [info] [command][a17c5cf3-f23b-4f19-8c34-e0f922400438] Process exited with code 0
2025-08-01 06:18:47.368 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7a490d49-2df4-42fc-a2ce-08869fe2fc7d] socks connection closed
2025-08-01 06:18:47.368 [info] [command][a17c5cf3-f23b-4f19-8c34-e0f922400438] Socket close event received
2025-08-01 06:18:47.408 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54301 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:19:47.375 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:19:47.377 [info] [command][96c88d48-c4d4-4d2c-93da-3b3c6f407341] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""96c88d48-c4d4-4d2c-93da-3b3c6f407341""}
2025-08-01 06:19:47.378 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a815e696-b187-402f-91cf-97d931118be0] received connection request
2025-08-01 06:19:47.378 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:19:47.426 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a815e696-b187-402f-91cf-97d931118be0] socks forwarding established
2025-08-01 06:19:47.470 [info] [command][96c88d48-c4d4-4d2c-93da-3b3c6f407341] Process exited with code 0
2025-08-01 06:19:47.471 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a815e696-b187-402f-91cf-97d931118be0] socks connection closed
2025-08-01 06:19:47.471 [info] [command][96c88d48-c4d4-4d2c-93da-3b3c6f407341] Socket close event received
2025-08-01 06:19:47.511 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54327 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:20:47.480 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:20:47.483 [info] [command][dd86fed6-08fd-4699-82ec-ca902732d766] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""dd86fed6-08fd-4699-82ec-ca902732d766""}
2025-08-01 06:20:47.483 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6a8ae039-6586-4211-ad1e-2d6be589796b] received connection request
2025-08-01 06:20:47.484 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:20:47.536 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6a8ae039-6586-4211-ad1e-2d6be589796b] socks forwarding established
2025-08-01 06:20:47.581 [info] [command][dd86fed6-08fd-4699-82ec-ca902732d766] Process exited with code 0
2025-08-01 06:20:47.581 [info] [command][dd86fed6-08fd-4699-82ec-ca902732d766] Socket close event received
2025-08-01 06:20:47.604 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6a8ae039-6586-4211-ad1e-2d6be589796b] socks connection closed
2025-08-01 06:20:47.639 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54340 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:21:47.592 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:21:47.594 [info] [command][05e2fba8-eeda-42fe-8d57-b80717c157ff] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""05e2fba8-eeda-42fe-8d57-b80717c157ff""}
2025-08-01 06:21:47.594 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][35594194-b7ec-4a8d-b86e-9da985990603] received connection request
2025-08-01 06:21:47.595 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:21:47.641 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35594194-b7ec-4a8d-b86e-9da985990603] socks forwarding established
2025-08-01 06:21:47.686 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][35594194-b7ec-4a8d-b86e-9da985990603] socks connection closed
2025-08-01 06:21:47.687 [info] [command][05e2fba8-eeda-42fe-8d57-b80717c157ff] Process exited with code 0
2025-08-01 06:21:47.687 [info] [command][05e2fba8-eeda-42fe-8d57-b80717c157ff] Socket close event received
2025-08-01 06:21:47.738 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54354 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:22:47.694 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:22:47.695 [info] [command][21d4f1fd-36e4-40f4-92b0-00963fad28c8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""21d4f1fd-36e4-40f4-92b0-00963fad28c8""}
2025-08-01 06:22:47.696 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b7918162-e640-49ee-80c9-7a059ad7fec1] received connection request
2025-08-01 06:22:47.696 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:22:47.743 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b7918162-e640-49ee-80c9-7a059ad7fec1] socks forwarding established
2025-08-01 06:22:47.787 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b7918162-e640-49ee-80c9-7a059ad7fec1] socks connection closed
2025-08-01 06:22:47.788 [info] [command][21d4f1fd-36e4-40f4-92b0-00963fad28c8] Process exited with code 0
2025-08-01 06:22:47.788 [info] [command][21d4f1fd-36e4-40f4-92b0-00963fad28c8] Socket close event received
2025-08-01 06:22:47.829 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54379 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:23:47.790 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:23:47.791 [info] [command][4ceaeb15-e0ff-4fc5-89a3-861df307e7ac] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4ceaeb15-e0ff-4fc5-89a3-861df307e7ac""}
2025-08-01 06:23:47.791 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b2e48a04-2339-437a-b3c5-cdc13e09770a] received connection request
2025-08-01 06:23:47.792 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:23:47.868 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2e48a04-2339-437a-b3c5-cdc13e09770a] socks forwarding established
2025-08-01 06:23:47.913 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2e48a04-2339-437a-b3c5-cdc13e09770a] socks connection closed
2025-08-01 06:23:47.914 [info] [command][4ceaeb15-e0ff-4fc5-89a3-861df307e7ac] Process exited with code 0
2025-08-01 06:23:47.914 [info] [command][4ceaeb15-e0ff-4fc5-89a3-861df307e7ac] Socket close event received
2025-08-01 06:23:47.978 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54406 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:24:47.923 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:24:47.924 [info] [command][0cc18ba7-3ba0-430d-b212-4cfa5ef544f8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0cc18ba7-3ba0-430d-b212-4cfa5ef544f8""}
2025-08-01 06:24:47.925 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][797de4d3-c081-41ab-9018-d007034eaa55] received connection request
2025-08-01 06:24:47.926 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:24:47.972 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][797de4d3-c081-41ab-9018-d007034eaa55] socks forwarding established
2025-08-01 06:24:48.018 [info] [command][0cc18ba7-3ba0-430d-b212-4cfa5ef544f8] Process exited with code 0
2025-08-01 06:24:48.018 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][797de4d3-c081-41ab-9018-d007034eaa55] socks connection closed
2025-08-01 06:24:48.018 [info] [command][0cc18ba7-3ba0-430d-b212-4cfa5ef544f8] Socket close event received
2025-08-01 06:24:48.058 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54422 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:25:48.025 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:25:48.026 [info] [command][d8e8a72d-241e-4048-9111-4c7ac0a36c3e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d8e8a72d-241e-4048-9111-4c7ac0a36c3e""}
2025-08-01 06:25:48.027 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b269365b-9266-4e6c-ab50-8431ff4f825e] received connection request
2025-08-01 06:25:48.027 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:25:48.072 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b269365b-9266-4e6c-ab50-8431ff4f825e] socks forwarding established
2025-08-01 06:25:48.120 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b269365b-9266-4e6c-ab50-8431ff4f825e] socks connection closed
2025-08-01 06:25:48.121 [info] [command][d8e8a72d-241e-4048-9111-4c7ac0a36c3e] Process exited with code 0
2025-08-01 06:25:48.121 [info] [command][d8e8a72d-241e-4048-9111-4c7ac0a36c3e] Socket close event received
2025-08-01 06:25:48.174 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54439 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:26:48.127 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:26:48.128 [info] [command][b7aee68c-4abb-4257-b3a4-50bc6affeda6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7aee68c-4abb-4257-b3a4-50bc6affeda6""}
2025-08-01 06:26:48.129 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a8c30954-7dad-4afd-9a97-bdb8b5517c65] received connection request
2025-08-01 06:26:48.129 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:26:48.172 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8c30954-7dad-4afd-9a97-bdb8b5517c65] socks forwarding established
2025-08-01 06:26:48.216 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a8c30954-7dad-4afd-9a97-bdb8b5517c65] socks connection closed
2025-08-01 06:26:48.216 [info] [command][b7aee68c-4abb-4257-b3a4-50bc6affeda6] Process exited with code 0
2025-08-01 06:26:48.216 [info] [command][b7aee68c-4abb-4257-b3a4-50bc6affeda6] Socket close event received
2025-08-01 06:26:48.259 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54458 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:27:48.224 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:27:48.225 [info] [command][fd2045b1-ac35-4422-9e9e-5fc348c4c372] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fd2045b1-ac35-4422-9e9e-5fc348c4c372""}
2025-08-01 06:27:48.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][706d1825-8c8a-466c-aaf1-82c45a7d5565] received connection request
2025-08-01 06:27:48.226 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:27:48.266 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][706d1825-8c8a-466c-aaf1-82c45a7d5565] socks forwarding established
2025-08-01 06:27:48.309 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][706d1825-8c8a-466c-aaf1-82c45a7d5565] socks connection closed
2025-08-01 06:27:48.309 [info] [command][fd2045b1-ac35-4422-9e9e-5fc348c4c372] Process exited with code 0
2025-08-01 06:27:48.309 [info] [command][fd2045b1-ac35-4422-9e9e-5fc348c4c372] Socket close event received
2025-08-01 06:27:48.349 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54495 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:28:48.311 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:28:48.312 [info] [command][1f5fa99e-3c28-470c-bcc2-e7c1763c7fc6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1f5fa99e-3c28-470c-bcc2-e7c1763c7fc6""}
2025-08-01 06:28:48.313 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][50ee7c82-8c4c-4a48-94b9-54ea398e1b2b] received connection request
2025-08-01 06:28:48.313 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:28:48.356 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50ee7c82-8c4c-4a48-94b9-54ea398e1b2b] socks forwarding established
2025-08-01 06:28:48.404 [info] [command][1f5fa99e-3c28-470c-bcc2-e7c1763c7fc6] Process exited with code 0
2025-08-01 06:28:48.405 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50ee7c82-8c4c-4a48-94b9-54ea398e1b2b] socks connection closed
2025-08-01 06:28:48.405 [info] [command][1f5fa99e-3c28-470c-bcc2-e7c1763c7fc6] Socket close event received
2025-08-01 06:28:48.447 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54524 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:29:48.411 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:29:48.412 [info] [command][326f2a64-9b2c-4fd2-85de-bbbcc0e054c0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""326f2a64-9b2c-4fd2-85de-bbbcc0e054c0""}
2025-08-01 06:29:48.413 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6f83785d-09be-4757-a720-371c60fcace3] received connection request
2025-08-01 06:29:48.414 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:29:48.456 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f83785d-09be-4757-a720-371c60fcace3] socks forwarding established
2025-08-01 06:29:48.502 [info] [command][326f2a64-9b2c-4fd2-85de-bbbcc0e054c0] Process exited with code 0
2025-08-01 06:29:48.502 [info] [command][326f2a64-9b2c-4fd2-85de-bbbcc0e054c0] Socket close event received
2025-08-01 06:29:48.503 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6f83785d-09be-4757-a720-371c60fcace3] socks connection closed
2025-08-01 06:29:48.543 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54539 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:30:48.512 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:30:48.513 [info] [command][70011855-ba56-4ef4-bbaf-c9c9f701ce01] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""70011855-ba56-4ef4-bbaf-c9c9f701ce01""}
2025-08-01 06:30:48.514 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0e8433dd-f206-4f3a-b3d2-e92784609e10] received connection request
2025-08-01 06:30:48.514 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:30:48.555 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0e8433dd-f206-4f3a-b3d2-e92784609e10] socks forwarding established
2025-08-01 06:30:48.606 [info] [command][70011855-ba56-4ef4-bbaf-c9c9f701ce01] Process exited with code 0
2025-08-01 06:30:48.606 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0e8433dd-f206-4f3a-b3d2-e92784609e10] socks connection closed
2025-08-01 06:30:48.606 [info] [command][70011855-ba56-4ef4-bbaf-c9c9f701ce01] Socket close event received
2025-08-01 06:30:48.647 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54553 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:31:48.606 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:31:48.607 [info] [command][a173640c-dc8a-4c33-95b7-0a78b6751e41] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a173640c-dc8a-4c33-95b7-0a78b6751e41""}
2025-08-01 06:31:48.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9475f9b1-ba8a-40f0-b60e-a86a975be3e5] received connection request
2025-08-01 06:31:48.608 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:31:48.650 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9475f9b1-ba8a-40f0-b60e-a86a975be3e5] socks forwarding established
2025-08-01 06:31:48.696 [info] [command][a173640c-dc8a-4c33-95b7-0a78b6751e41] Process exited with code 0
2025-08-01 06:31:48.696 [info] [command][a173640c-dc8a-4c33-95b7-0a78b6751e41] Socket close event received
2025-08-01 06:31:48.697 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9475f9b1-ba8a-40f0-b60e-a86a975be3e5] socks connection closed
2025-08-01 06:31:48.738 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54572 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:32:48.699 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:32:48.701 [info] [command][b35f7e7c-14cd-4e83-84ac-bf52b329fb71] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b35f7e7c-14cd-4e83-84ac-bf52b329fb71""}
2025-08-01 06:32:48.701 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][50111d3e-a05c-4823-9e95-e1507522ea53] received connection request
2025-08-01 06:32:48.702 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:32:48.743 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50111d3e-a05c-4823-9e95-e1507522ea53] socks forwarding established
2025-08-01 06:32:48.789 [info] [command][b35f7e7c-14cd-4e83-84ac-bf52b329fb71] Process exited with code 0
2025-08-01 06:32:48.790 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][50111d3e-a05c-4823-9e95-e1507522ea53] socks connection closed
2025-08-01 06:32:48.790 [info] [command][b35f7e7c-14cd-4e83-84ac-bf52b329fb71] Socket close event received
2025-08-01 06:32:48.832 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54611 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:33:48.794 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:33:48.795 [info] [command][2188be96-bec9-4fb9-97ab-6bc258b3dfc3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2188be96-bec9-4fb9-97ab-6bc258b3dfc3""}
2025-08-01 06:33:48.796 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][134f94ef-4c88-4bed-8813-097b95b12266] received connection request
2025-08-01 06:33:48.796 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:33:48.841 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][134f94ef-4c88-4bed-8813-097b95b12266] socks forwarding established
2025-08-01 06:33:48.886 [info] [command][2188be96-bec9-4fb9-97ab-6bc258b3dfc3] Process exited with code 0
2025-08-01 06:33:48.886 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][134f94ef-4c88-4bed-8813-097b95b12266] socks connection closed
2025-08-01 06:33:48.886 [info] [command][2188be96-bec9-4fb9-97ab-6bc258b3dfc3] Socket close event received
2025-08-01 06:33:48.936 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54648 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:34:48.893 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:34:48.894 [info] [command][1a3ed60a-6d4f-4dff-afea-6cd4823347c9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1a3ed60a-6d4f-4dff-afea-6cd4823347c9""}
2025-08-01 06:34:48.894 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1d199608-16c1-48a1-b46d-c23598b5b6b0] received connection request
2025-08-01 06:34:48.895 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:34:48.955 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d199608-16c1-48a1-b46d-c23598b5b6b0] socks forwarding established
2025-08-01 06:34:49.000 [info] [command][1a3ed60a-6d4f-4dff-afea-6cd4823347c9] Process exited with code 0
2025-08-01 06:34:49.001 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1d199608-16c1-48a1-b46d-c23598b5b6b0] socks connection closed
2025-08-01 06:34:49.001 [info] [command][1a3ed60a-6d4f-4dff-afea-6cd4823347c9] Socket close event received
2025-08-01 06:34:49.056 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54665 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:35:49.005 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:35:49.006 [info] [command][eba272b0-a19d-4e0d-a6f2-c69fdc561867] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eba272b0-a19d-4e0d-a6f2-c69fdc561867""}
2025-08-01 06:35:49.007 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][eb55bb3b-c324-4814-b51d-70e8964638d6] received connection request
2025-08-01 06:35:49.008 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:35:49.056 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb55bb3b-c324-4814-b51d-70e8964638d6] socks forwarding established
2025-08-01 06:35:49.101 [info] [command][eba272b0-a19d-4e0d-a6f2-c69fdc561867] Process exited with code 0
2025-08-01 06:35:49.101 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][eb55bb3b-c324-4814-b51d-70e8964638d6] socks connection closed
2025-08-01 06:35:49.101 [info] [command][eba272b0-a19d-4e0d-a6f2-c69fdc561867] Socket close event received
2025-08-01 06:35:49.140 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54680 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:36:49.111 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:36:49.112 [info] [command][87bbaa2e-2ea4-4f28-a40e-b2e8c96279df] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""87bbaa2e-2ea4-4f28-a40e-b2e8c96279df""}
2025-08-01 06:36:49.113 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][74e322c2-6684-4adb-8a83-612f9ea74eaa] received connection request
2025-08-01 06:36:49.114 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:36:49.158 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][74e322c2-6684-4adb-8a83-612f9ea74eaa] socks forwarding established
2025-08-01 06:36:49.204 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][74e322c2-6684-4adb-8a83-612f9ea74eaa] socks connection closed
2025-08-01 06:36:49.205 [info] [command][87bbaa2e-2ea4-4f28-a40e-b2e8c96279df] Process exited with code 0
2025-08-01 06:36:49.205 [info] [command][87bbaa2e-2ea4-4f28-a40e-b2e8c96279df] Socket close event received
2025-08-01 06:36:49.244 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54699 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:37:49.215 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:37:49.216 [info] [command][2e74f577-0556-4837-9e78-ac63bbc7a3c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2e74f577-0556-4837-9e78-ac63bbc7a3c7""}
2025-08-01 06:37:49.217 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3f5f9a93-7818-4841-a7a9-18a48a737c20] received connection request
2025-08-01 06:37:49.217 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 06:37:49.217 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:37:49.263 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3f5f9a93-7818-4841-a7a9-18a48a737c20] socks forwarding established
2025-08-01 06:37:49.308 [info] [command][2e74f577-0556-4837-9e78-ac63bbc7a3c7] Process exited with code 0
2025-08-01 06:37:49.308 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3f5f9a93-7818-4841-a7a9-18a48a737c20] socks connection closed
2025-08-01 06:37:49.308 [info] [command][2e74f577-0556-4837-9e78-ac63bbc7a3c7] Socket close event received
2025-08-01 06:37:49.347 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54727 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:38:49.319 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:38:49.320 [info] [command][1d66f01c-e9f9-4b3b-87aa-d53ea0548f71] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1d66f01c-e9f9-4b3b-87aa-d53ea0548f71""}
2025-08-01 06:38:49.321 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8b795840-7467-45a8-8115-cf0524a57199] received connection request
2025-08-01 06:38:49.321 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:38:49.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8b795840-7467-45a8-8115-cf0524a57199] socks forwarding established
2025-08-01 06:38:49.408 [info] [command][1d66f01c-e9f9-4b3b-87aa-d53ea0548f71] Process exited with code 0
2025-08-01 06:38:49.409 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8b795840-7467-45a8-8115-cf0524a57199] socks connection closed
2025-08-01 06:38:49.409 [info] [command][1d66f01c-e9f9-4b3b-87aa-d53ea0548f71] Socket close event received
2025-08-01 06:38:49.449 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54762 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:39:49.409 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:39:49.410 [info] [command][a75c604c-1d06-4bcf-bd34-dcb285be816c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a75c604c-1d06-4bcf-bd34-dcb285be816c""}
2025-08-01 06:39:49.410 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6df43cd4-2f6a-4fef-9b48-4792d90a382f] received connection request
2025-08-01 06:39:49.410 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:39:49.552 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6df43cd4-2f6a-4fef-9b48-4792d90a382f] socks forwarding established
2025-08-01 06:39:49.599 [info] [command][a75c604c-1d06-4bcf-bd34-dcb285be816c] Process exited with code 0
2025-08-01 06:39:49.600 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6df43cd4-2f6a-4fef-9b48-4792d90a382f] socks connection closed
2025-08-01 06:39:49.600 [info] [command][a75c604c-1d06-4bcf-bd34-dcb285be816c] Socket close event received
2025-08-01 06:39:49.641 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54778 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:40:49.605 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:40:49.608 [info] [command][0fe92e82-9627-4008-b2cd-195c84f39b9b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0fe92e82-9627-4008-b2cd-195c84f39b9b""}
2025-08-01 06:40:49.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cd985365-35c3-411d-8c5b-ca7ea7124671] received connection request
2025-08-01 06:40:49.609 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:40:49.650 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd985365-35c3-411d-8c5b-ca7ea7124671] socks forwarding established
2025-08-01 06:40:49.695 [info] [command][0fe92e82-9627-4008-b2cd-195c84f39b9b] Process exited with code 0
2025-08-01 06:40:49.695 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cd985365-35c3-411d-8c5b-ca7ea7124671] socks connection closed
2025-08-01 06:40:49.695 [info] [command][0fe92e82-9627-4008-b2cd-195c84f39b9b] Socket close event received
2025-08-01 06:40:49.735 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54800 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:41:49.700 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:41:49.701 [info] [command][6f6a7999-378a-4301-a661-1cdd6416aba6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6f6a7999-378a-4301-a661-1cdd6416aba6""}
2025-08-01 06:41:49.702 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e71ee63a-4d59-4e0b-a2a5-355ba301b092] received connection request
2025-08-01 06:41:49.703 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:41:49.746 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e71ee63a-4d59-4e0b-a2a5-355ba301b092] socks forwarding established
2025-08-01 06:41:49.790 [info] [command][6f6a7999-378a-4301-a661-1cdd6416aba6] Process exited with code 0
2025-08-01 06:41:49.790 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e71ee63a-4d59-4e0b-a2a5-355ba301b092] socks connection closed
2025-08-01 06:41:49.791 [info] [command][6f6a7999-378a-4301-a661-1cdd6416aba6] Socket close event received
2025-08-01 06:41:49.831 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54825 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:42:49.793 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:42:49.795 [info] [command][c34d2c7e-1312-4c87-a50d-befac991e33a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c34d2c7e-1312-4c87-a50d-befac991e33a""}
2025-08-01 06:42:49.796 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ffdfd4a9-1802-4e16-a222-40957b5f9a0f] received connection request
2025-08-01 06:42:49.796 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:42:49.841 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ffdfd4a9-1802-4e16-a222-40957b5f9a0f] socks forwarding established
2025-08-01 06:42:49.885 [info] [command][c34d2c7e-1312-4c87-a50d-befac991e33a] Process exited with code 0
2025-08-01 06:42:49.885 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ffdfd4a9-1802-4e16-a222-40957b5f9a0f] socks connection closed
2025-08-01 06:42:49.885 [info] [command][c34d2c7e-1312-4c87-a50d-befac991e33a] Socket close event received
2025-08-01 06:42:49.925 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54870 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:43:49.890 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:43:49.892 [info] [command][37035485-e8a0-4739-a089-ea13755f21ba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""37035485-e8a0-4739-a089-ea13755f21ba""}
2025-08-01 06:43:49.892 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8c9d7a03-ec59-4d97-be87-e4c2b76e6e03] received connection request
2025-08-01 06:43:49.893 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:43:49.934 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8c9d7a03-ec59-4d97-be87-e4c2b76e6e03] socks forwarding established
2025-08-01 06:43:49.979 [info] [command][37035485-e8a0-4739-a089-ea13755f21ba] Process exited with code 0
2025-08-01 06:43:49.979 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8c9d7a03-ec59-4d97-be87-e4c2b76e6e03] socks connection closed
2025-08-01 06:43:49.979 [info] [command][37035485-e8a0-4739-a089-ea13755f21ba] Socket close event received
2025-08-01 06:43:50.081 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54906 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:44:49.986 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:44:49.988 [info] [command][f35c2dc8-a8dc-4f54-a302-ac2bc66a27ba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f35c2dc8-a8dc-4f54-a302-ac2bc66a27ba""}
2025-08-01 06:44:49.988 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f4cb79e5-65ed-47fc-98ed-41a12a1a60c1] received connection request
2025-08-01 06:44:49.989 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:44:50.033 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4cb79e5-65ed-47fc-98ed-41a12a1a60c1] socks forwarding established
2025-08-01 06:44:50.079 [info] [command][f35c2dc8-a8dc-4f54-a302-ac2bc66a27ba] Process exited with code 0
2025-08-01 06:44:50.079 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4cb79e5-65ed-47fc-98ed-41a12a1a60c1] socks connection closed
2025-08-01 06:44:50.079 [info] [command][f35c2dc8-a8dc-4f54-a302-ac2bc66a27ba] Socket close event received
2025-08-01 06:44:50.119 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54926 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:45:50.082 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:45:50.084 [info] [command][57a6c375-db52-4768-87fc-30c6a5043957] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""57a6c375-db52-4768-87fc-30c6a5043957""}
2025-08-01 06:45:50.085 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ad0a061c-ad35-4e2e-a7bb-168162e1d0f8] received connection request
2025-08-01 06:45:50.085 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:45:50.130 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad0a061c-ad35-4e2e-a7bb-168162e1d0f8] socks forwarding established
2025-08-01 06:45:50.174 [info] [command][57a6c375-db52-4768-87fc-30c6a5043957] Process exited with code 0
2025-08-01 06:45:50.175 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ad0a061c-ad35-4e2e-a7bb-168162e1d0f8] socks connection closed
2025-08-01 06:45:50.175 [info] [command][57a6c375-db52-4768-87fc-30c6a5043957] Socket close event received
2025-08-01 06:45:50.213 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54951 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:46:50.178 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:46:50.179 [info] [command][79394e02-f232-4e38-a833-945e70e09d01] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""79394e02-f232-4e38-a833-945e70e09d01""}
2025-08-01 06:46:50.180 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][82daa58e-f680-4240-8ce2-f17535931ce0] received connection request
2025-08-01 06:46:50.180 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:46:50.227 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82daa58e-f680-4240-8ce2-f17535931ce0] socks forwarding established
2025-08-01 06:46:50.274 [info] [command][79394e02-f232-4e38-a833-945e70e09d01] Process exited with code 0
2025-08-01 06:46:50.274 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82daa58e-f680-4240-8ce2-f17535931ce0] socks connection closed
2025-08-01 06:46:50.274 [info] [command][79394e02-f232-4e38-a833-945e70e09d01] Socket close event received
2025-08-01 06:46:50.316 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 54971 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:47:50.284 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:47:50.285 [info] [command][d23d1d64-cd6b-4e58-a846-e3bf3c04f1ca] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d23d1d64-cd6b-4e58-a846-e3bf3c04f1ca""}
2025-08-01 06:47:50.286 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][db9b7cf6-ecac-4158-94f2-c5a111e10441] received connection request
2025-08-01 06:47:50.287 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:47:50.347 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db9b7cf6-ecac-4158-94f2-c5a111e10441] socks forwarding established
2025-08-01 06:47:50.683 [info] [command][d23d1d64-cd6b-4e58-a846-e3bf3c04f1ca] Process exited with code 0
2025-08-01 06:47:50.683 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db9b7cf6-ecac-4158-94f2-c5a111e10441] socks connection closed
2025-08-01 06:47:50.683 [info] [command][d23d1d64-cd6b-4e58-a846-e3bf3c04f1ca] Socket close event received
2025-08-01 06:47:50.724 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55001 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:48:50.690 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:48:50.690 [info] [command][69062ee2-31c6-40b6-92b0-fbfbc0fbfb46] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""69062ee2-31c6-40b6-92b0-fbfbc0fbfb46""}
2025-08-01 06:48:50.691 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][784a804e-46c5-4933-9deb-5945e9051e80] received connection request
2025-08-01 06:48:50.691 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:48:50.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][784a804e-46c5-4933-9deb-5945e9051e80] socks forwarding established
2025-08-01 06:48:50.878 [info] [command][69062ee2-31c6-40b6-92b0-fbfbc0fbfb46] Process exited with code 0
2025-08-01 06:48:50.878 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][784a804e-46c5-4933-9deb-5945e9051e80] socks connection closed
2025-08-01 06:48:50.878 [info] [command][69062ee2-31c6-40b6-92b0-fbfbc0fbfb46] Socket close event received
2025-08-01 06:48:50.918 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55035 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:49:50.886 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:49:50.888 [info] [command][3ef384cf-931b-435d-9e24-2818b62298c6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3ef384cf-931b-435d-9e24-2818b62298c6""}
2025-08-01 06:49:50.888 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0aa60571-0dc5-46cf-a388-d8d4e1674a34] received connection request
2025-08-01 06:49:50.889 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:49:50.962 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0aa60571-0dc5-46cf-a388-d8d4e1674a34] socks forwarding established
2025-08-01 06:49:51.023 [info] [command][3ef384cf-931b-435d-9e24-2818b62298c6] Process exited with code 0
2025-08-01 06:49:51.024 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0aa60571-0dc5-46cf-a388-d8d4e1674a34] socks connection closed
2025-08-01 06:49:51.024 [info] [command][3ef384cf-931b-435d-9e24-2818b62298c6] Socket close event received
2025-08-01 06:49:51.067 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55064 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:50:51.030 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:50:51.032 [info] [command][7aca3495-ff94-4f4b-83eb-8e4f702d6869] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7aca3495-ff94-4f4b-83eb-8e4f702d6869""}
2025-08-01 06:50:51.033 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e61f0461-9823-4617-b690-c39c50362175] received connection request
2025-08-01 06:50:51.034 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:50:51.077 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e61f0461-9823-4617-b690-c39c50362175] socks forwarding established
2025-08-01 06:50:51.123 [info] [command][7aca3495-ff94-4f4b-83eb-8e4f702d6869] Process exited with code 0
2025-08-01 06:50:51.123 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e61f0461-9823-4617-b690-c39c50362175] socks connection closed
2025-08-01 06:50:51.123 [info] [command][7aca3495-ff94-4f4b-83eb-8e4f702d6869] Socket close event received
2025-08-01 06:50:51.164 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55090 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:51:51.124 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:51:51.126 [info] [command][75b528b7-865b-4594-9267-ec34a47ea398] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""75b528b7-865b-4594-9267-ec34a47ea398""}
2025-08-01 06:51:51.127 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3e7e8c07-751a-4fc7-a27c-d23c18cff32a] received connection request
2025-08-01 06:51:51.127 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:51:51.167 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3e7e8c07-751a-4fc7-a27c-d23c18cff32a] socks forwarding established
2025-08-01 06:51:51.213 [info] [command][75b528b7-865b-4594-9267-ec34a47ea398] Process exited with code 0
2025-08-01 06:51:51.213 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3e7e8c07-751a-4fc7-a27c-d23c18cff32a] socks connection closed
2025-08-01 06:51:51.214 [info] [command][75b528b7-865b-4594-9267-ec34a47ea398] Socket close event received
2025-08-01 06:51:51.253 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55106 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:52:51.224 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:52:51.225 [info] [command][94a18dc0-820f-4c0e-878a-730d0cf773b7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""94a18dc0-820f-4c0e-878a-730d0cf773b7""}
2025-08-01 06:52:51.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6525aa60-a3be-4947-abe1-466d90e543ed] received connection request
2025-08-01 06:52:51.226 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:52:51.268 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6525aa60-a3be-4947-abe1-466d90e543ed] socks forwarding established
2025-08-01 06:52:51.312 [info] [command][94a18dc0-820f-4c0e-878a-730d0cf773b7] Process exited with code 0
2025-08-01 06:52:51.313 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6525aa60-a3be-4947-abe1-466d90e543ed] socks connection closed
2025-08-01 06:52:51.313 [info] [command][94a18dc0-820f-4c0e-878a-730d0cf773b7] Socket close event received
2025-08-01 06:52:51.351 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55138 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:53:51.317 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:53:51.319 [info] [command][ea5bf909-e3a2-4ea9-b783-d59efad0a620] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ea5bf909-e3a2-4ea9-b783-d59efad0a620""}
2025-08-01 06:53:51.320 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2872fe1d-e078-4ed6-a52f-fe843badd4d3] received connection request
2025-08-01 06:53:51.320 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:53:51.407 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2872fe1d-e078-4ed6-a52f-fe843badd4d3] socks forwarding established
2025-08-01 06:53:51.452 [info] [command][ea5bf909-e3a2-4ea9-b783-d59efad0a620] Process exited with code 0
2025-08-01 06:53:51.453 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2872fe1d-e078-4ed6-a52f-fe843badd4d3] socks connection closed
2025-08-01 06:53:51.453 [info] [command][ea5bf909-e3a2-4ea9-b783-d59efad0a620] Socket close event received
2025-08-01 06:53:51.493 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55167 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:54:51.461 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:54:51.463 [info] [command][988b8df7-57d5-41f7-8038-4d38d512da36] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""988b8df7-57d5-41f7-8038-4d38d512da36""}
2025-08-01 06:54:51.463 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b95874d4-622d-4f48-92ef-4e55ceb86e66] received connection request
2025-08-01 06:54:51.464 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:54:51.509 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b95874d4-622d-4f48-92ef-4e55ceb86e66] socks forwarding established
2025-08-01 06:54:51.554 [info] [command][988b8df7-57d5-41f7-8038-4d38d512da36] Process exited with code 0
2025-08-01 06:54:51.554 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b95874d4-622d-4f48-92ef-4e55ceb86e66] socks connection closed
2025-08-01 06:54:51.554 [info] [command][988b8df7-57d5-41f7-8038-4d38d512da36] Socket close event received
2025-08-01 06:54:51.596 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55190 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:55:51.557 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:55:51.559 [info] [command][bbd3a589-2f7f-4e4c-9532-8372f30131d4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bbd3a589-2f7f-4e4c-9532-8372f30131d4""}
2025-08-01 06:55:51.560 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][847f6abc-de2a-407b-bdec-78319fa6b14b] received connection request
2025-08-01 06:55:51.560 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:55:51.606 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][847f6abc-de2a-407b-bdec-78319fa6b14b] socks forwarding established
2025-08-01 06:55:51.651 [info] [command][bbd3a589-2f7f-4e4c-9532-8372f30131d4] Process exited with code 0
2025-08-01 06:55:51.652 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][847f6abc-de2a-407b-bdec-78319fa6b14b] socks connection closed
2025-08-01 06:55:51.652 [info] [command][bbd3a589-2f7f-4e4c-9532-8372f30131d4] Socket close event received
2025-08-01 06:55:51.692 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55219 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:56:51.658 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:56:51.660 [info] [command][2ddd77de-6b4d-4472-bc1e-c61384ee3289] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2ddd77de-6b4d-4472-bc1e-c61384ee3289""}
2025-08-01 06:56:51.661 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5f05e5c6-0125-4a49-94aa-36025309689a] received connection request
2025-08-01 06:56:51.661 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:56:51.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5f05e5c6-0125-4a49-94aa-36025309689a] socks forwarding established
2025-08-01 06:56:51.750 [info] [command][2ddd77de-6b4d-4472-bc1e-c61384ee3289] Process exited with code 0
2025-08-01 06:56:51.750 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5f05e5c6-0125-4a49-94aa-36025309689a] socks connection closed
2025-08-01 06:56:51.750 [info] [command][2ddd77de-6b4d-4472-bc1e-c61384ee3289] Socket close event received
2025-08-01 06:56:51.790 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55244 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:57:51.753 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:57:51.754 [info] [command][6691df0b-fd28-44dc-bf83-7622234b272c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6691df0b-fd28-44dc-bf83-7622234b272c""}
2025-08-01 06:57:51.755 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6fdc2125-2bad-472f-a653-9db8006e2dfc] received connection request
2025-08-01 06:57:51.756 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:57:51.800 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fdc2125-2bad-472f-a653-9db8006e2dfc] socks forwarding established
2025-08-01 06:57:51.846 [info] [command][6691df0b-fd28-44dc-bf83-7622234b272c] Process exited with code 0
2025-08-01 06:57:51.846 [info] [command][6691df0b-fd28-44dc-bf83-7622234b272c] Socket close event received
2025-08-01 06:57:51.847 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fdc2125-2bad-472f-a653-9db8006e2dfc] socks connection closed
2025-08-01 06:57:51.886 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55271 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:58:51.857 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:58:51.859 [info] [command][200a5e8f-6ceb-408c-9ffc-ec9803b06186] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""200a5e8f-6ceb-408c-9ffc-ec9803b06186""}
2025-08-01 06:58:51.860 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][98beacef-4771-4c98-ba1c-2394afe6b447] received connection request
2025-08-01 06:58:51.860 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:58:52.034 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][98beacef-4771-4c98-ba1c-2394afe6b447] socks forwarding established
2025-08-01 06:58:52.213 [info] [command][200a5e8f-6ceb-408c-9ffc-ec9803b06186] Process exited with code 0
2025-08-01 06:58:52.213 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][98beacef-4771-4c98-ba1c-2394afe6b447] socks connection closed
2025-08-01 06:58:52.213 [info] [command][200a5e8f-6ceb-408c-9ffc-ec9803b06186] Socket close event received
2025-08-01 06:58:52.252 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55306 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 06:59:52.223 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 06:59:52.225 [info] [command][337fb3a8-d9ab-40dd-8a21-197ff130b3d4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""337fb3a8-d9ab-40dd-8a21-197ff130b3d4""}
2025-08-01 06:59:52.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b8bb84c1-1519-45fe-846d-c8f0c5c04cd6] received connection request
2025-08-01 06:59:52.225 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 06:59:52.271 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b8bb84c1-1519-45fe-846d-c8f0c5c04cd6] socks forwarding established
2025-08-01 06:59:52.315 [info] [command][337fb3a8-d9ab-40dd-8a21-197ff130b3d4] Process exited with code 0
2025-08-01 06:59:52.315 [info] [command][337fb3a8-d9ab-40dd-8a21-197ff130b3d4] Socket close event received
2025-08-01 06:59:52.317 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b8bb84c1-1519-45fe-846d-c8f0c5c04cd6] socks connection closed
2025-08-01 06:59:52.358 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55324 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:00:52.319 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:00:52.321 [info] [command][b60b3bf4-c066-499c-a1f0-2291bc2acb4f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b60b3bf4-c066-499c-a1f0-2291bc2acb4f""}
2025-08-01 07:00:52.322 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d74f10eb-167b-4362-889c-0778c281bef9] received connection request
2025-08-01 07:00:52.322 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:00:52.364 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d74f10eb-167b-4362-889c-0778c281bef9] socks forwarding established
2025-08-01 07:00:52.410 [info] [command][b60b3bf4-c066-499c-a1f0-2291bc2acb4f] Process exited with code 0
2025-08-01 07:00:52.411 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d74f10eb-167b-4362-889c-0778c281bef9] socks connection closed
2025-08-01 07:00:52.411 [info] [command][b60b3bf4-c066-499c-a1f0-2291bc2acb4f] Socket close event received
2025-08-01 07:00:52.450 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55347 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:01:52.411 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:01:52.413 [info] [command][ac192cea-34d9-4104-abb2-52aea6239caa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ac192cea-34d9-4104-abb2-52aea6239caa""}
2025-08-01 07:01:52.414 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][842cb7b1-bf49-4e57-96de-837cd7e92dfa] received connection request
2025-08-01 07:01:52.414 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:01:52.457 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][842cb7b1-bf49-4e57-96de-837cd7e92dfa] socks forwarding established
2025-08-01 07:01:52.504 [info] [command][ac192cea-34d9-4104-abb2-52aea6239caa] Process exited with code 0
2025-08-01 07:01:52.504 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][842cb7b1-bf49-4e57-96de-837cd7e92dfa] socks connection closed
2025-08-01 07:01:52.504 [info] [command][ac192cea-34d9-4104-abb2-52aea6239caa] Socket close event received
2025-08-01 07:01:52.547 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55363 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:02:52.505 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:02:52.507 [info] [command][87648708-3b6a-437c-864c-7da19a891d1f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""87648708-3b6a-437c-864c-7da19a891d1f""}
2025-08-01 07:02:52.507 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0ce39cf7-d142-4945-90da-7e020eeb3702] received connection request
2025-08-01 07:02:52.508 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:02:52.630 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ce39cf7-d142-4945-90da-7e020eeb3702] socks forwarding established
2025-08-01 07:02:52.676 [info] [command][87648708-3b6a-437c-864c-7da19a891d1f] Process exited with code 0
2025-08-01 07:02:52.676 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0ce39cf7-d142-4945-90da-7e020eeb3702] socks connection closed
2025-08-01 07:02:52.676 [info] [command][87648708-3b6a-437c-864c-7da19a891d1f] Socket close event received
2025-08-01 07:02:52.716 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55389 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:03:52.685 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:03:52.686 [info] [command][bf38af8b-3e68-4b99-9b45-962d58866bcb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bf38af8b-3e68-4b99-9b45-962d58866bcb""}
2025-08-01 07:03:52.687 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4918f59a-250b-45e1-8254-515ca7cfdb0d] received connection request
2025-08-01 07:03:52.687 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:03:52.727 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4918f59a-250b-45e1-8254-515ca7cfdb0d] socks forwarding established
2025-08-01 07:03:52.771 [info] [command][bf38af8b-3e68-4b99-9b45-962d58866bcb] Process exited with code 0
2025-08-01 07:03:52.772 [info] [command][bf38af8b-3e68-4b99-9b45-962d58866bcb] Socket close event received
2025-08-01 07:03:52.773 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4918f59a-250b-45e1-8254-515ca7cfdb0d] socks connection closed
2025-08-01 07:03:52.811 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55422 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:04:52.775 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:04:52.777 [info] [command][e83725db-3165-43f7-8a0c-dfc7f6f9d31b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e83725db-3165-43f7-8a0c-dfc7f6f9d31b""}
2025-08-01 07:04:52.778 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1a67921c-949e-48f2-9c9e-82c172bc1995] received connection request
2025-08-01 07:04:52.778 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:04:52.824 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a67921c-949e-48f2-9c9e-82c172bc1995] socks forwarding established
2025-08-01 07:04:52.868 [info] [command][e83725db-3165-43f7-8a0c-dfc7f6f9d31b] Process exited with code 0
2025-08-01 07:04:52.869 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1a67921c-949e-48f2-9c9e-82c172bc1995] socks connection closed
2025-08-01 07:04:52.869 [info] [command][e83725db-3165-43f7-8a0c-dfc7f6f9d31b] Socket close event received
2025-08-01 07:04:52.908 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55434 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:05:52.876 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:05:52.878 [info] [command][a441a3ae-f0ec-4fa0-a8a2-a56fb9e929c0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a441a3ae-f0ec-4fa0-a8a2-a56fb9e929c0""}
2025-08-01 07:05:52.879 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b551b7ef-f7e6-45f0-8725-d911138e0067] received connection request
2025-08-01 07:05:52.880 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:05:52.923 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b551b7ef-f7e6-45f0-8725-d911138e0067] socks forwarding established
2025-08-01 07:05:52.968 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b551b7ef-f7e6-45f0-8725-d911138e0067] socks connection closed
2025-08-01 07:05:52.968 [info] [command][a441a3ae-f0ec-4fa0-a8a2-a56fb9e929c0] Process exited with code 0
2025-08-01 07:05:52.969 [info] [command][a441a3ae-f0ec-4fa0-a8a2-a56fb9e929c0] Socket close event received
2025-08-01 07:05:53.010 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55448 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:06:52.971 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:06:52.973 [info] [command][89e5687b-9dee-4c13-b41e-b70ed3cbb26f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""89e5687b-9dee-4c13-b41e-b70ed3cbb26f""}
2025-08-01 07:06:52.973 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][39a60970-593b-4579-ae3c-d366a32f8e68] received connection request
2025-08-01 07:06:52.973 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:06:53.018 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39a60970-593b-4579-ae3c-d366a32f8e68] socks forwarding established
2025-08-01 07:06:53.063 [info] [command][89e5687b-9dee-4c13-b41e-b70ed3cbb26f] Process exited with code 0
2025-08-01 07:06:53.063 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39a60970-593b-4579-ae3c-d366a32f8e68] socks connection closed
2025-08-01 07:06:53.063 [info] [command][89e5687b-9dee-4c13-b41e-b70ed3cbb26f] Socket close event received
2025-08-01 07:06:53.103 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55464 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:07:53.069 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:07:53.071 [info] [command][9102a065-789d-4e06-ad1d-fa1e9dd6bb3f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9102a065-789d-4e06-ad1d-fa1e9dd6bb3f""}
2025-08-01 07:07:53.071 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3ff99218-f8ad-4cb3-b7d1-b5b404a97865] received connection request
2025-08-01 07:07:53.072 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:07:53.115 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3ff99218-f8ad-4cb3-b7d1-b5b404a97865] socks forwarding established
2025-08-01 07:07:53.166 [info] [command][9102a065-789d-4e06-ad1d-fa1e9dd6bb3f] Process exited with code 0
2025-08-01 07:07:53.166 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3ff99218-f8ad-4cb3-b7d1-b5b404a97865] socks connection closed
2025-08-01 07:07:53.166 [info] [command][9102a065-789d-4e06-ad1d-fa1e9dd6bb3f] Socket close event received
2025-08-01 07:07:53.209 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55489 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:08:53.171 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:08:53.172 [info] [command][b5df4b6b-ce2b-42cd-b70c-9021001be733] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b5df4b6b-ce2b-42cd-b70c-9021001be733""}
2025-08-01 07:08:53.172 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][878cd8c6-0cef-4e8f-a950-e34f74c0b287] received connection request
2025-08-01 07:08:53.173 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:08:53.216 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][878cd8c6-0cef-4e8f-a950-e34f74c0b287] socks forwarding established
2025-08-01 07:08:53.261 [info] [command][b5df4b6b-ce2b-42cd-b70c-9021001be733] Process exited with code 0
2025-08-01 07:08:53.261 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][878cd8c6-0cef-4e8f-a950-e34f74c0b287] socks connection closed
2025-08-01 07:08:53.261 [info] [command][b5df4b6b-ce2b-42cd-b70c-9021001be733] Socket close event received
2025-08-01 07:08:53.300 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55522 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:09:53.266 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:09:53.268 [info] [command][f2e742f0-bbbf-4391-b2f6-739cb9ca24c4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f2e742f0-bbbf-4391-b2f6-739cb9ca24c4""}
2025-08-01 07:09:53.269 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d8e9a748-4548-4e6b-89ef-a8c1a76629b6] received connection request
2025-08-01 07:09:53.269 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:09:53.313 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d8e9a748-4548-4e6b-89ef-a8c1a76629b6] socks forwarding established
2025-08-01 07:09:53.357 [info] [command][f2e742f0-bbbf-4391-b2f6-739cb9ca24c4] Process exited with code 0
2025-08-01 07:09:53.357 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d8e9a748-4548-4e6b-89ef-a8c1a76629b6] socks connection closed
2025-08-01 07:09:53.357 [info] [command][f2e742f0-bbbf-4391-b2f6-739cb9ca24c4] Socket close event received
2025-08-01 07:09:53.396 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55539 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:10:53.359 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:10:53.361 [info] [command][83bad43b-fb5c-4228-aed4-61e80baef9be] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""83bad43b-fb5c-4228-aed4-61e80baef9be""}
2025-08-01 07:10:53.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3d06ba00-5e41-4539-9729-c311cb7335ba] received connection request
2025-08-01 07:10:53.363 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:10:53.407 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3d06ba00-5e41-4539-9729-c311cb7335ba] socks forwarding established
2025-08-01 07:10:53.453 [info] [command][83bad43b-fb5c-4228-aed4-61e80baef9be] Process exited with code 0
2025-08-01 07:10:53.453 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3d06ba00-5e41-4539-9729-c311cb7335ba] socks connection closed
2025-08-01 07:10:53.453 [info] [command][83bad43b-fb5c-4228-aed4-61e80baef9be] Socket close event received
2025-08-01 07:10:53.492 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55556 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:11:53.463 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:11:53.465 [info] [command][1a6a951e-02b2-40b7-a96e-282b4c8030e7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1a6a951e-02b2-40b7-a96e-282b4c8030e7""}
2025-08-01 07:11:53.465 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9fbc9bfb-f69c-4e46-8e30-04cd84e629fa] received connection request
2025-08-01 07:11:53.465 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:11:53.562 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9fbc9bfb-f69c-4e46-8e30-04cd84e629fa] socks forwarding established
2025-08-01 07:11:53.609 [info] [command][1a6a951e-02b2-40b7-a96e-282b4c8030e7] Process exited with code 0
2025-08-01 07:11:53.609 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9fbc9bfb-f69c-4e46-8e30-04cd84e629fa] socks connection closed
2025-08-01 07:11:53.609 [info] [command][1a6a951e-02b2-40b7-a96e-282b4c8030e7] Socket close event received
2025-08-01 07:11:53.650 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55575 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:12:53.613 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:12:53.615 [info] [command][132ac898-cb20-4f32-b893-fcb1d373eb98] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""132ac898-cb20-4f32-b893-fcb1d373eb98""}
2025-08-01 07:12:53.616 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4bccd7a2-14b6-4d15-9080-49222382558a] received connection request
2025-08-01 07:12:53.617 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:12:53.667 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bccd7a2-14b6-4d15-9080-49222382558a] socks forwarding established
2025-08-01 07:12:53.712 [info] [command][132ac898-cb20-4f32-b893-fcb1d373eb98] Process exited with code 0
2025-08-01 07:12:53.712 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4bccd7a2-14b6-4d15-9080-49222382558a] socks connection closed
2025-08-01 07:12:53.713 [info] [command][132ac898-cb20-4f32-b893-fcb1d373eb98] Socket close event received
2025-08-01 07:12:53.753 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55604 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:13:53.715 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:13:53.717 [info] [command][34e36ec3-acb8-44d1-8771-efc35cbf13c7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""34e36ec3-acb8-44d1-8771-efc35cbf13c7""}
2025-08-01 07:13:53.717 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][831a67be-3ee9-440b-afb3-2a28d9cfc21b] received connection request
2025-08-01 07:13:53.718 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:13:53.764 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][831a67be-3ee9-440b-afb3-2a28d9cfc21b] socks forwarding established
2025-08-01 07:13:53.810 [info] [command][34e36ec3-acb8-44d1-8771-efc35cbf13c7] Process exited with code 0
2025-08-01 07:13:53.811 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][831a67be-3ee9-440b-afb3-2a28d9cfc21b] socks connection closed
2025-08-01 07:13:53.811 [info] [command][34e36ec3-acb8-44d1-8771-efc35cbf13c7] Socket close event received
2025-08-01 07:13:53.852 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55636 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:14:53.817 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:14:53.818 [info] [command][2538ae8c-b3a4-427a-ab25-afec13f907d6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2538ae8c-b3a4-427a-ab25-afec13f907d6""}
2025-08-01 07:14:53.819 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][660f3908-ddf7-463e-b80d-4f627dd7f3d3] received connection request
2025-08-01 07:14:53.820 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:14:53.862 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][660f3908-ddf7-463e-b80d-4f627dd7f3d3] socks forwarding established
2025-08-01 07:14:53.904 [info] [command][2538ae8c-b3a4-427a-ab25-afec13f907d6] Process exited with code 0
2025-08-01 07:14:53.904 [info] [command][2538ae8c-b3a4-427a-ab25-afec13f907d6] Socket close event received
2025-08-01 07:14:53.905 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][660f3908-ddf7-463e-b80d-4f627dd7f3d3] socks connection closed
2025-08-01 07:14:53.944 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55656 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:15:53.911 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:15:53.913 [info] [command][37705f81-307b-415f-9e20-1b2f2a34bdc6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""37705f81-307b-415f-9e20-1b2f2a34bdc6""}
2025-08-01 07:15:53.914 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e4faaaa0-3af3-4b6f-a963-c7bd67034e62] received connection request
2025-08-01 07:15:53.914 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:15:53.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4faaaa0-3af3-4b6f-a963-c7bd67034e62] socks forwarding established
2025-08-01 07:15:54.008 [info] [command][37705f81-307b-415f-9e20-1b2f2a34bdc6] Process exited with code 0
2025-08-01 07:15:54.008 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4faaaa0-3af3-4b6f-a963-c7bd67034e62] socks connection closed
2025-08-01 07:15:54.008 [info] [command][37705f81-307b-415f-9e20-1b2f2a34bdc6] Socket close event received
2025-08-01 07:15:54.048 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55673 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:16:54.016 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:16:54.019 [info] [command][74242844-e868-4ef0-a3b5-df784dd93038] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""74242844-e868-4ef0-a3b5-df784dd93038""}
2025-08-01 07:16:54.020 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f63251a2-6078-4d0a-b989-fd854e97166b] received connection request
2025-08-01 07:16:54.020 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:16:54.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f63251a2-6078-4d0a-b989-fd854e97166b] socks forwarding established
2025-08-01 07:16:54.111 [info] [command][74242844-e868-4ef0-a3b5-df784dd93038] Process exited with code 0
2025-08-01 07:16:54.111 [info] [command][74242844-e868-4ef0-a3b5-df784dd93038] Socket close event received
2025-08-01 07:16:54.150 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f63251a2-6078-4d0a-b989-fd854e97166b] socks connection closed
2025-08-01 07:16:54.154 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55741 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:17:54.113 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:17:54.115 [info] [command][c2f84def-460d-4d96-b25d-31f85c1e24a6] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c2f84def-460d-4d96-b25d-31f85c1e24a6""}
2025-08-01 07:17:54.116 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dff02421-32d2-4e22-9150-ebad2ff737e1] received connection request
2025-08-01 07:17:54.116 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:17:54.157 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dff02421-32d2-4e22-9150-ebad2ff737e1] socks forwarding established
2025-08-01 07:17:54.202 [info] [command][c2f84def-460d-4d96-b25d-31f85c1e24a6] Process exited with code 0
2025-08-01 07:17:54.203 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dff02421-32d2-4e22-9150-ebad2ff737e1] socks connection closed
2025-08-01 07:17:54.203 [info] [command][c2f84def-460d-4d96-b25d-31f85c1e24a6] Socket close event received
2025-08-01 07:17:54.242 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55770 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:18:54.210 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:18:54.213 [info] [command][27538c24-d08e-456f-a630-666261519898] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""27538c24-d08e-456f-a630-666261519898""}
2025-08-01 07:18:54.213 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][39b46a06-9bca-45f8-844d-fbac889f92a7] received connection request
2025-08-01 07:18:54.214 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:18:54.258 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39b46a06-9bca-45f8-844d-fbac889f92a7] socks forwarding established
2025-08-01 07:18:54.303 [info] [command][27538c24-d08e-456f-a630-666261519898] Process exited with code 0
2025-08-01 07:18:54.303 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39b46a06-9bca-45f8-844d-fbac889f92a7] socks connection closed
2025-08-01 07:18:54.303 [info] [command][27538c24-d08e-456f-a630-666261519898] Socket close event received
2025-08-01 07:18:54.346 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55802 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:19:54.311 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:19:54.313 [info] [command][6fda734e-c374-4e27-85c5-f29db510c3d4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6fda734e-c374-4e27-85c5-f29db510c3d4""}
2025-08-01 07:19:54.314 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][95e15c50-9ac0-4881-9fb2-5b60b6e5150a] received connection request
2025-08-01 07:19:54.314 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:19:54.356 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][95e15c50-9ac0-4881-9fb2-5b60b6e5150a] socks forwarding established
2025-08-01 07:19:54.401 [info] [command][6fda734e-c374-4e27-85c5-f29db510c3d4] Process exited with code 0
2025-08-01 07:19:54.401 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][95e15c50-9ac0-4881-9fb2-5b60b6e5150a] socks connection closed
2025-08-01 07:19:54.401 [info] [command][6fda734e-c374-4e27-85c5-f29db510c3d4] Socket close event received
2025-08-01 07:19:54.442 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55821 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:20:54.401 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:20:54.403 [info] [command][7bbb4e72-09c1-4f6f-84e8-61cbdcb5ae14] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7bbb4e72-09c1-4f6f-84e8-61cbdcb5ae14""}
2025-08-01 07:20:54.404 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7187b9ba-dbb2-4390-ba0c-c61ac11b0427] received connection request
2025-08-01 07:20:54.405 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:20:54.450 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7187b9ba-dbb2-4390-ba0c-c61ac11b0427] socks forwarding established
2025-08-01 07:20:54.499 [info] [command][7bbb4e72-09c1-4f6f-84e8-61cbdcb5ae14] Process exited with code 0
2025-08-01 07:20:54.499 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7187b9ba-dbb2-4390-ba0c-c61ac11b0427] socks connection closed
2025-08-01 07:20:54.499 [info] [command][7bbb4e72-09c1-4f6f-84e8-61cbdcb5ae14] Socket close event received
2025-08-01 07:20:54.540 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55834 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:21:54.503 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:21:54.504 [info] [command][9f469ab9-2669-43c2-9cfc-c4b0502498df] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9f469ab9-2669-43c2-9cfc-c4b0502498df""}
2025-08-01 07:21:54.504 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][20ad22ca-4dcf-465f-b11c-c806f0665ccd] received connection request
2025-08-01 07:21:54.505 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:21:54.548 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][20ad22ca-4dcf-465f-b11c-c806f0665ccd] socks forwarding established
2025-08-01 07:21:54.596 [info] [command][9f469ab9-2669-43c2-9cfc-c4b0502498df] Process exited with code 0
2025-08-01 07:21:54.596 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][20ad22ca-4dcf-465f-b11c-c806f0665ccd] socks connection closed
2025-08-01 07:21:54.596 [info] [command][9f469ab9-2669-43c2-9cfc-c4b0502498df] Socket close event received
2025-08-01 07:21:54.636 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55847 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:22:54.601 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:22:54.602 [info] [command][103548ec-5945-4480-9cc8-5866dd898730] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""103548ec-5945-4480-9cc8-5866dd898730""}
2025-08-01 07:22:54.602 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7b606aff-14ee-47fe-9d44-da79e1a757c4] received connection request
2025-08-01 07:22:54.602 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:22:54.651 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7b606aff-14ee-47fe-9d44-da79e1a757c4] socks forwarding established
2025-08-01 07:22:54.694 [info] [command][103548ec-5945-4480-9cc8-5866dd898730] Process exited with code 0
2025-08-01 07:22:54.695 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7b606aff-14ee-47fe-9d44-da79e1a757c4] socks connection closed
2025-08-01 07:22:54.695 [info] [command][103548ec-5945-4480-9cc8-5866dd898730] Socket close event received
2025-08-01 07:22:54.733 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55873 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:23:54.695 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:23:54.696 [info] [command][7becb108-a9b2-468c-a0aa-c419def31b18] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""7becb108-a9b2-468c-a0aa-c419def31b18""}
2025-08-01 07:23:54.697 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4db20b84-d7de-4490-8daa-50ca077224aa] received connection request
2025-08-01 07:23:54.697 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:23:54.744 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4db20b84-d7de-4490-8daa-50ca077224aa] socks forwarding established
2025-08-01 07:23:54.789 [info] [command][7becb108-a9b2-468c-a0aa-c419def31b18] Process exited with code 0
2025-08-01 07:23:54.790 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4db20b84-d7de-4490-8daa-50ca077224aa] socks connection closed
2025-08-01 07:23:54.790 [info] [command][7becb108-a9b2-468c-a0aa-c419def31b18] Socket close event received
2025-08-01 07:23:54.840 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55899 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:24:54.801 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:24:54.802 [info] [command][501adc00-830e-44c1-a6a1-486103285537] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""501adc00-830e-44c1-a6a1-486103285537""}
2025-08-01 07:24:54.803 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d9b7671c-17fb-4a82-b217-cc7415ffaf89] received connection request
2025-08-01 07:24:54.803 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:24:54.849 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d9b7671c-17fb-4a82-b217-cc7415ffaf89] socks forwarding established
2025-08-01 07:24:54.894 [info] [command][501adc00-830e-44c1-a6a1-486103285537] Process exited with code 0
2025-08-01 07:24:54.895 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d9b7671c-17fb-4a82-b217-cc7415ffaf89] socks connection closed
2025-08-01 07:24:54.895 [info] [command][501adc00-830e-44c1-a6a1-486103285537] Socket close event received
2025-08-01 07:24:54.933 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55911 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:25:54.904 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:25:54.905 [info] [command][a54738ca-495d-47c8-85f8-93f63bef5a40] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a54738ca-495d-47c8-85f8-93f63bef5a40""}
2025-08-01 07:25:54.906 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c69ab00f-5157-44c5-af2c-6f899f1f0208] received connection request
2025-08-01 07:25:54.907 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:25:55.140 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c69ab00f-5157-44c5-af2c-6f899f1f0208] socks forwarding established
2025-08-01 07:25:55.282 [info] [command][a54738ca-495d-47c8-85f8-93f63bef5a40] Process exited with code 0
2025-08-01 07:25:55.282 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c69ab00f-5157-44c5-af2c-6f899f1f0208] socks connection closed
2025-08-01 07:25:55.282 [info] [command][a54738ca-495d-47c8-85f8-93f63bef5a40] Socket close event received
2025-08-01 07:25:55.322 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55928 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:26:55.290 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:26:55.292 [info] [command][99163490-173f-4981-b629-0d43ddb246af] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""99163490-173f-4981-b629-0d43ddb246af""}
2025-08-01 07:26:55.292 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][66f76693-d6a6-4e73-9c12-42dc4bb3ecf5] received connection request
2025-08-01 07:26:55.293 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:26:55.452 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][66f76693-d6a6-4e73-9c12-42dc4bb3ecf5] socks forwarding established
2025-08-01 07:26:55.497 [info] [command][99163490-173f-4981-b629-0d43ddb246af] Process exited with code 0
2025-08-01 07:26:55.498 [info] [command][99163490-173f-4981-b629-0d43ddb246af] Socket close event received
2025-08-01 07:26:55.513 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][66f76693-d6a6-4e73-9c12-42dc4bb3ecf5] socks connection closed
2025-08-01 07:26:55.542 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55947 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:27:55.508 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:27:55.509 [info] [command][b74752d6-52fd-47df-a15a-98e3e99e3707] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b74752d6-52fd-47df-a15a-98e3e99e3707""}
2025-08-01 07:27:55.510 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5366bda2-3370-4772-a650-e01446294dbd] received connection request
2025-08-01 07:27:55.510 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:27:55.555 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5366bda2-3370-4772-a650-e01446294dbd] socks forwarding established
2025-08-01 07:27:55.601 [info] [command][b74752d6-52fd-47df-a15a-98e3e99e3707] Process exited with code 0
2025-08-01 07:27:55.602 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5366bda2-3370-4772-a650-e01446294dbd] socks connection closed
2025-08-01 07:27:55.602 [info] [command][b74752d6-52fd-47df-a15a-98e3e99e3707] Socket close event received
2025-08-01 07:27:55.642 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 55975 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:28:55.607 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:28:55.608 [info] [command][8d9aef5c-bb25-4cba-9564-59dc61c66b1c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8d9aef5c-bb25-4cba-9564-59dc61c66b1c""}
2025-08-01 07:28:55.608 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2b5299cc-b76e-44f8-88f1-c00e46e86c0a] received connection request
2025-08-01 07:28:55.609 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:28:55.650 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2b5299cc-b76e-44f8-88f1-c00e46e86c0a] socks forwarding established
2025-08-01 07:28:55.695 [info] [command][8d9aef5c-bb25-4cba-9564-59dc61c66b1c] Process exited with code 0
2025-08-01 07:28:55.695 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2b5299cc-b76e-44f8-88f1-c00e46e86c0a] socks connection closed
2025-08-01 07:28:55.695 [info] [command][8d9aef5c-bb25-4cba-9564-59dc61c66b1c] Socket close event received
2025-08-01 07:28:55.735 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56008 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:29:55.701 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:29:55.702 [info] [command][b2ea2553-1a43-4276-a86d-b5e7282c07a2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b2ea2553-1a43-4276-a86d-b5e7282c07a2""}
2025-08-01 07:29:55.703 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6dc678a1-46b8-423a-94bc-109b0afcd6ad] received connection request
2025-08-01 07:29:55.703 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:29:55.747 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6dc678a1-46b8-423a-94bc-109b0afcd6ad] socks forwarding established
2025-08-01 07:29:55.792 [info] [command][b2ea2553-1a43-4276-a86d-b5e7282c07a2] Process exited with code 0
2025-08-01 07:29:55.792 [info] [command][b2ea2553-1a43-4276-a86d-b5e7282c07a2] Socket close event received
2025-08-01 07:29:55.793 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6dc678a1-46b8-423a-94bc-109b0afcd6ad] socks connection closed
2025-08-01 07:29:55.833 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56028 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:30:55.796 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:30:55.797 [info] [command][591b116a-b0bd-46aa-8f99-baaba4d73cbc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""591b116a-b0bd-46aa-8f99-baaba4d73cbc""}
2025-08-01 07:30:55.798 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7a63bd16-cdd1-49fe-bd06-42061364a9ac] received connection request
2025-08-01 07:30:55.798 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:30:55.912 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7a63bd16-cdd1-49fe-bd06-42061364a9ac] socks forwarding established
2025-08-01 07:30:55.958 [info] [command][591b116a-b0bd-46aa-8f99-baaba4d73cbc] Process exited with code 0
2025-08-01 07:30:55.958 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7a63bd16-cdd1-49fe-bd06-42061364a9ac] socks connection closed
2025-08-01 07:30:55.958 [info] [command][591b116a-b0bd-46aa-8f99-baaba4d73cbc] Socket close event received
2025-08-01 07:30:56.000 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56046 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:31:55.960 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:31:55.961 [info] [command][68dbad70-39a0-4582-9746-5349ed49ae89] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""68dbad70-39a0-4582-9746-5349ed49ae89""}
2025-08-01 07:31:55.962 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][485b14c6-352b-4ada-9434-7f42321b01bc] received connection request
2025-08-01 07:31:55.962 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:31:56.007 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][485b14c6-352b-4ada-9434-7f42321b01bc] socks forwarding established
2025-08-01 07:31:56.052 [info] [command][68dbad70-39a0-4582-9746-5349ed49ae89] Process exited with code 0
2025-08-01 07:31:56.052 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][485b14c6-352b-4ada-9434-7f42321b01bc] socks connection closed
2025-08-01 07:31:56.052 [info] [command][68dbad70-39a0-4582-9746-5349ed49ae89] Socket close event received
2025-08-01 07:31:56.092 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56065 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:32:56.062 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:32:56.063 [info] [command][80237501-18fb-434d-8ea0-d1a667fb352b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""80237501-18fb-434d-8ea0-d1a667fb352b""}
2025-08-01 07:32:56.064 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dd9792d4-d8c1-4b7d-82f4-c82d744f53dc] received connection request
2025-08-01 07:32:56.064 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:32:56.112 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dd9792d4-d8c1-4b7d-82f4-c82d744f53dc] socks forwarding established
2025-08-01 07:32:56.155 [info] [command][80237501-18fb-434d-8ea0-d1a667fb352b] Process exited with code 0
2025-08-01 07:32:56.155 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dd9792d4-d8c1-4b7d-82f4-c82d744f53dc] socks connection closed
2025-08-01 07:32:56.156 [info] [command][80237501-18fb-434d-8ea0-d1a667fb352b] Socket close event received
2025-08-01 07:32:56.194 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56159 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:33:56.161 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:33:56.162 [info] [command][9ffb28f5-8899-4f9c-9a87-d4e604d999eb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9ffb28f5-8899-4f9c-9a87-d4e604d999eb""}
2025-08-01 07:33:56.163 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f106a1eb-76d8-463d-8458-a289ca803747] received connection request
2025-08-01 07:33:56.164 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:33:56.210 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f106a1eb-76d8-463d-8458-a289ca803747] socks forwarding established
2025-08-01 07:33:56.256 [info] [command][9ffb28f5-8899-4f9c-9a87-d4e604d999eb] Process exited with code 0
2025-08-01 07:33:56.256 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f106a1eb-76d8-463d-8458-a289ca803747] socks connection closed
2025-08-01 07:33:56.256 [info] [command][9ffb28f5-8899-4f9c-9a87-d4e604d999eb] Socket close event received
2025-08-01 07:33:56.298 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56188 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:34:56.257 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:34:56.259 [info] [command][9909346d-6920-45ee-a493-01fe1674c8dd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9909346d-6920-45ee-a493-01fe1674c8dd""}
2025-08-01 07:34:56.259 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e6a2978e-f013-4de6-937a-bea06caf101a] received connection request
2025-08-01 07:34:56.260 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:34:56.308 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e6a2978e-f013-4de6-937a-bea06caf101a] socks forwarding established
2025-08-01 07:34:56.353 [info] [command][9909346d-6920-45ee-a493-01fe1674c8dd] Process exited with code 0
2025-08-01 07:34:56.354 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e6a2978e-f013-4de6-937a-bea06caf101a] socks connection closed
2025-08-01 07:34:56.354 [info] [command][9909346d-6920-45ee-a493-01fe1674c8dd] Socket close event received
2025-08-01 07:34:56.396 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56201 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:35:56.360 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:35:56.361 [info] [command][1f61cad4-4e56-4a29-88e7-2752437bd47b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1f61cad4-4e56-4a29-88e7-2752437bd47b""}
2025-08-01 07:35:56.362 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a027bc1a-47cb-46d4-b878-9902b6b98dbb] received connection request
2025-08-01 07:35:56.362 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:35:56.405 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a027bc1a-47cb-46d4-b878-9902b6b98dbb] socks forwarding established
2025-08-01 07:35:56.472 [info] [command][1f61cad4-4e56-4a29-88e7-2752437bd47b] Process exited with code 0
2025-08-01 07:35:56.472 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a027bc1a-47cb-46d4-b878-9902b6b98dbb] socks connection closed
2025-08-01 07:35:56.472 [info] [command][1f61cad4-4e56-4a29-88e7-2752437bd47b] Socket close event received
2025-08-01 07:35:56.513 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56215 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:36:56.478 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:36:56.479 [info] [command][fb8df01f-fc7b-4a1f-8a1d-ebee91d43095] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fb8df01f-fc7b-4a1f-8a1d-ebee91d43095""}
2025-08-01 07:36:56.479 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][8161a151-ad96-4c68-b515-7ea473fcb881] received connection request
2025-08-01 07:36:56.480 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:36:56.528 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8161a151-ad96-4c68-b515-7ea473fcb881] socks forwarding established
2025-08-01 07:36:56.572 [info] [command][fb8df01f-fc7b-4a1f-8a1d-ebee91d43095] Process exited with code 0
2025-08-01 07:36:56.573 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][8161a151-ad96-4c68-b515-7ea473fcb881] socks connection closed
2025-08-01 07:36:56.573 [info] [command][fb8df01f-fc7b-4a1f-8a1d-ebee91d43095] Socket close event received
2025-08-01 07:36:56.612 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56233 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:37:56.583 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:37:56.584 [info] [command][4d3927bf-49d9-4dec-aee7-36abf8b0d9d0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4d3927bf-49d9-4dec-aee7-36abf8b0d9d0""}
2025-08-01 07:37:56.585 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d40c16a7-12d9-4799-a7e9-34e9d773c63e] received connection request
2025-08-01 07:37:56.585 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:37:56.634 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d40c16a7-12d9-4799-a7e9-34e9d773c63e] socks forwarding established
2025-08-01 07:37:56.678 [info] [command][4d3927bf-49d9-4dec-aee7-36abf8b0d9d0] Process exited with code 0
2025-08-01 07:37:56.678 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d40c16a7-12d9-4799-a7e9-34e9d773c63e] socks connection closed
2025-08-01 07:37:56.678 [info] [command][4d3927bf-49d9-4dec-aee7-36abf8b0d9d0] Socket close event received
2025-08-01 07:37:56.718 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56263 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:38:56.688 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:38:56.689 [info] [command][bace9763-bf9c-4b9a-8462-2ed399576ebe] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bace9763-bf9c-4b9a-8462-2ed399576ebe""}
2025-08-01 07:38:56.690 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][82348324-9d15-47fd-b39e-27a23abca98e] received connection request
2025-08-01 07:38:56.690 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:38:56.734 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82348324-9d15-47fd-b39e-27a23abca98e] socks forwarding established
2025-08-01 07:38:56.781 [info] [command][bace9763-bf9c-4b9a-8462-2ed399576ebe] Process exited with code 0
2025-08-01 07:38:56.782 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][82348324-9d15-47fd-b39e-27a23abca98e] socks connection closed
2025-08-01 07:38:56.782 [info] [command][bace9763-bf9c-4b9a-8462-2ed399576ebe] Socket close event received
2025-08-01 07:38:56.823 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56301 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:39:56.784 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:39:56.785 [info] [command][a7f03d49-d394-4ef8-bf60-5f0c315f7cd7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a7f03d49-d394-4ef8-bf60-5f0c315f7cd7""}
2025-08-01 07:39:56.786 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ee73859b-7544-4a75-8027-a482810fef12] received connection request
2025-08-01 07:39:56.786 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:39:56.834 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee73859b-7544-4a75-8027-a482810fef12] socks forwarding established
2025-08-01 07:39:56.880 [info] [command][a7f03d49-d394-4ef8-bf60-5f0c315f7cd7] Process exited with code 0
2025-08-01 07:39:56.880 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee73859b-7544-4a75-8027-a482810fef12] socks connection closed
2025-08-01 07:39:56.880 [info] [command][a7f03d49-d394-4ef8-bf60-5f0c315f7cd7] Socket close event received
2025-08-01 07:39:56.922 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56335 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:40:56.881 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:40:56.882 [info] [command][3d14e3af-370d-4ba4-8ff8-d285b31124be] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3d14e3af-370d-4ba4-8ff8-d285b31124be""}
2025-08-01 07:40:56.882 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6c4f47ae-963c-413e-890c-123d1a689a62] received connection request
2025-08-01 07:40:56.883 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 07:40:56.883 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:40:57.045 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c4f47ae-963c-413e-890c-123d1a689a62] socks forwarding established
2025-08-01 07:40:57.090 [info] [command][3d14e3af-370d-4ba4-8ff8-d285b31124be] Process exited with code 0
2025-08-01 07:40:57.090 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6c4f47ae-963c-413e-890c-123d1a689a62] socks connection closed
2025-08-01 07:40:57.091 [info] [command][3d14e3af-370d-4ba4-8ff8-d285b31124be] Socket close event received
2025-08-01 07:40:57.132 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56349 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:41:57.092 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:41:57.094 [info] [command][3023c3f7-d18f-40f1-9f61-d2e52a18fcf5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3023c3f7-d18f-40f1-9f61-d2e52a18fcf5""}
2025-08-01 07:41:57.094 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3c95f67e-6325-4049-9f5c-aa0d4acbc19e] received connection request
2025-08-01 07:41:57.095 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:41:57.140 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3c95f67e-6325-4049-9f5c-aa0d4acbc19e] socks forwarding established
2025-08-01 07:41:57.186 [info] [command][3023c3f7-d18f-40f1-9f61-d2e52a18fcf5] Process exited with code 0
2025-08-01 07:41:57.186 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3c95f67e-6325-4049-9f5c-aa0d4acbc19e] socks connection closed
2025-08-01 07:41:57.186 [info] [command][3023c3f7-d18f-40f1-9f61-d2e52a18fcf5] Socket close event received
2025-08-01 07:41:57.228 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56369 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:42:57.191 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:42:57.192 [info] [command][282226f3-b0a3-4a03-9677-540718bcff6b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""282226f3-b0a3-4a03-9677-540718bcff6b""}
2025-08-01 07:42:57.193 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e1e43e74-6ca9-4aae-8262-8ccd291b8581] received connection request
2025-08-01 07:42:57.193 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:42:57.238 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e1e43e74-6ca9-4aae-8262-8ccd291b8581] socks forwarding established
2025-08-01 07:42:57.282 [info] [command][282226f3-b0a3-4a03-9677-540718bcff6b] Process exited with code 0
2025-08-01 07:42:57.283 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e1e43e74-6ca9-4aae-8262-8ccd291b8581] socks connection closed
2025-08-01 07:42:57.283 [info] [command][282226f3-b0a3-4a03-9677-540718bcff6b] Socket close event received
2025-08-01 07:42:57.321 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56406 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:43:57.282 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:43:57.284 [info] [command][11c0ebdf-155b-42ad-9332-afe0f592c51b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""11c0ebdf-155b-42ad-9332-afe0f592c51b""}
2025-08-01 07:43:57.284 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f44bb9f0-d490-4360-8c10-15b2de9902f8] received connection request
2025-08-01 07:43:57.285 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:43:57.326 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f44bb9f0-d490-4360-8c10-15b2de9902f8] socks forwarding established
2025-08-01 07:43:57.383 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f44bb9f0-d490-4360-8c10-15b2de9902f8] socks connection closed
2025-08-01 07:43:57.383 [info] [command][11c0ebdf-155b-42ad-9332-afe0f592c51b] Process exited with code 0
2025-08-01 07:43:57.383 [info] [command][11c0ebdf-155b-42ad-9332-afe0f592c51b] Socket close event received
2025-08-01 07:43:57.423 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56452 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:44:57.393 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:44:57.394 [info] [command][85a9c468-95a7-46cf-b66d-17a9cc703acb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""85a9c468-95a7-46cf-b66d-17a9cc703acb""}
2025-08-01 07:44:57.395 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][39d077bc-3749-4a4c-b066-da7b48aafca6] received connection request
2025-08-01 07:44:57.396 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:44:57.502 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39d077bc-3749-4a4c-b066-da7b48aafca6] socks forwarding established
2025-08-01 07:44:57.641 [info] [command][85a9c468-95a7-46cf-b66d-17a9cc703acb] Process exited with code 0
2025-08-01 07:44:57.642 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][39d077bc-3749-4a4c-b066-da7b48aafca6] socks connection closed
2025-08-01 07:44:57.642 [info] [command][85a9c468-95a7-46cf-b66d-17a9cc703acb] Socket close event received
2025-08-01 07:44:57.886 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56470 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:45:57.642 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:45:57.643 [info] [command][8c35dd4d-fba9-4f30-8aaa-665caa36341f] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8c35dd4d-fba9-4f30-8aaa-665caa36341f""}
2025-08-01 07:45:57.644 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][62acc6b4-069b-4b84-931a-5f9e95f752fe] received connection request
2025-08-01 07:45:57.644 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:45:57.686 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62acc6b4-069b-4b84-931a-5f9e95f752fe] socks forwarding established
2025-08-01 07:45:57.730 [info] [command][8c35dd4d-fba9-4f30-8aaa-665caa36341f] Process exited with code 0
2025-08-01 07:45:57.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62acc6b4-069b-4b84-931a-5f9e95f752fe] socks connection closed
2025-08-01 07:45:57.731 [info] [command][8c35dd4d-fba9-4f30-8aaa-665caa36341f] Socket close event received
2025-08-01 07:45:57.770 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56491 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:46:57.732 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:46:57.735 [info] [command][4ff3cfed-7b65-47e5-8ded-ca5b4f02cd8c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4ff3cfed-7b65-47e5-8ded-ca5b4f02cd8c""}
2025-08-01 07:46:57.735 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e8129abe-687c-46ae-bdf1-1820c0fc9e55] received connection request
2025-08-01 07:46:57.736 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:46:57.785 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e8129abe-687c-46ae-bdf1-1820c0fc9e55] socks forwarding established
2025-08-01 07:46:57.830 [info] [command][4ff3cfed-7b65-47e5-8ded-ca5b4f02cd8c] Process exited with code 0
2025-08-01 07:46:57.831 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e8129abe-687c-46ae-bdf1-1820c0fc9e55] socks connection closed
2025-08-01 07:46:57.831 [info] [command][4ff3cfed-7b65-47e5-8ded-ca5b4f02cd8c] Socket close event received
2025-08-01 07:46:57.870 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56523 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:47:57.833 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:47:57.834 [info] [command][f2055b44-c5d4-4948-8e84-fe620c904711] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f2055b44-c5d4-4948-8e84-fe620c904711""}
2025-08-01 07:47:57.834 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d4bc01fd-ee60-4917-be13-4ea526adf16d] received connection request
2025-08-01 07:47:57.834 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:47:57.876 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d4bc01fd-ee60-4917-be13-4ea526adf16d] socks forwarding established
2025-08-01 07:47:57.921 [info] [command][f2055b44-c5d4-4948-8e84-fe620c904711] Process exited with code 0
2025-08-01 07:47:57.922 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d4bc01fd-ee60-4917-be13-4ea526adf16d] socks connection closed
2025-08-01 07:47:57.922 [info] [command][f2055b44-c5d4-4948-8e84-fe620c904711] Socket close event received
2025-08-01 07:47:57.962 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56551 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:48:57.929 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:48:57.931 [info] [command][1223b94f-60af-4511-9cf6-b91aac6ae2ba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1223b94f-60af-4511-9cf6-b91aac6ae2ba""}
2025-08-01 07:48:57.932 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b2350b42-f2c3-46d8-8595-b2a508667de9] received connection request
2025-08-01 07:48:57.932 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:48:57.974 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2350b42-f2c3-46d8-8595-b2a508667de9] socks forwarding established
2025-08-01 07:48:58.021 [info] [command][1223b94f-60af-4511-9cf6-b91aac6ae2ba] Process exited with code 0
2025-08-01 07:48:58.022 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b2350b42-f2c3-46d8-8595-b2a508667de9] socks connection closed
2025-08-01 07:48:58.022 [info] [command][1223b94f-60af-4511-9cf6-b91aac6ae2ba] Socket close event received
2025-08-01 07:48:58.068 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56631 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:49:58.023 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:49:58.024 [info] [command][4d73e162-3221-4547-897f-136fe5fa32ec] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4d73e162-3221-4547-897f-136fe5fa32ec""}
2025-08-01 07:49:58.025 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c05fd456-4a14-405d-b156-bc7b2a2f5781] received connection request
2025-08-01 07:49:58.025 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:49:58.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c05fd456-4a14-405d-b156-bc7b2a2f5781] socks forwarding established
2025-08-01 07:49:58.116 [info] [command][4d73e162-3221-4547-897f-136fe5fa32ec] Process exited with code 0
2025-08-01 07:49:58.116 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c05fd456-4a14-405d-b156-bc7b2a2f5781] socks connection closed
2025-08-01 07:49:58.116 [info] [command][4d73e162-3221-4547-897f-136fe5fa32ec] Socket close event received
2025-08-01 07:49:58.158 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56664 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:50:58.123 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:50:58.126 [info] [command][bebf39f6-5ba6-4c2b-b3d1-460ba666ea8a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bebf39f6-5ba6-4c2b-b3d1-460ba666ea8a""}
2025-08-01 07:50:58.126 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e2bbc445-8a0a-4b08-986e-fd1ce2ff38cf] received connection request
2025-08-01 07:50:58.127 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:50:58.171 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e2bbc445-8a0a-4b08-986e-fd1ce2ff38cf] socks forwarding established
2025-08-01 07:50:58.218 [info] [command][bebf39f6-5ba6-4c2b-b3d1-460ba666ea8a] Process exited with code 0
2025-08-01 07:50:58.218 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e2bbc445-8a0a-4b08-986e-fd1ce2ff38cf] socks connection closed
2025-08-01 07:50:58.219 [info] [command][bebf39f6-5ba6-4c2b-b3d1-460ba666ea8a] Socket close event received
2025-08-01 07:50:58.259 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56681 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:51:58.223 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:51:58.225 [info] [command][59611f90-f426-4844-8a3c-105f22fce50a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""59611f90-f426-4844-8a3c-105f22fce50a""}
2025-08-01 07:51:58.225 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][77ddb978-f2e0-46f5-8959-0e4bb56b47ac] received connection request
2025-08-01 07:51:58.226 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:51:58.272 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][77ddb978-f2e0-46f5-8959-0e4bb56b47ac] socks forwarding established
2025-08-01 07:51:58.318 [info] [command][59611f90-f426-4844-8a3c-105f22fce50a] Process exited with code 0
2025-08-01 07:51:58.318 [info] [command][59611f90-f426-4844-8a3c-105f22fce50a] Socket close event received
2025-08-01 07:51:58.319 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][77ddb978-f2e0-46f5-8959-0e4bb56b47ac] socks connection closed
2025-08-01 07:51:58.359 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56709 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:52:58.327 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:52:58.329 [info] [command][74a9c488-9223-4a6e-928a-bfe7f73dc72b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""74a9c488-9223-4a6e-928a-bfe7f73dc72b""}
2025-08-01 07:52:58.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d7e3437b-ef3c-419d-93a7-2f9f9d95d057] received connection request
2025-08-01 07:52:58.331 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:52:58.376 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d7e3437b-ef3c-419d-93a7-2f9f9d95d057] socks forwarding established
2025-08-01 07:52:58.420 [info] [command][74a9c488-9223-4a6e-928a-bfe7f73dc72b] Process exited with code 0
2025-08-01 07:52:58.421 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d7e3437b-ef3c-419d-93a7-2f9f9d95d057] socks connection closed
2025-08-01 07:52:58.421 [info] [command][74a9c488-9223-4a6e-928a-bfe7f73dc72b] Socket close event received
2025-08-01 07:52:58.460 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56751 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:53:58.429 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:53:58.431 [info] [command][41855fe8-3443-425c-a2f1-33fec5dc19ad] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""41855fe8-3443-425c-a2f1-33fec5dc19ad""}
2025-08-01 07:53:58.431 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5662c430-c406-4bde-b66f-dfa87339f414] received connection request
2025-08-01 07:53:58.432 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:53:58.627 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5662c430-c406-4bde-b66f-dfa87339f414] socks forwarding established
2025-08-01 07:53:58.762 [info] [command][41855fe8-3443-425c-a2f1-33fec5dc19ad] Process exited with code 0
2025-08-01 07:53:58.763 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5662c430-c406-4bde-b66f-dfa87339f414] socks connection closed
2025-08-01 07:53:58.763 [info] [command][41855fe8-3443-425c-a2f1-33fec5dc19ad] Socket close event received
2025-08-01 07:53:58.895 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56861 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:54:58.769 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:54:58.770 [info] [command][a08adf55-de42-4cfd-8029-1cc7983ed620] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a08adf55-de42-4cfd-8029-1cc7983ed620""}
2025-08-01 07:54:58.770 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][683c70f2-a92f-4c84-8df0-7474c6a57f8e] received connection request
2025-08-01 07:54:58.771 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:54:58.848 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][683c70f2-a92f-4c84-8df0-7474c6a57f8e] socks forwarding established
2025-08-01 07:54:58.894 [info] [command][a08adf55-de42-4cfd-8029-1cc7983ed620] Process exited with code 0
2025-08-01 07:54:58.894 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][683c70f2-a92f-4c84-8df0-7474c6a57f8e] socks connection closed
2025-08-01 07:54:58.894 [info] [command][a08adf55-de42-4cfd-8029-1cc7983ed620] Socket close event received
2025-08-01 07:54:59.009 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56879 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:55:58.901 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:55:58.903 [info] [command][2641cd46-4a70-4861-b1ed-ac5907a79126] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2641cd46-4a70-4861-b1ed-ac5907a79126""}
2025-08-01 07:55:58.904 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][58aa312c-114a-4587-8d0b-ad07aadb134e] received connection request
2025-08-01 07:55:58.904 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:55:58.947 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58aa312c-114a-4587-8d0b-ad07aadb134e] socks forwarding established
2025-08-01 07:55:58.991 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58aa312c-114a-4587-8d0b-ad07aadb134e] socks connection closed
2025-08-01 07:55:58.991 [info] [command][2641cd46-4a70-4861-b1ed-ac5907a79126] Process exited with code 0
2025-08-01 07:55:58.991 [info] [command][2641cd46-4a70-4861-b1ed-ac5907a79126] Socket close event received
2025-08-01 07:55:59.030 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56890 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:56:59.001 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:56:59.003 [info] [command][fc75cfdb-6779-44c7-ba6e-ecfe7019b296] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""fc75cfdb-6779-44c7-ba6e-ecfe7019b296""}
2025-08-01 07:56:59.004 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1b2dc255-aa63-47a4-a6cf-2606d662c610] received connection request
2025-08-01 07:56:59.004 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:56:59.045 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1b2dc255-aa63-47a4-a6cf-2606d662c610] socks forwarding established
2025-08-01 07:56:59.089 [info] [command][fc75cfdb-6779-44c7-ba6e-ecfe7019b296] Process exited with code 0
2025-08-01 07:56:59.090 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1b2dc255-aa63-47a4-a6cf-2606d662c610] socks connection closed
2025-08-01 07:56:59.090 [info] [command][fc75cfdb-6779-44c7-ba6e-ecfe7019b296] Socket close event received
2025-08-01 07:56:59.129 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56904 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:57:59.097 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:57:59.098 [info] [command][5b3fc91f-4680-464a-bfb4-ac2fe96b8fa1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5b3fc91f-4680-464a-bfb4-ac2fe96b8fa1""}
2025-08-01 07:57:59.098 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c196c61f-a686-4667-9360-26a51770d0ef] received connection request
2025-08-01 07:57:59.099 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:57:59.140 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c196c61f-a686-4667-9360-26a51770d0ef] socks forwarding established
2025-08-01 07:57:59.185 [info] [command][5b3fc91f-4680-464a-bfb4-ac2fe96b8fa1] Process exited with code 0
2025-08-01 07:57:59.185 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c196c61f-a686-4667-9360-26a51770d0ef] socks connection closed
2025-08-01 07:57:59.185 [info] [command][5b3fc91f-4680-464a-bfb4-ac2fe96b8fa1] Socket close event received
2025-08-01 07:57:59.225 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56929 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:58:59.190 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:58:59.192 [info] [command][1d6a9aac-ec3d-4c3d-a305-a3be2a0e5404] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1d6a9aac-ec3d-4c3d-a305-a3be2a0e5404""}
2025-08-01 07:58:59.193 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][85203037-9ef8-4fc1-941f-9e3b18605cac] received connection request
2025-08-01 07:58:59.193 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:58:59.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85203037-9ef8-4fc1-941f-9e3b18605cac] socks forwarding established
2025-08-01 07:58:59.285 [info] [command][1d6a9aac-ec3d-4c3d-a305-a3be2a0e5404] Process exited with code 0
2025-08-01 07:58:59.286 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][85203037-9ef8-4fc1-941f-9e3b18605cac] socks connection closed
2025-08-01 07:58:59.286 [info] [command][1d6a9aac-ec3d-4c3d-a305-a3be2a0e5404] Socket close event received
2025-08-01 07:58:59.348 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56959 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 07:59:59.291 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 07:59:59.292 [info] [command][12626df2-1acf-40c8-ad39-52752f5e605b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""12626df2-1acf-40c8-ad39-52752f5e605b""}
2025-08-01 07:59:59.293 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2c533ae8-55a0-4983-aaed-08e3bc562f35] received connection request
2025-08-01 07:59:59.293 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 07:59:59.334 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c533ae8-55a0-4983-aaed-08e3bc562f35] socks forwarding established
2025-08-01 07:59:59.381 [info] [command][12626df2-1acf-40c8-ad39-52752f5e605b] Process exited with code 0
2025-08-01 07:59:59.382 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2c533ae8-55a0-4983-aaed-08e3bc562f35] socks connection closed
2025-08-01 07:59:59.382 [info] [command][12626df2-1acf-40c8-ad39-52752f5e605b] Socket close event received
2025-08-01 07:59:59.425 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 56979 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:00:59.389 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:00:59.391 [info] [command][0462e718-ba9b-4ee6-87cb-d75bf325c815] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0462e718-ba9b-4ee6-87cb-d75bf325c815""}
2025-08-01 08:00:59.392 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][44082290-a731-4040-8a90-dfca8d092d1f] received connection request
2025-08-01 08:00:59.392 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:00:59.468 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][44082290-a731-4040-8a90-dfca8d092d1f] socks forwarding established
2025-08-01 08:00:59.512 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][44082290-a731-4040-8a90-dfca8d092d1f] socks connection closed
2025-08-01 08:00:59.513 [info] [command][0462e718-ba9b-4ee6-87cb-d75bf325c815] Process exited with code 0
2025-08-01 08:00:59.513 [info] [command][0462e718-ba9b-4ee6-87cb-d75bf325c815] Socket close event received
2025-08-01 08:00:59.577 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57004 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:01:59.522 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:01:59.523 [info] [command][2fcb7b03-75fd-418d-a233-19ca454ad3f9] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2fcb7b03-75fd-418d-a233-19ca454ad3f9""}
2025-08-01 08:01:59.523 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][67234454-d2cd-4a15-9cf3-c511045efc10] received connection request
2025-08-01 08:01:59.523 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:01:59.569 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67234454-d2cd-4a15-9cf3-c511045efc10] socks forwarding established
2025-08-01 08:01:59.614 [info] [command][2fcb7b03-75fd-418d-a233-19ca454ad3f9] Process exited with code 0
2025-08-01 08:01:59.614 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][67234454-d2cd-4a15-9cf3-c511045efc10] socks connection closed
2025-08-01 08:01:59.614 [info] [command][2fcb7b03-75fd-418d-a233-19ca454ad3f9] Socket close event received
2025-08-01 08:01:59.653 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57014 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:02:59.620 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:02:59.621 [info] [command][f220c59a-4c86-490a-b9fa-e6704a1b1208] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f220c59a-4c86-490a-b9fa-e6704a1b1208""}
2025-08-01 08:02:59.622 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][dd0901b1-13ef-4623-8d6f-79de149dd2d7] received connection request
2025-08-01 08:02:59.623 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:02:59.681 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dd0901b1-13ef-4623-8d6f-79de149dd2d7] socks forwarding established
2025-08-01 08:02:59.726 [info] [command][f220c59a-4c86-490a-b9fa-e6704a1b1208] Process exited with code 0
2025-08-01 08:02:59.726 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][dd0901b1-13ef-4623-8d6f-79de149dd2d7] socks connection closed
2025-08-01 08:02:59.726 [info] [command][f220c59a-4c86-490a-b9fa-e6704a1b1208] Socket close event received
2025-08-01 08:02:59.766 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57045 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:03:59.731 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:03:59.733 [info] [command][6a9feb99-f327-4081-96c3-62fc60383ac5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6a9feb99-f327-4081-96c3-62fc60383ac5""}
2025-08-01 08:03:59.734 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][40981935-0dbb-4805-8757-10e1daf5692f] received connection request
2025-08-01 08:03:59.734 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:03:59.914 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][40981935-0dbb-4805-8757-10e1daf5692f] socks forwarding established
2025-08-01 08:03:59.963 [info] [command][6a9feb99-f327-4081-96c3-62fc60383ac5] Process exited with code 0
2025-08-01 08:03:59.963 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][40981935-0dbb-4805-8757-10e1daf5692f] socks connection closed
2025-08-01 08:03:59.963 [info] [command][6a9feb99-f327-4081-96c3-62fc60383ac5] Socket close event received
2025-08-01 08:04:00.010 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57075 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:04:59.972 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:04:59.975 [info] [command][5b83564b-66fa-4afe-b5e1-55d23f5ff200] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5b83564b-66fa-4afe-b5e1-55d23f5ff200""}
2025-08-01 08:04:59.976 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ab5ac198-a503-439b-82bb-46f1d73d8825] received connection request
2025-08-01 08:04:59.976 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:05:00.023 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab5ac198-a503-439b-82bb-46f1d73d8825] socks forwarding established
2025-08-01 08:05:00.068 [info] [command][5b83564b-66fa-4afe-b5e1-55d23f5ff200] Process exited with code 0
2025-08-01 08:05:00.068 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ab5ac198-a503-439b-82bb-46f1d73d8825] socks connection closed
2025-08-01 08:05:00.068 [info] [command][5b83564b-66fa-4afe-b5e1-55d23f5ff200] Socket close event received
2025-08-01 08:05:00.112 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57092 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:06:00.072 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:06:00.073 [info] [command][4ba04e69-f4fe-4ffe-93cb-aac58fd0047b] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""4ba04e69-f4fe-4ffe-93cb-aac58fd0047b""}
2025-08-01 08:06:00.074 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][59604cc6-24bb-4a5a-a105-dc7fd3c29ead] received connection request
2025-08-01 08:06:00.074 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:06:00.118 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][59604cc6-24bb-4a5a-a105-dc7fd3c29ead] socks forwarding established
2025-08-01 08:06:00.163 [info] [command][4ba04e69-f4fe-4ffe-93cb-aac58fd0047b] Process exited with code 0
2025-08-01 08:06:00.163 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][59604cc6-24bb-4a5a-a105-dc7fd3c29ead] socks connection closed
2025-08-01 08:06:00.163 [info] [command][4ba04e69-f4fe-4ffe-93cb-aac58fd0047b] Socket close event received
2025-08-01 08:06:00.208 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57106 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:07:00.164 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:07:00.166 [info] [command][1ac10e59-d53a-4b68-82b7-00fb141c6806] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1ac10e59-d53a-4b68-82b7-00fb141c6806""}
2025-08-01 08:07:00.167 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f4bdfe78-7022-4f8c-aacf-f568472254f0] received connection request
2025-08-01 08:07:00.168 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:07:00.223 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4bdfe78-7022-4f8c-aacf-f568472254f0] socks forwarding established
2025-08-01 08:07:00.269 [info] [command][1ac10e59-d53a-4b68-82b7-00fb141c6806] Process exited with code 0
2025-08-01 08:07:00.269 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f4bdfe78-7022-4f8c-aacf-f568472254f0] socks connection closed
2025-08-01 08:07:00.269 [info] [command][1ac10e59-d53a-4b68-82b7-00fb141c6806] Socket close event received
2025-08-01 08:07:00.312 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57135 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:08:00.274 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:08:00.275 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][58368738-98d5-48dd-b6b3-7f58523acda2] received connection request
2025-08-01 08:08:00.276 [info] [command][cd45150c-de31-46a3-ba74-3537f48c6e8d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""cd45150c-de31-46a3-ba74-3537f48c6e8d""}
2025-08-01 08:08:00.276 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:08:00.316 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58368738-98d5-48dd-b6b3-7f58523acda2] socks forwarding established
2025-08-01 08:08:00.362 [info] [command][cd45150c-de31-46a3-ba74-3537f48c6e8d] Process exited with code 0
2025-08-01 08:08:00.363 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][58368738-98d5-48dd-b6b3-7f58523acda2] socks connection closed
2025-08-01 08:08:00.363 [info] [command][cd45150c-de31-46a3-ba74-3537f48c6e8d] Socket close event received
2025-08-01 08:08:00.404 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57166 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:09:00.365 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:09:00.367 [info] [command][5c45cab9-65ad-4961-a19e-f0c3a619d111] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5c45cab9-65ad-4961-a19e-f0c3a619d111""}
2025-08-01 08:09:00.367 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a1a6f923-02e8-4f27-9c9e-cb3bbfe1e8d5] received connection request
2025-08-01 08:09:00.367 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:09:00.424 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1a6f923-02e8-4f27-9c9e-cb3bbfe1e8d5] socks forwarding established
2025-08-01 08:09:00.472 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a1a6f923-02e8-4f27-9c9e-cb3bbfe1e8d5] socks connection closed
2025-08-01 08:09:00.473 [info] [command][5c45cab9-65ad-4961-a19e-f0c3a619d111] Process exited with code 0
2025-08-01 08:09:00.473 [info] [command][5c45cab9-65ad-4961-a19e-f0c3a619d111] Socket close event received
2025-08-01 08:09:00.748 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57200 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:10:00.473 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:10:00.475 [info] [command][e44fa87b-028f-452c-8f54-9ce3d836fc77] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e44fa87b-028f-452c-8f54-9ce3d836fc77""}
2025-08-01 08:10:00.476 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9ea91854-3321-4ca3-8ea9-2277e6b34297] received connection request
2025-08-01 08:10:00.476 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:10:00.517 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9ea91854-3321-4ca3-8ea9-2277e6b34297] socks forwarding established
2025-08-01 08:10:00.562 [info] [command][e44fa87b-028f-452c-8f54-9ce3d836fc77] Process exited with code 0
2025-08-01 08:10:00.562 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9ea91854-3321-4ca3-8ea9-2277e6b34297] socks connection closed
2025-08-01 08:10:00.562 [info] [command][e44fa87b-028f-452c-8f54-9ce3d836fc77] Socket close event received
2025-08-01 08:10:00.601 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57233 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:11:00.570 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:11:00.572 [info] [command][991c561f-f3f3-451d-9efd-1cb2fd9c493d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""991c561f-f3f3-451d-9efd-1cb2fd9c493d""}
2025-08-01 08:11:00.573 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][c3613be2-1a18-41eb-97fa-f26e9a5707a7] received connection request
2025-08-01 08:11:00.573 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:11:00.614 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3613be2-1a18-41eb-97fa-f26e9a5707a7] socks forwarding established
2025-08-01 08:11:00.660 [info] [command][991c561f-f3f3-451d-9efd-1cb2fd9c493d] Process exited with code 0
2025-08-01 08:11:00.660 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][c3613be2-1a18-41eb-97fa-f26e9a5707a7] socks connection closed
2025-08-01 08:11:00.661 [info] [command][991c561f-f3f3-451d-9efd-1cb2fd9c493d] Socket close event received
2025-08-01 08:11:00.702 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57251 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:12:00.661 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:12:00.663 [info] [command][66b08bb4-7b67-40f8-ab55-7709f04199c2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""66b08bb4-7b67-40f8-ab55-7709f04199c2""}
2025-08-01 08:12:00.664 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0bb5be0b-200a-4795-9909-14e2514668c3] received connection request
2025-08-01 08:12:00.664 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:12:00.711 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0bb5be0b-200a-4795-9909-14e2514668c3] socks forwarding established
2025-08-01 08:12:00.757 [info] [command][66b08bb4-7b67-40f8-ab55-7709f04199c2] Process exited with code 0
2025-08-01 08:12:00.758 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0bb5be0b-200a-4795-9909-14e2514668c3] socks connection closed
2025-08-01 08:12:00.758 [info] [command][66b08bb4-7b67-40f8-ab55-7709f04199c2] Socket close event received
2025-08-01 08:12:00.796 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57266 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:13:00.760 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:13:00.762 [info] [command][ebbabffe-c19a-44f4-897c-9c6b016a553d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ebbabffe-c19a-44f4-897c-9c6b016a553d""}
2025-08-01 08:13:00.763 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][09714689-56aa-4614-9328-bea6d7dfb28d] received connection request
2025-08-01 08:13:00.763 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:13:00.872 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][09714689-56aa-4614-9328-bea6d7dfb28d] socks forwarding established
2025-08-01 08:13:00.938 [info] [command][ebbabffe-c19a-44f4-897c-9c6b016a553d] Process exited with code 0
2025-08-01 08:13:00.938 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][09714689-56aa-4614-9328-bea6d7dfb28d] socks connection closed
2025-08-01 08:13:00.938 [info] [command][ebbabffe-c19a-44f4-897c-9c6b016a553d] Socket close event received
2025-08-01 08:13:00.980 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57309 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:14:00.948 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:14:00.950 [info] [command][33e8a218-a05a-43c6-8bd8-6d424b1d628c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""33e8a218-a05a-43c6-8bd8-6d424b1d628c""}
2025-08-01 08:14:00.951 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1c557dcb-c3be-4c5b-8fdf-4619083f9a07] received connection request
2025-08-01 08:14:00.952 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:14:00.995 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c557dcb-c3be-4c5b-8fdf-4619083f9a07] socks forwarding established
2025-08-01 08:14:01.040 [info] [command][33e8a218-a05a-43c6-8bd8-6d424b1d628c] Process exited with code 0
2025-08-01 08:14:01.040 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c557dcb-c3be-4c5b-8fdf-4619083f9a07] socks connection closed
2025-08-01 08:14:01.040 [info] [command][33e8a218-a05a-43c6-8bd8-6d424b1d628c] Socket close event received
2025-08-01 08:14:01.082 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57344 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:15:01.045 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:15:01.047 [info] [command][a0b39b90-fa52-439d-8584-4a6b1591e2d2] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a0b39b90-fa52-439d-8584-4a6b1591e2d2""}
2025-08-01 08:15:01.048 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][13f25455-78ea-410d-a41e-fcd5c54606b9] received connection request
2025-08-01 08:15:01.048 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:15:01.093 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13f25455-78ea-410d-a41e-fcd5c54606b9] socks forwarding established
2025-08-01 08:15:01.137 [info] [command][a0b39b90-fa52-439d-8584-4a6b1591e2d2] Process exited with code 0
2025-08-01 08:15:01.137 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][13f25455-78ea-410d-a41e-fcd5c54606b9] socks connection closed
2025-08-01 08:15:01.137 [info] [command][a0b39b90-fa52-439d-8584-4a6b1591e2d2] Socket close event received
2025-08-01 08:15:01.176 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57357 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:16:01.140 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:16:01.141 [info] [command][ff91c953-9d21-4a24-ab3c-8ad7ebac2154] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""ff91c953-9d21-4a24-ab3c-8ad7ebac2154""}
2025-08-01 08:16:01.141 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][4c2e9cb6-e5d7-4479-b9d8-cb88bf3c8b18] received connection request
2025-08-01 08:16:01.142 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:16:01.186 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4c2e9cb6-e5d7-4479-b9d8-cb88bf3c8b18] socks forwarding established
2025-08-01 08:16:01.228 [info] [command][ff91c953-9d21-4a24-ab3c-8ad7ebac2154] Process exited with code 0
2025-08-01 08:16:01.228 [info] [command][ff91c953-9d21-4a24-ab3c-8ad7ebac2154] Socket close event received
2025-08-01 08:16:01.229 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][4c2e9cb6-e5d7-4479-b9d8-cb88bf3c8b18] socks connection closed
2025-08-01 08:16:01.270 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57370 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:17:01.236 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:17:01.239 [info] [command][312034d1-0695-4a85-8c1f-aacb00cf3aa3] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""312034d1-0695-4a85-8c1f-aacb00cf3aa3""}
2025-08-01 08:17:01.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][33bfddd6-2581-4676-ac52-5f2363a9365f] received connection request
2025-08-01 08:17:01.240 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:17:01.288 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33bfddd6-2581-4676-ac52-5f2363a9365f] socks forwarding established
2025-08-01 08:17:01.333 [info] [command][312034d1-0695-4a85-8c1f-aacb00cf3aa3] Process exited with code 0
2025-08-01 08:17:01.333 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][33bfddd6-2581-4676-ac52-5f2363a9365f] socks connection closed
2025-08-01 08:17:01.333 [info] [command][312034d1-0695-4a85-8c1f-aacb00cf3aa3] Socket close event received
2025-08-01 08:17:01.373 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57391 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:18:01.339 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:18:01.340 [info] [command][8f259d2c-cf42-4ed3-a595-b557ce0d65aa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8f259d2c-cf42-4ed3-a595-b557ce0d65aa""}
2025-08-01 08:18:01.341 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3dd505de-1f22-4fde-83ed-f458d3d46d20] received connection request
2025-08-01 08:18:01.341 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:18:01.405 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3dd505de-1f22-4fde-83ed-f458d3d46d20] socks forwarding established
2025-08-01 08:18:01.615 [info] [command][8f259d2c-cf42-4ed3-a595-b557ce0d65aa] Process exited with code 0
2025-08-01 08:18:01.615 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3dd505de-1f22-4fde-83ed-f458d3d46d20] socks connection closed
2025-08-01 08:18:01.615 [info] [command][8f259d2c-cf42-4ed3-a595-b557ce0d65aa] Socket close event received
2025-08-01 08:18:01.654 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57428 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:19:01.625 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:19:01.627 [info] [command][f689f591-4c8e-4b8d-a4e7-fe9fd18c4dd0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f689f591-4c8e-4b8d-a4e7-fe9fd18c4dd0""}
2025-08-01 08:19:01.628 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9014928d-3677-42a9-b6d1-535d61e01a6a] received connection request
2025-08-01 08:19:01.628 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:19:01.676 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9014928d-3677-42a9-b6d1-535d61e01a6a] socks forwarding established
2025-08-01 08:19:01.720 [info] [command][f689f591-4c8e-4b8d-a4e7-fe9fd18c4dd0] Process exited with code 0
2025-08-01 08:19:01.720 [info] [command][f689f591-4c8e-4b8d-a4e7-fe9fd18c4dd0] Socket close event received
2025-08-01 08:19:01.759 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9014928d-3677-42a9-b6d1-535d61e01a6a] socks connection closed
2025-08-01 08:19:01.759 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57499 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:20:01.728 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:20:01.730 [info] [command][bef74917-669b-4c1d-b462-80ac47563027] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""bef74917-669b-4c1d-b462-80ac47563027""}
2025-08-01 08:20:01.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][0c2e67c2-5299-40c0-b674-fce4418ce234] received connection request
2025-08-01 08:20:01.732 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:20:01.775 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c2e67c2-5299-40c0-b674-fce4418ce234] socks forwarding established
2025-08-01 08:20:01.819 [info] [command][bef74917-669b-4c1d-b462-80ac47563027] Process exited with code 0
2025-08-01 08:20:01.819 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][0c2e67c2-5299-40c0-b674-fce4418ce234] socks connection closed
2025-08-01 08:20:01.820 [info] [command][bef74917-669b-4c1d-b462-80ac47563027] Socket close event received
2025-08-01 08:20:01.859 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57515 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:21:01.823 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:21:01.826 [info] [command][326adad6-503e-4b3b-993c-0bcbb6501bfc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""326adad6-503e-4b3b-993c-0bcbb6501bfc""}
2025-08-01 08:21:01.826 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b311641f-72ff-4df9-adf4-6f0f82a0e4cc] received connection request
2025-08-01 08:21:01.827 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:21:01.874 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b311641f-72ff-4df9-adf4-6f0f82a0e4cc] socks forwarding established
2025-08-01 08:21:01.919 [info] [command][326adad6-503e-4b3b-993c-0bcbb6501bfc] Process exited with code 0
2025-08-01 08:21:01.920 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b311641f-72ff-4df9-adf4-6f0f82a0e4cc] socks connection closed
2025-08-01 08:21:01.920 [info] [command][326adad6-503e-4b3b-993c-0bcbb6501bfc] Socket close event received
2025-08-01 08:21:01.965 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57539 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:22:01.923 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:22:01.924 [info] [command][c1b9cb96-40c6-4f9a-a4d6-7b5cbc1fc591] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c1b9cb96-40c6-4f9a-a4d6-7b5cbc1fc591""}
2025-08-01 08:22:01.924 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][76a7b2ba-9754-4573-8edf-7d0ac1280aeb] received connection request
2025-08-01 08:22:01.925 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:22:02.036 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76a7b2ba-9754-4573-8edf-7d0ac1280aeb] socks forwarding established
2025-08-01 08:22:02.170 [info] [command][c1b9cb96-40c6-4f9a-a4d6-7b5cbc1fc591] Process exited with code 0
2025-08-01 08:22:02.170 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76a7b2ba-9754-4573-8edf-7d0ac1280aeb] socks connection closed
2025-08-01 08:22:02.170 [info] [command][c1b9cb96-40c6-4f9a-a4d6-7b5cbc1fc591] Socket close event received
2025-08-01 08:22:02.209 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57555 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:23:02.174 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:23:02.175 [info] [command][47f4ac16-105b-4865-99c4-3670d05ecc1a] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""47f4ac16-105b-4865-99c4-3670d05ecc1a""}
2025-08-01 08:23:02.176 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6e37b865-4798-4462-b775-ca248b08f194] received connection request
2025-08-01 08:23:02.177 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:23:02.264 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6e37b865-4798-4462-b775-ca248b08f194] socks forwarding established
2025-08-01 08:23:02.312 [info] [command][47f4ac16-105b-4865-99c4-3670d05ecc1a] Process exited with code 0
2025-08-01 08:23:02.313 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6e37b865-4798-4462-b775-ca248b08f194] socks connection closed
2025-08-01 08:23:02.313 [info] [command][47f4ac16-105b-4865-99c4-3670d05ecc1a] Socket close event received
2025-08-01 08:23:02.353 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57586 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:24:02.323 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:24:02.324 [info] [command][53163760-474b-4331-99dd-aa1b1bec9921] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""53163760-474b-4331-99dd-aa1b1bec9921""}
2025-08-01 08:24:02.325 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][9b6aa8d6-335d-4f25-af54-8ee2608846e9] received connection request
2025-08-01 08:24:02.325 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:24:02.374 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9b6aa8d6-335d-4f25-af54-8ee2608846e9] socks forwarding established
2025-08-01 08:24:02.421 [info] [command][53163760-474b-4331-99dd-aa1b1bec9921] Process exited with code 0
2025-08-01 08:24:02.422 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][9b6aa8d6-335d-4f25-af54-8ee2608846e9] socks connection closed
2025-08-01 08:24:02.422 [info] [command][53163760-474b-4331-99dd-aa1b1bec9921] Socket close event received
2025-08-01 08:24:02.463 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57615 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:25:02.432 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:25:02.433 [info] [command][1756ed8d-ef03-490d-bc28-98997e9312c8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""1756ed8d-ef03-490d-bc28-98997e9312c8""}
2025-08-01 08:25:02.433 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][55eb3bf1-adfb-4d42-a60c-d3eb7dc3c6fc] received connection request
2025-08-01 08:25:02.434 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:25:02.480 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][55eb3bf1-adfb-4d42-a60c-d3eb7dc3c6fc] socks forwarding established
2025-08-01 08:25:02.524 [info] [command][1756ed8d-ef03-490d-bc28-98997e9312c8] Process exited with code 0
2025-08-01 08:25:02.524 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][55eb3bf1-adfb-4d42-a60c-d3eb7dc3c6fc] socks connection closed
2025-08-01 08:25:02.524 [info] [command][1756ed8d-ef03-490d-bc28-98997e9312c8] Socket close event received
2025-08-01 08:25:02.567 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57631 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:26:02.531 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:26:02.532 [info] [command][33414f95-62f5-431a-b5f8-260ac4fe92a0] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""33414f95-62f5-431a-b5f8-260ac4fe92a0""}
2025-08-01 08:26:02.532 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][a3c43ec9-f671-4299-a572-85a6b7037e06] received connection request
2025-08-01 08:26:02.533 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:26:02.578 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a3c43ec9-f671-4299-a572-85a6b7037e06] socks forwarding established
2025-08-01 08:26:02.624 [info] [command][33414f95-62f5-431a-b5f8-260ac4fe92a0] Process exited with code 0
2025-08-01 08:26:02.624 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][a3c43ec9-f671-4299-a572-85a6b7037e06] socks connection closed
2025-08-01 08:26:02.624 [info] [command][33414f95-62f5-431a-b5f8-260ac4fe92a0] Socket close event received
2025-08-01 08:26:02.664 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57644 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:27:02.634 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:27:02.635 [info] [command][d4fe05d9-739c-4b14-ba3f-ca80962aeefc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d4fe05d9-739c-4b14-ba3f-ca80962aeefc""}
2025-08-01 08:27:02.636 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e4c5438a-a604-4bf5-8de4-c20aa427b1a0] received connection request
2025-08-01 08:27:02.636 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:27:02.682 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4c5438a-a604-4bf5-8de4-c20aa427b1a0] socks forwarding established
2025-08-01 08:27:02.729 [info] [command][d4fe05d9-739c-4b14-ba3f-ca80962aeefc] Process exited with code 0
2025-08-01 08:27:02.729 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e4c5438a-a604-4bf5-8de4-c20aa427b1a0] socks connection closed
2025-08-01 08:27:02.729 [info] [command][d4fe05d9-739c-4b14-ba3f-ca80962aeefc] Socket close event received
2025-08-01 08:27:02.770 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57662 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:28:02.738 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:28:02.739 [info] [command][15d0344f-273b-4d1a-8f9f-c82978618b12] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""15d0344f-273b-4d1a-8f9f-c82978618b12""}
2025-08-01 08:28:02.740 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][466cf5d7-4a69-4c43-9c59-a2b92cd60334] received connection request
2025-08-01 08:28:02.740 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:28:02.783 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][466cf5d7-4a69-4c43-9c59-a2b92cd60334] socks forwarding established
2025-08-01 08:28:02.829 [info] [command][15d0344f-273b-4d1a-8f9f-c82978618b12] Process exited with code 0
2025-08-01 08:28:02.829 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][466cf5d7-4a69-4c43-9c59-a2b92cd60334] socks connection closed
2025-08-01 08:28:02.829 [info] [command][15d0344f-273b-4d1a-8f9f-c82978618b12] Socket close event received
2025-08-01 08:28:02.870 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57689 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:29:02.839 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:29:02.840 [info] [command][0cf4a153-750e-49d6-b310-a5e5de20eabd] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0cf4a153-750e-49d6-b310-a5e5de20eabd""}
2025-08-01 08:29:02.841 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][facb29a7-a616-4c6f-b98e-455b306ebaf0] received connection request
2025-08-01 08:29:02.842 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:29:02.885 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][facb29a7-a616-4c6f-b98e-455b306ebaf0] socks forwarding established
2025-08-01 08:29:02.929 [info] [command][0cf4a153-750e-49d6-b310-a5e5de20eabd] Process exited with code 0
2025-08-01 08:29:02.930 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][facb29a7-a616-4c6f-b98e-455b306ebaf0] socks connection closed
2025-08-01 08:29:02.930 [info] [command][0cf4a153-750e-49d6-b310-a5e5de20eabd] Socket close event received
2025-08-01 08:29:02.969 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57756 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:30:02.934 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:30:02.935 [info] [command][d2d0fd06-92db-452b-a4a9-0c3268de6ffe] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""d2d0fd06-92db-452b-a4a9-0c3268de6ffe""}
2025-08-01 08:30:02.936 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][db25ef8b-3d27-498f-b96b-fba540db3cbf] received connection request
2025-08-01 08:30:02.936 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:30:02.984 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db25ef8b-3d27-498f-b96b-fba540db3cbf] socks forwarding established
2025-08-01 08:30:03.028 [info] [command][d2d0fd06-92db-452b-a4a9-0c3268de6ffe] Process exited with code 0
2025-08-01 08:30:03.028 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][db25ef8b-3d27-498f-b96b-fba540db3cbf] socks connection closed
2025-08-01 08:30:03.028 [info] [command][d2d0fd06-92db-452b-a4a9-0c3268de6ffe] Socket close event received
2025-08-01 08:30:03.069 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57776 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:31:03.035 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:31:03.036 [info] [command][5275cad6-1f22-47d6-b57f-3cdead75ecdf] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5275cad6-1f22-47d6-b57f-3cdead75ecdf""}
2025-08-01 08:31:03.036 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][97502a0c-c650-4997-844e-6a839a3f5b4e] received connection request
2025-08-01 08:31:03.037 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:31:03.084 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97502a0c-c650-4997-844e-6a839a3f5b4e] socks forwarding established
2025-08-01 08:31:03.131 [info] [command][5275cad6-1f22-47d6-b57f-3cdead75ecdf] Process exited with code 0
2025-08-01 08:31:03.131 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][97502a0c-c650-4997-844e-6a839a3f5b4e] socks connection closed
2025-08-01 08:31:03.131 [info] [command][5275cad6-1f22-47d6-b57f-3cdead75ecdf] Socket close event received
2025-08-01 08:31:03.171 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57792 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:32:03.142 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:32:03.142 [info] [command][f17e1286-0dc7-4272-9ab8-4c307696e9b5] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f17e1286-0dc7-4272-9ab8-4c307696e9b5""}
2025-08-01 08:32:03.143 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][f211298c-4ed3-462e-988c-f7b0a12923ca] received connection request
2025-08-01 08:32:03.144 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:32:03.189 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f211298c-4ed3-462e-988c-f7b0a12923ca] socks forwarding established
2025-08-01 08:32:03.236 [info] [command][f17e1286-0dc7-4272-9ab8-4c307696e9b5] Process exited with code 0
2025-08-01 08:32:03.236 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][f211298c-4ed3-462e-988c-f7b0a12923ca] socks connection closed
2025-08-01 08:32:03.236 [info] [command][f17e1286-0dc7-4272-9ab8-4c307696e9b5] Socket close event received
2025-08-01 08:32:03.293 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57804 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:33:03.238 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:33:03.238 [info] [command][5df1ce17-dddc-46d7-b115-f928857d6e58] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""5df1ce17-dddc-46d7-b115-f928857d6e58""}
2025-08-01 08:33:03.239 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][da2df10d-e2e9-4e75-ac13-89cc0a325009] received connection request
2025-08-01 08:33:03.239 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:33:03.285 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][da2df10d-e2e9-4e75-ac13-89cc0a325009] socks forwarding established
2025-08-01 08:33:03.330 [info] [command][5df1ce17-dddc-46d7-b115-f928857d6e58] Process exited with code 0
2025-08-01 08:33:03.330 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][da2df10d-e2e9-4e75-ac13-89cc0a325009] socks connection closed
2025-08-01 08:33:03.330 [info] [command][5df1ce17-dddc-46d7-b115-f928857d6e58] Socket close event received
2025-08-01 08:33:03.371 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57844 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:34:03.338 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:34:03.339 [info] [command][b7e1a2d6-2cca-4557-94a4-0328131324ec] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7e1a2d6-2cca-4557-94a4-0328131324ec""}
2025-08-01 08:34:03.340 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5370ba6c-0543-40a8-a8dd-5fae5a3b73df] received connection request
2025-08-01 08:34:03.341 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:34:03.381 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5370ba6c-0543-40a8-a8dd-5fae5a3b73df] socks forwarding established
2025-08-01 08:34:03.425 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5370ba6c-0543-40a8-a8dd-5fae5a3b73df] socks connection closed
2025-08-01 08:34:03.425 [info] [command][b7e1a2d6-2cca-4557-94a4-0328131324ec] Process exited with code 0
2025-08-01 08:34:03.426 [info] [command][b7e1a2d6-2cca-4557-94a4-0328131324ec] Socket close event received
2025-08-01 08:34:03.464 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57872 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:35:03.432 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:35:03.433 [info] [command][b8ec52db-c079-492d-97b2-274d2375ee2d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b8ec52db-c079-492d-97b2-274d2375ee2d""}
2025-08-01 08:35:03.434 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1c612a0b-3419-4a57-abd8-51bbba608863] received connection request
2025-08-01 08:35:03.434 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:35:03.516 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c612a0b-3419-4a57-abd8-51bbba608863] socks forwarding established
2025-08-01 08:35:03.561 [info] [command][b8ec52db-c079-492d-97b2-274d2375ee2d] Process exited with code 0
2025-08-01 08:35:03.561 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c612a0b-3419-4a57-abd8-51bbba608863] socks connection closed
2025-08-01 08:35:03.561 [info] [command][b8ec52db-c079-492d-97b2-274d2375ee2d] Socket close event received
2025-08-01 08:35:03.599 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57892 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:36:03.568 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:36:03.569 [info] [command][24977a10-65d5-4614-bf68-0923c33aacd4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""24977a10-65d5-4614-bf68-0923c33aacd4""}
2025-08-01 08:36:03.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][e9234fd7-d9d9-441c-9ab0-5e766868d9f2] received connection request
2025-08-01 08:36:03.571 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:36:03.612 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9234fd7-d9d9-441c-9ab0-5e766868d9f2] socks forwarding established
2025-08-01 08:36:03.740 [info] [command][24977a10-65d5-4614-bf68-0923c33aacd4] Process exited with code 0
2025-08-01 08:36:03.740 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][e9234fd7-d9d9-441c-9ab0-5e766868d9f2] socks connection closed
2025-08-01 08:36:03.740 [info] [command][24977a10-65d5-4614-bf68-0923c33aacd4] Socket close event received
2025-08-01 08:36:03.781 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57908 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:37:03.741 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:37:03.743 [info] [command][b7ef8807-27bc-4545-bd39-d30801ef9322] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b7ef8807-27bc-4545-bd39-d30801ef9322""}
2025-08-01 08:37:03.744 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6b635b13-3e19-472c-8b65-9b8f95349d7e] received connection request
2025-08-01 08:37:03.744 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:37:03.789 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b635b13-3e19-472c-8b65-9b8f95349d7e] socks forwarding established
2025-08-01 08:37:03.853 [info] [command][b7ef8807-27bc-4545-bd39-d30801ef9322] Process exited with code 0
2025-08-01 08:37:03.853 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6b635b13-3e19-472c-8b65-9b8f95349d7e] socks connection closed
2025-08-01 08:37:03.853 [info] [command][b7ef8807-27bc-4545-bd39-d30801ef9322] Socket close event received
2025-08-01 08:37:03.895 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57932 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:38:03.857 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:38:03.859 [info] [command][34f7b461-a617-44dd-8caf-d747aa5359c4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""34f7b461-a617-44dd-8caf-d747aa5359c4""}
2025-08-01 08:38:03.859 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5ef82703-1b88-4d39-b6f9-ecb5af4bb8ec] received connection request
2025-08-01 08:38:03.860 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:38:03.904 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ef82703-1b88-4d39-b6f9-ecb5af4bb8ec] socks forwarding established
2025-08-01 08:38:03.953 [info] [command][34f7b461-a617-44dd-8caf-d747aa5359c4] Process exited with code 0
2025-08-01 08:38:03.954 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5ef82703-1b88-4d39-b6f9-ecb5af4bb8ec] socks connection closed
2025-08-01 08:38:03.954 [info] [command][34f7b461-a617-44dd-8caf-d747aa5359c4] Socket close event received
2025-08-01 08:38:03.998 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57965 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:39:03.964 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:39:03.965 [info] [command][a5143162-d99e-4812-a599-f5e427b08a12] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a5143162-d99e-4812-a599-f5e427b08a12""}
2025-08-01 08:39:03.965 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bfece25b-68a8-468c-8606-9a61479bb906] received connection request
2025-08-01 08:39:03.966 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:39:04.013 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bfece25b-68a8-468c-8606-9a61479bb906] socks forwarding established
2025-08-01 08:39:04.060 [info] [command][a5143162-d99e-4812-a599-f5e427b08a12] Process exited with code 0
2025-08-01 08:39:04.060 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bfece25b-68a8-468c-8606-9a61479bb906] socks connection closed
2025-08-01 08:39:04.060 [info] [command][a5143162-d99e-4812-a599-f5e427b08a12] Socket close event received
2025-08-01 08:39:04.100 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 57999 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:40:04.069 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:40:04.070 [info] [command][81fbc6ed-6e49-41f9-94de-8a102a8fdbba] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""81fbc6ed-6e49-41f9-94de-8a102a8fdbba""}
2025-08-01 08:40:04.071 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][953f3f95-5018-49a0-b7bc-59410ef1c0ce] received connection request
2025-08-01 08:40:04.072 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:40:04.117 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][953f3f95-5018-49a0-b7bc-59410ef1c0ce] socks forwarding established
2025-08-01 08:40:04.161 [info] [command][81fbc6ed-6e49-41f9-94de-8a102a8fdbba] Process exited with code 0
2025-08-01 08:40:04.161 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][953f3f95-5018-49a0-b7bc-59410ef1c0ce] socks connection closed
2025-08-01 08:40:04.162 [info] [command][81fbc6ed-6e49-41f9-94de-8a102a8fdbba] Socket close event received
2025-08-01 08:40:04.201 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58013 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:41:04.172 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:41:04.173 [info] [command][332a4637-e2e5-4b7c-af5e-34f8e56652f7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""332a4637-e2e5-4b7c-af5e-34f8e56652f7""}
2025-08-01 08:41:04.174 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][76818121-d837-4231-b72b-1850447b51d9] received connection request
2025-08-01 08:41:04.174 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:41:04.218 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76818121-d837-4231-b72b-1850447b51d9] socks forwarding established
2025-08-01 08:41:04.315 [info] [command][332a4637-e2e5-4b7c-af5e-34f8e56652f7] Process exited with code 0
2025-08-01 08:41:04.315 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][76818121-d837-4231-b72b-1850447b51d9] socks connection closed
2025-08-01 08:41:04.315 [info] [command][332a4637-e2e5-4b7c-af5e-34f8e56652f7] Socket close event received
2025-08-01 08:41:04.358 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58031 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:42:04.379 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:42:04.587 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][030dfe70-e40d-4a26-b370-96e2934b6174] received connection request
2025-08-01 08:42:04.587 [info] [command][794f8b7c-e6e9-4aa6-aa20-78f1436bad78] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""794f8b7c-e6e9-4aa6-aa20-78f1436bad78""}
2025-08-01 08:42:04.681 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:42:04.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][030dfe70-e40d-4a26-b370-96e2934b6174] socks forwarding established
2025-08-01 08:42:04.880 [info] [command][794f8b7c-e6e9-4aa6-aa20-78f1436bad78] Process exited with code 0
2025-08-01 08:42:04.882 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][030dfe70-e40d-4a26-b370-96e2934b6174] socks connection closed
2025-08-01 08:42:04.882 [info] [command][794f8b7c-e6e9-4aa6-aa20-78f1436bad78] Socket close event received
2025-08-01 08:42:04.951 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58061 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:43:04.889 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:43:04.890 [info] [command][be23feea-8683-4b68-ad42-d6611be791a4] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""be23feea-8683-4b68-ad42-d6611be791a4""}
2025-08-01 08:43:04.890 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][321f8f49-860a-45ab-ad70-b3c539d5cdb3] received connection request
2025-08-01 08:43:04.890 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:43:04.931 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][321f8f49-860a-45ab-ad70-b3c539d5cdb3] socks forwarding established
2025-08-01 08:43:04.974 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][321f8f49-860a-45ab-ad70-b3c539d5cdb3] socks connection closed
2025-08-01 08:43:04.974 [info] [command][be23feea-8683-4b68-ad42-d6611be791a4] Process exited with code 0
2025-08-01 08:43:04.974 [info] [command][be23feea-8683-4b68-ad42-d6611be791a4] Socket close event received
2025-08-01 08:43:05.037 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58104 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:44:04.985 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:44:04.985 [info] [command][c60e546c-622e-4cb0-8c2c-21f84a20b67c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""c60e546c-622e-4cb0-8c2c-21f84a20b67c""}
2025-08-01 08:44:04.985 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d6f6cf92-1953-4e8c-8b05-e97c6c820f90] received connection request
2025-08-01 08:44:04.986 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 08:44:04.986 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:44:05.024 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d6f6cf92-1953-4e8c-8b05-e97c6c820f90] socks forwarding established
2025-08-01 08:44:05.067 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d6f6cf92-1953-4e8c-8b05-e97c6c820f90] socks connection closed
2025-08-01 08:44:05.068 [info] [command][c60e546c-622e-4cb0-8c2c-21f84a20b67c] Process exited with code 0
2025-08-01 08:44:05.068 [info] [command][c60e546c-622e-4cb0-8c2c-21f84a20b67c] Socket close event received
2025-08-01 08:44:05.108 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58123 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:45:05.078 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:45:05.079 [info] [command][2c36ac48-9464-4e22-83b4-1cbf7aa54951] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""2c36ac48-9464-4e22-83b4-1cbf7aa54951""}
2025-08-01 08:45:05.079 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][62300320-7680-4a9a-bfc5-d74ca942a08b] received connection request
2025-08-01 08:45:05.079 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:45:05.121 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62300320-7680-4a9a-bfc5-d74ca942a08b] socks forwarding established
2025-08-01 08:45:05.166 [info] [command][2c36ac48-9464-4e22-83b4-1cbf7aa54951] Process exited with code 0
2025-08-01 08:45:05.166 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62300320-7680-4a9a-bfc5-d74ca942a08b] socks connection closed
2025-08-01 08:45:05.166 [info] [command][2c36ac48-9464-4e22-83b4-1cbf7aa54951] Socket close event received
2025-08-01 08:45:05.205 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58177 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:46:05.175 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:46:05.177 [info] [command][26594658-43b9-4138-bced-4017c9e30b2e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""26594658-43b9-4138-bced-4017c9e30b2e""}
2025-08-01 08:46:05.177 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d25d0e9a-92f9-42de-bcaf-fafcb742bfd3] received connection request
2025-08-01 08:46:05.177 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:46:05.223 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d25d0e9a-92f9-42de-bcaf-fafcb742bfd3] socks forwarding established
2025-08-01 08:46:05.265 [info] [command][26594658-43b9-4138-bced-4017c9e30b2e] Process exited with code 0
2025-08-01 08:46:05.266 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d25d0e9a-92f9-42de-bcaf-fafcb742bfd3] socks connection closed
2025-08-01 08:46:05.266 [info] [command][26594658-43b9-4138-bced-4017c9e30b2e] Socket close event received
2025-08-01 08:46:05.394 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58192 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:47:05.278 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:47:05.279 [info] [command][59eb3b67-9f0b-4f8e-8b53-5a216584e1ad] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""59eb3b67-9f0b-4f8e-8b53-5a216584e1ad""}
2025-08-01 08:47:05.279 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][5d580099-9769-4883-b92c-5cfb93524559] received connection request
2025-08-01 08:47:05.279 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 08:47:05.280 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:47:05.319 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d580099-9769-4883-b92c-5cfb93524559] socks forwarding established
2025-08-01 08:47:05.365 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][5d580099-9769-4883-b92c-5cfb93524559] socks connection closed
2025-08-01 08:47:05.365 [info] [command][59eb3b67-9f0b-4f8e-8b53-5a216584e1ad] Process exited with code 0
2025-08-01 08:47:05.365 [info] [command][59eb3b67-9f0b-4f8e-8b53-5a216584e1ad] Socket close event received
2025-08-01 08:47:05.406 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58217 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:48:05.367 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:48:05.368 [info] [command][797f0de9-9c60-4911-a02c-2778b7f5d604] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""797f0de9-9c60-4911-a02c-2778b7f5d604""}
2025-08-01 08:48:05.368 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6fb20b07-cfe3-41fd-b63f-01f7bb77d270] received connection request
2025-08-01 08:48:05.369 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.

2025-08-01 08:48:05.369 [info] (ssh_tunnel) stderr: debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:48:05.408 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fb20b07-cfe3-41fd-b63f-01f7bb77d270] socks forwarding established
2025-08-01 08:48:05.451 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fb20b07-cfe3-41fd-b63f-01f7bb77d270] socks connection closed
2025-08-01 08:48:05.451 [info] [command][797f0de9-9c60-4911-a02c-2778b7f5d604] Process exited with code 0
2025-08-01 08:48:05.451 [info] [command][797f0de9-9c60-4911-a02c-2778b7f5d604] Socket close event received
2025-08-01 08:48:05.520 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58245 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:49:05.460 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:49:05.463 [info] [command][b71307f0-2522-4cd9-bced-dea1922bb543] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b71307f0-2522-4cd9-bced-dea1922bb543""}
2025-08-01 08:49:05.464 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][62c90ebf-f3ef-4145-90b6-4dcb305f9461] received connection request
2025-08-01 08:49:05.464 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:49:05.509 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62c90ebf-f3ef-4145-90b6-4dcb305f9461] socks forwarding established
2025-08-01 08:49:05.555 [info] [command][b71307f0-2522-4cd9-bced-dea1922bb543] Process exited with code 0
2025-08-01 08:49:05.556 [info] [command][b71307f0-2522-4cd9-bced-dea1922bb543] Socket close event received
2025-08-01 08:49:05.598 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58266 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:49:05.598 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][62c90ebf-f3ef-4145-90b6-4dcb305f9461] socks connection closed
2025-08-01 08:50:05.566 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:50:05.567 [info] [command][39e38a2a-907c-4c92-a0b3-9530067efdfc] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""39e38a2a-907c-4c92-a0b3-9530067efdfc""}
2025-08-01 08:50:05.568 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][bed5ac55-0e30-4858-b18e-b2aa758be92c] received connection request
2025-08-01 08:50:05.568 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:50:05.675 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bed5ac55-0e30-4858-b18e-b2aa758be92c] socks forwarding established
2025-08-01 08:50:05.721 [info] [command][39e38a2a-907c-4c92-a0b3-9530067efdfc] Process exited with code 0
2025-08-01 08:50:05.721 [info] [command][39e38a2a-907c-4c92-a0b3-9530067efdfc] Socket close event received
2025-08-01 08:50:05.759 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][bed5ac55-0e30-4858-b18e-b2aa758be92c] socks connection closed
2025-08-01 08:50:05.762 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58283 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:51:05.729 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:51:05.731 [info] [command][77466e94-4a47-4945-959e-e6413faaefaa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""77466e94-4a47-4945-959e-e6413faaefaa""}
2025-08-01 08:51:05.731 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][efdfd193-547c-4528-8d99-f50d0308f2fa] received connection request
2025-08-01 08:51:05.732 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:51:05.807 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efdfd193-547c-4528-8d99-f50d0308f2fa] socks forwarding established
2025-08-01 08:51:05.862 [info] [command][77466e94-4a47-4945-959e-e6413faaefaa] Process exited with code 0
2025-08-01 08:51:05.862 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][efdfd193-547c-4528-8d99-f50d0308f2fa] socks connection closed
2025-08-01 08:51:05.862 [info] [command][77466e94-4a47-4945-959e-e6413faaefaa] Socket close event received
2025-08-01 08:51:05.902 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58302 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:52:05.864 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:52:05.870 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][ee9f05eb-4b5b-4871-a849-13d0f2b083e6] received connection request
2025-08-01 08:52:05.870 [info] [command][82a01978-735c-4b15-a396-5f573d3243f7] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""82a01978-735c-4b15-a396-5f573d3243f7""}
2025-08-01 08:52:05.870 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:52:05.912 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee9f05eb-4b5b-4871-a849-13d0f2b083e6] socks forwarding established
2025-08-01 08:52:05.957 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][ee9f05eb-4b5b-4871-a849-13d0f2b083e6] socks connection closed
2025-08-01 08:52:05.957 [info] [command][82a01978-735c-4b15-a396-5f573d3243f7] Process exited with code 0
2025-08-01 08:52:05.957 [info] [command][82a01978-735c-4b15-a396-5f573d3243f7] Socket close event received
2025-08-01 08:52:05.997 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58330 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:53:05.968 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:53:05.970 [info] [command][e8479223-9d17-4964-84c3-2093eb790597] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""e8479223-9d17-4964-84c3-2093eb790597""}
2025-08-01 08:53:05.970 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][284fbbeb-6b4e-4560-95df-a509bc435dd1] received connection request
2025-08-01 08:53:05.971 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:53:06.020 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][284fbbeb-6b4e-4560-95df-a509bc435dd1] socks forwarding established
2025-08-01 08:53:06.065 [info] [command][e8479223-9d17-4964-84c3-2093eb790597] Process exited with code 0
2025-08-01 08:53:06.065 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][284fbbeb-6b4e-4560-95df-a509bc435dd1] socks connection closed
2025-08-01 08:53:06.065 [info] [command][e8479223-9d17-4964-84c3-2093eb790597] Socket close event received
2025-08-01 08:53:06.104 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58363 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:54:06.075 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:54:06.077 [info] [command][db3423fc-d8fa-4939-a216-f5fe851b405c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""db3423fc-d8fa-4939-a216-f5fe851b405c""}
2025-08-01 08:54:06.078 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][cb9a928b-0bc6-4b5f-955a-19c4761cb7c9] received connection request
2025-08-01 08:54:06.079 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:54:06.119 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb9a928b-0bc6-4b5f-955a-19c4761cb7c9] socks forwarding established
2025-08-01 08:54:06.163 [info] [command][db3423fc-d8fa-4939-a216-f5fe851b405c] Process exited with code 0
2025-08-01 08:54:06.164 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][cb9a928b-0bc6-4b5f-955a-19c4761cb7c9] socks connection closed
2025-08-01 08:54:06.164 [info] [command][db3423fc-d8fa-4939-a216-f5fe851b405c] Socket close event received
2025-08-01 08:54:06.203 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58381 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:55:06.174 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:55:06.175 [info] [command][425c296e-4132-441e-9196-b141a69bf3a8] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""425c296e-4132-441e-9196-b141a69bf3a8""}
2025-08-01 08:55:06.176 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d2873d67-ac19-4333-8ed8-d85a577fc50d] received connection request
2025-08-01 08:55:06.176 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:55:06.223 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d2873d67-ac19-4333-8ed8-d85a577fc50d] socks forwarding established
2025-08-01 08:55:06.278 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d2873d67-ac19-4333-8ed8-d85a577fc50d] socks connection closed
2025-08-01 08:55:06.278 [info] [command][425c296e-4132-441e-9196-b141a69bf3a8] Process exited with code 0
2025-08-01 08:55:06.278 [info] [command][425c296e-4132-441e-9196-b141a69bf3a8] Socket close event received
2025-08-01 08:55:06.318 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58406 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:56:06.283 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:56:06.285 [info] [command][6cfb5fbd-ac45-4b3a-ab92-ef5cf998bdcb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6cfb5fbd-ac45-4b3a-ab92-ef5cf998bdcb""}
2025-08-01 08:56:06.286 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][63b03c8e-73dc-4292-819d-55c6d38a6763] received connection request
2025-08-01 08:56:06.286 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:56:06.328 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][63b03c8e-73dc-4292-819d-55c6d38a6763] socks forwarding established
2025-08-01 08:56:06.374 [info] [command][6cfb5fbd-ac45-4b3a-ab92-ef5cf998bdcb] Process exited with code 0
2025-08-01 08:56:06.375 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][63b03c8e-73dc-4292-819d-55c6d38a6763] socks connection closed
2025-08-01 08:56:06.375 [info] [command][6cfb5fbd-ac45-4b3a-ab92-ef5cf998bdcb] Socket close event received
2025-08-01 08:56:06.414 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58417 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:57:07.372 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:57:08.391 [info] [command][0864b45e-2002-4768-98a9-4bf0a26577ef] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""0864b45e-2002-4768-98a9-4bf0a26577ef""}
2025-08-01 08:57:08.391 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1915005e-97ea-4a05-b1c9-85e85c6534f8] received connection request
2025-08-01 08:57:09.367 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:57:10.383 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1915005e-97ea-4a05-b1c9-85e85c6534f8] socks forwarding established
2025-08-01 08:57:11.356 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1915005e-97ea-4a05-b1c9-85e85c6534f8] socks connection closed
2025-08-01 08:57:11.356 [info] [command][0864b45e-2002-4768-98a9-4bf0a26577ef] Process exited with code 0
2025-08-01 08:57:11.356 [info] [command][0864b45e-2002-4768-98a9-4bf0a26577ef] Socket close event received
2025-08-01 08:57:12.275 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58496 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:58:11.361 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:58:11.363 [info] [command][8e45d6dd-7a6e-4698-8780-8c58d47dcf87] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8e45d6dd-7a6e-4698-8780-8c58d47dcf87""}
2025-08-01 08:58:11.363 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][7e552f71-c08b-47c4-93e8-e05ee0faa97c] received connection request
2025-08-01 08:58:11.364 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:58:11.411 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7e552f71-c08b-47c4-93e8-e05ee0faa97c] socks forwarding established
2025-08-01 08:58:11.466 [info] [command][8e45d6dd-7a6e-4698-8780-8c58d47dcf87] Process exited with code 0
2025-08-01 08:58:11.467 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][7e552f71-c08b-47c4-93e8-e05ee0faa97c] socks connection closed
2025-08-01 08:58:11.467 [info] [command][8e45d6dd-7a6e-4698-8780-8c58d47dcf87] Socket close event received
2025-08-01 08:58:11.506 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58622 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 08:59:11.477 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 08:59:11.478 [info] [command][3a123485-a82a-4aa0-af14-66c237244542] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""3a123485-a82a-4aa0-af14-66c237244542""}
2025-08-01 08:59:11.478 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][443ebb4f-5b05-464b-b82b-35a5e11923e9] received connection request
2025-08-01 08:59:11.478 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 08:59:11.522 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][443ebb4f-5b05-464b-b82b-35a5e11923e9] socks forwarding established
2025-08-01 08:59:11.568 [info] [command][3a123485-a82a-4aa0-af14-66c237244542] Process exited with code 0
2025-08-01 08:59:11.568 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][443ebb4f-5b05-464b-b82b-35a5e11923e9] socks connection closed
2025-08-01 08:59:11.568 [info] [command][3a123485-a82a-4aa0-af14-66c237244542] Socket close event received
2025-08-01 08:59:11.631 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58639 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:00:11.569 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:00:11.570 [info] [command][eabbf07a-bc04-46cc-86bb-cfd0e7e3c06c] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""eabbf07a-bc04-46cc-86bb-cfd0e7e3c06c""}
2025-08-01 09:00:11.570 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][b9f87926-f74b-4d09-a781-46bbf0ef791b] received connection request
2025-08-01 09:00:11.571 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:00:11.614 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9f87926-f74b-4d09-a781-46bbf0ef791b] socks forwarding established
2025-08-01 09:00:11.659 [info] [command][eabbf07a-bc04-46cc-86bb-cfd0e7e3c06c] Process exited with code 0
2025-08-01 09:00:11.659 [info] [command][eabbf07a-bc04-46cc-86bb-cfd0e7e3c06c] Socket close event received
2025-08-01 09:00:11.701 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58656 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:00:11.702 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][b9f87926-f74b-4d09-a781-46bbf0ef791b] socks connection closed
2025-08-01 09:01:11.661 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:01:11.662 [info] [command][f9ac7fb3-5a29-4fc6-ad62-407fb73053b1] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""f9ac7fb3-5a29-4fc6-ad62-407fb73053b1""}
2025-08-01 09:01:11.662 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][29dbc065-ebc6-4588-a3a8-b17186767f1a] received connection request
2025-08-01 09:01:11.663 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:01:11.705 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29dbc065-ebc6-4588-a3a8-b17186767f1a] socks forwarding established
2025-08-01 09:01:11.751 [info] [command][f9ac7fb3-5a29-4fc6-ad62-407fb73053b1] Process exited with code 0
2025-08-01 09:01:11.751 [info] [command][f9ac7fb3-5a29-4fc6-ad62-407fb73053b1] Socket close event received
2025-08-01 09:01:11.752 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][29dbc065-ebc6-4588-a3a8-b17186767f1a] socks connection closed
2025-08-01 09:01:11.793 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58680 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:02:11.751 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:02:11.752 [info] [command][a7fd9d87-0880-4ac2-8080-c7a361053bfb] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""a7fd9d87-0880-4ac2-8080-c7a361053bfb""}
2025-08-01 09:02:11.753 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][52116040-0a1d-4e08-841b-53db887d8f9a] received connection request
2025-08-01 09:02:11.753 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:02:11.796 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52116040-0a1d-4e08-841b-53db887d8f9a] socks forwarding established
2025-08-01 09:02:11.843 [info] [command][a7fd9d87-0880-4ac2-8080-c7a361053bfb] Process exited with code 0
2025-08-01 09:02:11.843 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][52116040-0a1d-4e08-841b-53db887d8f9a] socks connection closed
2025-08-01 09:02:11.843 [info] [command][a7fd9d87-0880-4ac2-8080-c7a361053bfb] Socket close event received
2025-08-01 09:02:11.883 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58764 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:03:11.845 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:03:11.847 [info] [command][433f3598-b96f-4fc2-9d9f-8a5938ccf255] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""433f3598-b96f-4fc2-9d9f-8a5938ccf255""}
2025-08-01 09:03:11.848 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][3a9d9e53-0a8a-453b-8723-d8c95957014c] received connection request
2025-08-01 09:03:11.849 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:03:11.891 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3a9d9e53-0a8a-453b-8723-d8c95957014c] socks forwarding established
2025-08-01 09:03:11.937 [info] [command][433f3598-b96f-4fc2-9d9f-8a5938ccf255] Process exited with code 0
2025-08-01 09:03:11.937 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][3a9d9e53-0a8a-453b-8723-d8c95957014c] socks connection closed
2025-08-01 09:03:11.937 [info] [command][433f3598-b96f-4fc2-9d9f-8a5938ccf255] Socket close event received
2025-08-01 09:03:11.977 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58818 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:04:11.942 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:04:11.945 [info] [command][edd61262-f6ee-4c56-ad13-0746cc478614] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""edd61262-f6ee-4c56-ad13-0746cc478614""}
2025-08-01 09:04:11.946 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][d97be02c-d434-42ae-aebe-0717a3917d57] received connection request
2025-08-01 09:04:11.946 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:04:11.994 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d97be02c-d434-42ae-aebe-0717a3917d57] socks forwarding established
2025-08-01 09:04:12.040 [info] [command][edd61262-f6ee-4c56-ad13-0746cc478614] Process exited with code 0
2025-08-01 09:04:12.040 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][d97be02c-d434-42ae-aebe-0717a3917d57] socks connection closed
2025-08-01 09:04:12.040 [info] [command][edd61262-f6ee-4c56-ad13-0746cc478614] Socket close event received
2025-08-01 09:04:12.083 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58834 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:05:12.050 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:05:12.053 [info] [command][b0b83bcd-adb6-46c9-a0ec-bdbf9d680963] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""b0b83bcd-adb6-46c9-a0ec-bdbf9d680963""}
2025-08-01 09:05:12.054 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][1c3f1dd0-568e-43df-9120-9ad1eec79744] received connection request
2025-08-01 09:05:12.054 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:05:12.097 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c3f1dd0-568e-43df-9120-9ad1eec79744] socks forwarding established
2025-08-01 09:05:12.145 [info] [command][b0b83bcd-adb6-46c9-a0ec-bdbf9d680963] Process exited with code 0
2025-08-01 09:05:12.145 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][1c3f1dd0-568e-43df-9120-9ad1eec79744] socks connection closed
2025-08-01 09:05:12.145 [info] [command][b0b83bcd-adb6-46c9-a0ec-bdbf9d680963] Socket close event received
2025-08-01 09:05:12.186 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58856 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:06:12.150 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:06:12.152 [info] [command][8d560c40-4748-45c5-9e7b-f24ae92d5a03] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""8d560c40-4748-45c5-9e7b-f24ae92d5a03""}
2025-08-01 09:06:12.153 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][2ec5a867-a216-4212-9598-c0df61a1b99f] received connection request
2025-08-01 09:06:12.153 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:06:12.194 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2ec5a867-a216-4212-9598-c0df61a1b99f] socks forwarding established
2025-08-01 09:06:12.246 [info] [command][8d560c40-4748-45c5-9e7b-f24ae92d5a03] Process exited with code 0
2025-08-01 09:06:12.247 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][2ec5a867-a216-4212-9598-c0df61a1b99f] socks connection closed
2025-08-01 09:06:12.247 [info] [command][8d560c40-4748-45c5-9e7b-f24ae92d5a03] Socket close event received
2025-08-01 09:06:12.289 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58870 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:07:12.250 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:07:12.252 [info] [command][9f8cc91f-bb73-4831-843c-632ab6a0078e] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""9f8cc91f-bb73-4831-843c-632ab6a0078e""}
2025-08-01 09:07:12.253 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][6fc74e59-6645-42a7-bb94-0e1f59a75b17] received connection request
2025-08-01 09:07:12.253 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:07:12.294 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fc74e59-6645-42a7-bb94-0e1f59a75b17] socks forwarding established
2025-08-01 09:07:12.339 [info] [command][9f8cc91f-bb73-4831-843c-632ab6a0078e] Process exited with code 0
2025-08-01 09:07:12.340 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][6fc74e59-6645-42a7-bb94-0e1f59a75b17] socks connection closed
2025-08-01 09:07:12.340 [info] [command][9f8cc91f-bb73-4831-843c-632ab6a0078e] Socket close event received
2025-08-01 09:07:12.380 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58903 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:08:12.346 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:08:12.347 [info] [command][17a5743f-55fc-47fb-9972-371830f049aa] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""17a5743f-55fc-47fb-9972-371830f049aa""}
2025-08-01 09:08:12.348 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:32959][750286e2-a75f-400e-ad03-68e692be9f38] received connection request
2025-08-01 09:08:12.349 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:08:12.399 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][750286e2-a75f-400e-ad03-68e692be9f38] socks forwarding established
2025-08-01 09:08:12.444 [info] [command][17a5743f-55fc-47fb-9972-371830f049aa] Process exited with code 0
2025-08-01 09:08:12.445 [info] [forwarding][multiplex][127.0.0.1:53505 -> 127.0.0.1:53495 -> 127.0.0.1:32959][750286e2-a75f-400e-ad03-68e692be9f38] socks connection closed
2025-08-01 09:08:12.445 [info] [command][17a5743f-55fc-47fb-9972-371830f049aa] Socket close event received
2025-08-01 09:08:12.486 [info] (ssh_tunnel) stderr: debug1: channel 5: free: direct-tcpip: listening port 53495 for 127.0.0.1 port 32959, connect from 127.0.0.1 port 58973 to 127.0.0.1 port 53495, nchannels 6

2025-08-01 09:09:00.714 [info] Resolving ssh remote authority 'login.haicore.berlin' (Unparsed 'ssh-remote+7b22686f73744e616d65223a226c6f67696e2e686169636f72652e6265726c696e227d') (attempt #2)
2025-08-01 09:09:00.715 [info] Received re-connection request; checking to see if existing connection is still valid
2025-08-01 09:09:00.725 [info] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:36751][3093f780-c8c6-44a7-89f0-2f225fda0d78] received connection request
2025-08-01 09:09:00.754 [info] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:36751][adbc35b9-2afb-406d-ac87-888576afc57e] received connection request
2025-08-01 09:09:00.778 [info] (ssh_tunnel) stderr: debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 5: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)
debug1: Connection to port 53495 forwarding to socks port 0 requested.
debug1: channel 6: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:09:10.720 [error] Unexpected error while checking if existing connection is still valid Timeout while checking if existing connection is still valid
2025-08-01 09:09:10.721 [error] Failed to connect to Cursor server at http://127.0.0.1:53504, attempt 1 of 3 This operation was aborted
2025-08-01 09:09:10.772 [info] Terminating existing SSH process with pid: 84097
2025-08-01 09:09:10.772 [info] Using configured platform linux for remote host login.haicore.berlin
2025-08-01 09:09:10.772 [info] (ssh_tunnel): exit: code=null signal=SIGKILL
2025-08-01 09:09:10.773 [info] Using askpass script: /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/launchSSHAskpass.sh with javascript file /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/sshAskClient.js. Askpass handle: /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor-ssh-tgv0nD/socket.sock
2025-08-01 09:09:10.775 [info] Launching SSH server via shell with command: cat ""/var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor_remote_install_76251.sh"" | ssh -v -T -D 58992 login.haicore.berlin bash --login -c bash
2025-08-01 09:09:10.776 [info] Started installation script. Waiting for it to finish...
2025-08-01 09:09:10.776 [info] Waiting for server to install via process(6792)...
2025-08-01 09:09:10.781 [info] (ssh_tunnel) stderr: OpenSSH_9.9p1, LibreSSL 3.3.6
debug1: Reading configuration data /Users/franzsrambical/.ssh/config

2025-08-01 09:09:10.781 [info] (ssh_tunnel) stderr: debug1: /Users/franzsrambical/.ssh/config line 22: Applying options for login.haicore.berlin
debug1: Reading configuration data /etc/ssh/ssh_config

2025-08-01 09:09:10.782 [info] (ssh_tunnel) stderr: debug1: Reading configuration data /etc/ssh/ssh_config.d/100-macos.conf
debug1: /etc/ssh/ssh_config.d/100-macos.conf line 1: Applying options for *

2025-08-01 09:09:10.782 [info] (ssh_tunnel) stderr: debug1: Reading configuration data /etc/ssh/crypto.conf

2025-08-01 09:09:10.783 [info] (ssh_tunnel) stderr: debug1: Authenticator provider $SSH_SK_PROVIDER did not resolve; disabling
debug1: Connecting to login.haicore.berlin port 22.

2025-08-01 09:09:11.729 [error] Failed to connect to Cursor server at http://127.0.0.1:53504, attempt 2 of 3 This operation was aborted
2025-08-01 09:09:12.453 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:53505
2025-08-01 09:09:12.733 [error] Failed to connect to Cursor server at http://127.0.0.1:53504, attempt 3 of 3 This operation was aborted
2025-08-01 09:09:12.733 [error] Could not re-use existing SOCKS connection; attempting to re-establish SOCKS forwarding Failed to connect to Cursor code server. Ensure that your remote host ssh config has 'AllowTcpForwarding yes' in '/etc/ssh/sshd_config'. Please check the logs and try reinstalling the server.
2025-08-01 09:09:12.733 [error] Could not re-establish SOCKS forwarding; re-establishing entire SSH connection Remote server is not set
2025-08-01 09:14:18.269 [info] [command][6247dca8-fb78-4a7e-a0e3-bf4d3f373530] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""4e2b026f-f3bc-45e2-b182-05301748651e"",""id"":""6247dca8-fb78-4a7e-a0e3-bf4d3f373530""}
2025-08-01 09:14:18.343 [error] [forwarding][code][127.0.0.1:53504 -> unknown}][a3590c0a-bda0-4345-87a3-d3e7612f806d] remote server not configured
2025-08-01 09:14:18.343 [error] [forwarding][multiplex][127.0.0.1:53505 -> unknown}][e820cc67-9a63-48c8-bb78-2daf844c3e4e] remote server not configured
2025-08-01 09:14:18.343 [error] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:53495 -> 127.0.0.1:36751][3093f780-c8c6-44a7-89f0-2f225fda0d78] error while creating socks forwarding Socket closed
2025-08-01 09:14:18.343 [error] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:53495 -> 127.0.0.1:36751][adbc35b9-2afb-406d-ac87-888576afc57e] error while creating socks forwarding Socket closed
2025-08-01 09:14:18.343 [info] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:53495 -> 127.0.0.1:36751][c1f42692-4142-4ca4-87fe-b02325efdc0f] socks connection closed
2025-08-01 09:14:18.344 [info] [forwarding][code][127.0.0.1:53504 -> 127.0.0.1:53495 -> 127.0.0.1:36751][ca5c1054-7923-423b-bac1-bbf7f4a43a78] socks connection closed
2025-08-01 09:14:18.345 [error] Error installing server: Failed to install server within the timeout
2025-08-01 09:14:18.345 [error] Failed to connect after 1 attempts: Failed to install server within the timeout
2025-08-01 09:14:18.345 [info] Deleting local script /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor_remote_install_76251.sh
2025-08-01 09:14:18.347 [info] (ssh_tunnel) stderr: ssh: connect to host login.haicore.berlin port 22: Undefined error: 0

2025-08-01 09:14:18.347 [error] [command][6247dca8-fb78-4a7e-a0e3-bf4d3f373530] Socket error: Error: read ECONNRESET
2025-08-01 09:14:18.347 [info] (ssh_tunnel): exit: code=255 signal=null
2025-08-01 09:14:18.348 [info] [command][6247dca8-fb78-4a7e-a0e3-bf4d3f373530] Socket close event received
2025-08-01 09:14:18.348 [error] Error resolving SSH authority Failed to install server within the timeout
2025-08-01 09:34:52.826 [info] Resolving ssh remote authority 'login.haicore.berlin' (Unparsed 'ssh-remote+7b22686f73744e616d65223a226c6f67696e2e686169636f72652e6265726c696e227d') (attempt #1)
2025-08-01 09:34:52.833 [info] SSH askpass server listening on /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor-ssh-N1Vsk8/socket.sock
2025-08-01 09:34:52.834 [info] Using configured platform linux for remote host login.haicore.berlin
2025-08-01 09:34:52.835 [info] Using askpass script: /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/launchSSHAskpass.sh with javascript file /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/sshAskClient.js. Askpass handle: /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor-ssh-N1Vsk8/socket.sock
2025-08-01 09:34:52.838 [info] Launching SSH server via shell with command: cat ""/var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor_remote_install_31848.sh"" | ssh -v -T -D 60072 login.haicore.berlin bash --login -c bash
2025-08-01 09:34:52.838 [info] Started installation script. Waiting for it to finish...
2025-08-01 09:34:52.838 [info] Waiting for server to install via process(7539)...
2025-08-01 09:34:52.847 [info] (ssh_tunnel) stderr: OpenSSH_9.9p1, LibreSSL 3.3.6
debug1: Reading configuration data /Users/franzsrambical/.ssh/config

2025-08-01 09:34:52.847 [info] (ssh_tunnel) stderr: debug1: /Users/franzsrambical/.ssh/config line 22: Applying options for login.haicore.berlin
debug1: Reading configuration data /etc/ssh/ssh_config
debug1: Reading configuration data /etc/ssh/ssh_config.d/100-macos.conf
debug1: /etc/ssh/ssh_config.d/100-macos.conf line 1: Applying options for *

2025-08-01 09:34:52.847 [info] (ssh_tunnel) stderr: debug1: Reading configuration data /etc/ssh/crypto.conf

2025-08-01 09:34:52.847 [info] (ssh_tunnel) stderr: debug1: Authenticator provider $SSH_SK_PROVIDER did not resolve; disabling
debug1: Connecting to login.haicore.berlin port 22.

2025-08-01 09:34:52.965 [info] (ssh_tunnel) stderr: debug1: Connection established.

2025-08-01 09:34:52.966 [info] (ssh_tunnel) stderr: debug1: identity file /Users/franzsrambical/.ssh/id_ed25519 type 3

2025-08-01 09:34:52.966 [info] (ssh_tunnel) stderr: debug1: identity file /Users/franzsrambical/.ssh/id_ed25519-cert type -1

2025-08-01 09:34:52.966 [info] (ssh_tunnel) stderr: debug1: Local version string SSH-2.0-OpenSSH_9.9

2025-08-01 09:34:53.011 [info] (ssh_tunnel) stderr: debug1: Remote protocol version 2.0, remote software version OpenSSH_8.7
debug1: compat_banner: match: OpenSSH_8.7 pat OpenSSH* compat 0x04000000
debug1: Authenticating to login.haicore.berlin:22 as 'franz.srambical'

2025-08-01 09:34:53.012 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /Users/franzsrambical/.ssh/known_hosts2: No such file or directory

2025-08-01 09:34:53.013 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts: No such file or directory
debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts2: No such file or directory
debug1: SSH2_MSG_KEXINIT sent

2025-08-01 09:34:53.050 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_KEXINIT received
debug1: kex: algorithm: ecdh-sha2-nistp256
debug1: kex: host key algorithm: ssh-ed25519
debug1: kex: server->client cipher: aes128-gcm@openssh.com MAC: <implicit> compression: none

2025-08-01 09:34:53.050 [info] (ssh_tunnel) stderr: debug1: kex: client->server cipher: aes128-gcm@openssh.com MAC: <implicit> compression: none

2025-08-01 09:34:53.051 [info] (ssh_tunnel) stderr: debug1: expecting SSH2_MSG_KEX_ECDH_REPLY

2025-08-01 09:34:53.092 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_KEX_ECDH_REPLY received
debug1: Server host key: ssh-ed25519 SHA256:3/BGZ1UNXR9SufKdsZVtx4Yd+kZTnZzSvRH0l6rtbvo

2025-08-01 09:34:53.093 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /Users/franzsrambical/.ssh/known_hosts2: No such file or directory

2025-08-01 09:34:53.093 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts: No such file or directory
debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts2: No such file or directory
debug1: Host 'login.haicore.berlin' is known and matches the ED25519 host key.
debug1: Found key in /Users/franzsrambical/.ssh/known_hosts:17

2025-08-01 09:34:53.097 [info] (ssh_tunnel) stderr: debug1: ssh_packet_send2_wrapped: resetting send seqnr 3
debug1: rekey out after 4294967296 blocks
debug1: SSH2_MSG_NEWKEYS sent
debug1: expecting SSH2_MSG_NEWKEYS

2025-08-01 09:34:53.097 [info] (ssh_tunnel) stderr: debug1: ssh_packet_read_poll2: resetting read seqnr 3
debug1: SSH2_MSG_NEWKEYS received
debug1: rekey in after 4294967296 blocks
debug1: SSH2_MSG_EXT_INFO received
debug1: kex_ext_info_client_parse: server-sig-algs=<ssh-ed25519,sk-ssh-ed25519@openssh.com,ssh-rsa,rsa-sha2-256,rsa-sha2-512,ssh-dss,ecdsa-sha2-nistp256,ecdsa-sha2-nistp384,ecdsa-sha2-nistp521,sk-ecdsa-sha2-nistp256@openssh.com,webauthn-sk-ecdsa-sha2-nistp256@openssh.com>

2025-08-01 09:34:53.225 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_SERVICE_ACCEPT received

2025-08-01 09:34:53.297 [info] (ssh_tunnel) stderr: debug1: Authentications that can continue: publickey,gssapi-keyex,gssapi-with-mic,password,keyboard-interactive
debug1: Next authentication method: publickey

2025-08-01 09:34:53.302 [info] (ssh_tunnel) stderr: debug1: get_agent_identities: bound agent to hostkey

2025-08-01 09:34:53.303 [info] (ssh_tunnel) stderr: debug1: get_agent_identities: ssh_fetch_identitylist: agent contains no identities
debug1: Will attempt key: /Users/franzsrambical/.ssh/id_ed25519 ED25519 SHA256:Iq/4xZ7XCaNSqbQ9N2LsZ+UYwGSzaQ+kLJHcOv/9ogQ explicit
debug1: Offering public key: /Users/franzsrambical/.ssh/id_ed25519 ED25519 SHA256:Iq/4xZ7XCaNSqbQ9N2LsZ+UYwGSzaQ+kLJHcOv/9ogQ explicit

2025-08-01 09:34:53.354 [info] (ssh_tunnel) stderr: debug1: Authentications that can continue: publickey,gssapi-keyex,gssapi-with-mic,password,keyboard-interactive
debug1: Next authentication method: keyboard-interactive

2025-08-01 09:34:53.438 [info] (ssh_tunnel) stderr: debug1: read_passphrase: requested to askpass

2025-08-01 09:34:53.721 [info] Askpass server received request: POST /
2025-08-01 09:34:53.721 [info] Askpass server received request body: {""request"":""(franz.srambical@login.haicore.berlin) Password: ""}
2025-08-01 09:34:53.721 [info] Received SSH askpass request: (franz.srambical@login.haicore.berlin) Password: 
2025-08-01 09:34:57.426 [error] Password authentication cancelled
2025-08-01 09:34:57.428 [info] (ssh_tunnel) stderr: Server returned status code: 500

2025-08-01 09:35:00.064 [info] Resolving ssh remote authority 'login.haicore.berlin' (Unparsed 'ssh-remote+7b22686f73744e616d65223a226c6f67696e2e686169636f72652e6265726c696e227d') (attempt #1)
2025-08-01 09:35:00.070 [info] SSH askpass server listening on /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor-ssh-Ze6kcV/socket.sock
2025-08-01 09:35:00.071 [info] Using configured platform linux for remote host login.haicore.berlin
2025-08-01 09:35:00.072 [info] Using askpass script: /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/launchSSHAskpass.sh with javascript file /Users/franzsrambical/.cursor/extensions/anysphere.remote-ssh-1.0.23/dist/scripts/sshAskClient.js. Askpass handle: /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor-ssh-Ze6kcV/socket.sock
2025-08-01 09:35:00.076 [info] Launching SSH server via shell with command: cat ""/var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor_remote_install_45994.sh"" | ssh -v -T -D 60081 login.haicore.berlin bash --login -c bash
2025-08-01 09:35:00.076 [info] Started installation script. Waiting for it to finish...
2025-08-01 09:35:00.076 [info] Waiting for server to install via process(7554)...
2025-08-01 09:35:00.083 [info] (ssh_tunnel) stderr: OpenSSH_9.9p1, LibreSSL 3.3.6
debug1: Reading configuration data /Users/franzsrambical/.ssh/config

2025-08-01 09:35:00.083 [info] (ssh_tunnel) stderr: debug1: /Users/franzsrambical/.ssh/config line 22: Applying options for login.haicore.berlin
debug1: Reading configuration data /etc/ssh/ssh_config
debug1: Reading configuration data /etc/ssh/ssh_config.d/100-macos.conf
debug1: /etc/ssh/ssh_config.d/100-macos.conf line 1: Applying options for *
debug1: Reading configuration data /etc/ssh/crypto.conf

2025-08-01 09:35:00.083 [info] (ssh_tunnel) stderr: debug1: Authenticator provider $SSH_SK_PROVIDER did not resolve; disabling
debug1: Connecting to login.haicore.berlin port 22.

2025-08-01 09:35:00.129 [info] (ssh_tunnel) stderr: debug1: Connection established.

2025-08-01 09:35:00.129 [info] (ssh_tunnel) stderr: debug1: identity file /Users/franzsrambical/.ssh/id_ed25519 type 3
debug1: identity file /Users/franzsrambical/.ssh/id_ed25519-cert type -1

2025-08-01 09:35:00.129 [info] (ssh_tunnel) stderr: debug1: Local version string SSH-2.0-OpenSSH_9.9

2025-08-01 09:35:00.178 [info] (ssh_tunnel) stderr: debug1: Remote protocol version 2.0, remote software version OpenSSH_8.7
debug1: compat_banner: match: OpenSSH_8.7 pat OpenSSH* compat 0x04000000
debug1: Authenticating to login.haicore.berlin:22 as 'franz.srambical'

2025-08-01 09:35:00.179 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /Users/franzsrambical/.ssh/known_hosts2: No such file or directory
debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts: No such file or directory

2025-08-01 09:35:00.179 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts2: No such file or directory
debug1: SSH2_MSG_KEXINIT sent

2025-08-01 09:35:00.218 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_KEXINIT received
debug1: kex: algorithm: ecdh-sha2-nistp256
debug1: kex: host key algorithm: ssh-ed25519
debug1: kex: server->client cipher: aes128-gcm@openssh.com MAC: <implicit> compression: none
debug1: kex: client->server cipher: aes128-gcm@openssh.com MAC: <implicit> compression: none

2025-08-01 09:35:00.218 [info] (ssh_tunnel) stderr: debug1: expecting SSH2_MSG_KEX_ECDH_REPLY

2025-08-01 09:35:00.260 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_KEX_ECDH_REPLY received

2025-08-01 09:35:00.260 [info] (ssh_tunnel) stderr: debug1: Server host key: ssh-ed25519 SHA256:3/BGZ1UNXR9SufKdsZVtx4Yd+kZTnZzSvRH0l6rtbvo

2025-08-01 09:35:00.261 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /Users/franzsrambical/.ssh/known_hosts2: No such file or directory
debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts: No such file or directory

2025-08-01 09:35:00.261 [info] (ssh_tunnel) stderr: debug1: load_hostkeys: fopen /etc/ssh/ssh_known_hosts2: No such file or directory
debug1: Host 'login.haicore.berlin' is known and matches the ED25519 host key.
debug1: Found key in /Users/franzsrambical/.ssh/known_hosts:17

2025-08-01 09:35:00.265 [info] (ssh_tunnel) stderr: debug1: ssh_packet_send2_wrapped: resetting send seqnr 3
debug1: rekey out after 4294967296 blocks

2025-08-01 09:35:00.265 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_NEWKEYS sent
debug1: expecting SSH2_MSG_NEWKEYS

2025-08-01 09:35:00.265 [info] (ssh_tunnel) stderr: debug1: ssh_packet_read_poll2: resetting read seqnr 3
debug1: SSH2_MSG_NEWKEYS received
debug1: rekey in after 4294967296 blocks
debug1: SSH2_MSG_EXT_INFO received
debug1: kex_ext_info_client_parse: server-sig-algs=<ssh-ed25519,sk-ssh-ed25519@openssh.com,ssh-rsa,rsa-sha2-256,rsa-sha2-512,ssh-dss,ecdsa-sha2-nistp256,ecdsa-sha2-nistp384,ecdsa-sha2-nistp521,sk-ecdsa-sha2-nistp256@openssh.com,webauthn-sk-ecdsa-sha2-nistp256@openssh.com>

2025-08-01 09:35:00.386 [info] (ssh_tunnel) stderr: debug1: SSH2_MSG_SERVICE_ACCEPT received

2025-08-01 09:35:00.430 [info] (ssh_tunnel) stderr: debug1: Authentications that can continue: publickey,gssapi-keyex,gssapi-with-mic,password,keyboard-interactive
debug1: Next authentication method: publickey

2025-08-01 09:35:00.432 [info] (ssh_tunnel) stderr: debug1: get_agent_identities: bound agent to hostkey

2025-08-01 09:35:00.432 [info] (ssh_tunnel) stderr: debug1: get_agent_identities: ssh_fetch_identitylist: agent contains no identities
debug1: Will attempt key: /Users/franzsrambical/.ssh/id_ed25519 ED25519 SHA256:Iq/4xZ7XCaNSqbQ9N2LsZ+UYwGSzaQ+kLJHcOv/9ogQ explicit
debug1: Offering public key: /Users/franzsrambical/.ssh/id_ed25519 ED25519 SHA256:Iq/4xZ7XCaNSqbQ9N2LsZ+UYwGSzaQ+kLJHcOv/9ogQ explicit

2025-08-01 09:35:00.520 [info] (ssh_tunnel) stderr: debug1: Server accepts key: /Users/franzsrambical/.ssh/id_ed25519 ED25519 SHA256:Iq/4xZ7XCaNSqbQ9N2LsZ+UYwGSzaQ+kLJHcOv/9ogQ explicit

2025-08-01 09:35:00.601 [info] (ssh_tunnel) stderr: Authenticated to login.haicore.berlin ([141.80.150.4]:22) using ""publickey"".
debug1: Local connections to LOCALHOST:60081 forwarded to remote address socks:0

2025-08-01 09:35:00.601 [info] (ssh_tunnel) stderr: debug1: Local forwarding listening on ::1 port 60081.

2025-08-01 09:35:00.602 [info] (ssh_tunnel) stderr: debug1: channel 0: new port-listener [port listener] (inactive timeout: 0)
debug1: Local forwarding listening on 127.0.0.1 port 60081.
debug1: channel 1: new port-listener [port listener] (inactive timeout: 0)
debug1: channel 2: new session [client-session] (inactive timeout: 0)
debug1: Requesting no-more-sessions@openssh.com

2025-08-01 09:35:00.602 [info] (ssh_tunnel) stderr: debug1: Entering interactive session.
debug1: pledge: filesystem

2025-08-01 09:35:00.795 [info] (ssh_tunnel) stderr: debug1: client_input_global_request: rtype hostkeys-00@openssh.com want_reply 0

2025-08-01 09:35:00.795 [info] (ssh_tunnel) stderr: debug1: client_input_hostkeys: searching /Users/franzsrambical/.ssh/known_hosts for login.haicore.berlin / (none)

2025-08-01 09:35:00.798 [info] (ssh_tunnel) stderr: debug1: client_input_hostkeys: searching /Users/franzsrambical/.ssh/known_hosts2 for login.haicore.berlin / (none)
debug1: client_input_hostkeys: hostkeys file /Users/franzsrambical/.ssh/known_hosts2 does not exist
debug1: client_input_hostkeys: no new or deprecated keys from server
debug1: Remote: /home/franz.srambical/.ssh/authorized_keys:1: key options: agent-forwarding port-forwarding pty user-rc x11-forwarding

2025-08-01 09:35:00.833 [info] (ssh_tunnel) stderr: debug1: Remote: /home/franz.srambical/.ssh/authorized_keys:1: key options: agent-forwarding port-forwarding pty user-rc x11-forwarding
debug1: Sending environment.

2025-08-01 09:35:00.833 [info] (ssh_tunnel) stderr: debug1: Sending command: bash --login -c bash
debug1: pledge: network

2025-08-01 09:35:01.292 [info] (ssh_tunnel) stdout: Using TMP_DIR: /run/user/961800067
Locking /run/user/961800067/cursor-remote-lock.c403edc4db82e26fa41a0903d75ac6d0
Server script already installed in /home/franz.srambical/.cursor-server/bin/a8e95743c5268be73767c46944a71f4465d05c90/bin/cursor-server
Checking node executable
v20.18.2

2025-08-01 09:35:01.298 [info] (ssh_tunnel) stdout: Checking for running multiplex server: /home/franz.srambical/.cursor-server/bin/multiplex-server/45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15.js

2025-08-01 09:35:01.315 [info] (ssh_tunnel) stdout: Running multiplex server: 

2025-08-01 09:35:01.316 [info] (ssh_tunnel) stdout: Creating multiplex server token file /run/user/961800067/cursor-remote-multiplex.token.c403edc4db82e26fa41a0903d75ac6d0.45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15

2025-08-01 09:35:01.320 [info] (ssh_tunnel) stdout: Creating directory for multiplex server: /home/franz.srambical/.cursor-server/bin/multiplex-server

2025-08-01 09:35:01.322 [info] (ssh_tunnel) stdout: Writing multiplex server script to /home/franz.srambical/.cursor-server/bin/multiplex-server/45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15.js

2025-08-01 09:35:01.332 [info] (ssh_tunnel) stdout: Starting multiplex server: /home/franz.srambical/.cursor-server/bin/a8e95743c5268be73767c46944a71f4465d05c90/node /home/franz.srambical/.cursor-server/bin/multiplex-server/45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15.js 1049ddbe-3284-492e-b7dd-41f38a3105dc
Multiplex server started with PID 311292 and wrote pid to file /run/user/961800067/cursor-remote-multiplex.pid.c403edc4db82e26fa41a0903d75ac6d0.45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15
Reading multiplex server token file /run/user/961800067/cursor-remote-multiplex.token.c403edc4db82e26fa41a0903d75ac6d0.45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15
Multiplex server token file found
Reading multiplex server log file /run/user/961800067/cursor-remote-multiplex.log.c403edc4db82e26fa41a0903d75ac6d0.45e440a0fc5a5d12380c7a83a49ab82c55f715a5d60292da31f8d75730a9ee15

2025-08-01 09:35:01.840 [info] (ssh_tunnel) stdout: Checking for code servers

2025-08-01 09:35:01.855 [info] (ssh_tunnel) stdout: Code server script is not running
Creating code server token file /run/user/961800067/cursor-remote-code.token.c403edc4db82e26fa41a0903d75ac6d0

2025-08-01 09:35:01.856 [info] (ssh_tunnel) stdout: Starting code server script /home/franz.srambical/.cursor-server/bin/a8e95743c5268be73767c46944a71f4465d05c90/bin/cursor-server --start-server --host=127.0.0.1 --port 0  --connection-token-file /run/user/961800067/cursor-remote-code.token.c403edc4db82e26fa41a0903d75ac6d0 --telemetry-level off --enable-remote-auto-shutdown --accept-server-license-terms &> /run/user/961800067/cursor-remote-code.log.c403edc4db82e26fa41a0903d75ac6d0 &

2025-08-01 09:35:01.857 [info] (ssh_tunnel) stdout: Code server started with PID 311316 and wrote pid to file /run/user/961800067/cursor-remote-code.pid.c403edc4db82e26fa41a0903d75ac6d0

2025-08-01 09:35:01.859 [info] (ssh_tunnel) stdout: Code server log file is /run/user/961800067/cursor-remote-code.log.c403edc4db82e26fa41a0903d75ac6d0

2025-08-01 09:35:02.415 [info] (ssh_tunnel) stdout: b41f16eb63b64385d25a3022: start
exitCode==0==
nodeExecutable==/home/franz.srambical/.cursor-server/bin/a8e95743c5268be73767c46944a71f4465d05c90/node==
errorMessage====
isFatalError==false==
multiplexListeningOn==43445==
multiplexConnectionToken==1049ddbe-3284-492e-b7dd-41f38a3105dc==
codeListeningOn==42663==
codeConnectionToken==541fe3b4-8d0c-4bfb-8c97-da67e297099e==
detectedPlatform==linux==
arch==x64==
SSH_AUTH_SOCK====
b41f16eb63b64385d25a3022: end
Unlocking /run/user/961800067/cursor-remote-lock.c403edc4db82e26fa41a0903d75ac6d0

2025-08-01 09:35:02.417 [info] Server install command exit code:  0
2025-08-01 09:35:02.417 [info] Deleting local script /var/folders/nn/241fnlwx03d7k7qt2jg98txr0000gn/T/cursor_remote_install_45994.sh
2025-08-01 09:35:02.419 [info] [forwarding][code] creating new forwarding server
2025-08-01 09:35:02.419 [info] [forwarding][code] server listening on 127.0.0.1:60083
2025-08-01 09:35:02.419 [info] [forwarding][code] Set up server
2025-08-01 09:35:02.419 [info] [remote-ssh] codeListeningOn (remote=[object Object]; local=[object Object]) codeConnectionToken: 541fe3b4-8d0c-4bfb-8c97-da67e297099e
2025-08-01 09:35:02.420 [info] [forwarding][multiplex] creating new forwarding server
2025-08-01 09:35:02.420 [info] [forwarding][multiplex] server listening on 127.0.0.1:60084
2025-08-01 09:35:02.420 [info] [forwarding][multiplex] Set up server
2025-08-01 09:35:02.424 [info] [remote-ssh] multiplexListeningOn (remote=[object Object]; local=[object Object]) multiplexConnectionToken: 1049ddbe-3284-492e-b7dd-41f38a3105dc
2025-08-01 09:35:02.425 [info] [remote-ssh] Pinging remote server on port 127.0.0.1:60084
2025-08-01 09:35:02.427 [info] [remote-ssh] Resolved exec server. Socks port: 60081
2025-08-01 09:35:02.427 [info] Setting up 0 default forwarded ports
2025-08-01 09:35:02.427 [info] [remote-ssh] Resolved authority: {""host"":""127.0.0.1"",""port"":60083,""connectionToken"":""541fe3b4-8d0c-4bfb-8c97-da67e297099e"",""extensionHostEnv"":{}}. Socks port: 60081
2025-08-01 09:35:02.429 [info] [forwarding][multiplex][127.0.0.1:60084 -> 127.0.0.1:43445][b0a1460c-960d-429b-bd2f-3b513f61db53] received connection request
2025-08-01 09:35:02.430 [info] [command][6b98f8a8-3796-435e-889b-6dc6a62b567d] Sending command request: {""command"":""echo"",""args"":[""1""],""env"":{},""token"":""1049ddbe-3284-492e-b7dd-41f38a3105dc"",""id"":""6b98f8a8-3796-435e-889b-6dc6a62b567d""}
2025-08-01 09:35:02.431 [info] (ssh_tunnel) stderr: debug1: Connection to port 60081 forwarding to socks port 0 requested.
debug1: channel 3: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:35:02.476 [info] [forwarding][multiplex][127.0.0.1:60084 -> 127.0.0.1:60081 -> 127.0.0.1:43445][b0a1460c-960d-429b-bd2f-3b513f61db53] socks forwarding established
2025-08-01 09:35:02.477 [info] [forwarding][code][127.0.0.1:60083 -> 127.0.0.1:42663][f05ec0ab-3c52-4a80-b97e-e98e5fa33137] received connection request
2025-08-01 09:35:02.477 [info] (ssh_tunnel) stderr: debug1: Connection to port 60081 forwarding to socks port 0 requested.
debug1: channel 4: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:35:02.529 [info] [command][6b98f8a8-3796-435e-889b-6dc6a62b567d] Process exited with code 0
2025-08-01 09:35:02.529 [info] [command][6b98f8a8-3796-435e-889b-6dc6a62b567d] Socket close event received
2025-08-01 09:35:02.530 [info] [forwarding][code][127.0.0.1:60083 -> 127.0.0.1:60081 -> 127.0.0.1:42663][f05ec0ab-3c52-4a80-b97e-e98e5fa33137] socks forwarding established
2025-08-01 09:35:02.530 [info] [forwarding][multiplex][127.0.0.1:60084 -> 127.0.0.1:60081 -> 127.0.0.1:43445][b0a1460c-960d-429b-bd2f-3b513f61db53] socks connection closed
2025-08-01 09:35:02.571 [info] (ssh_tunnel) stderr: debug1: channel 3: free: direct-tcpip: listening port 60081 for 127.0.0.1 port 43445, connect from 127.0.0.1 port 60086 to 127.0.0.1 port 60081, nchannels 5

2025-08-01 09:35:02.588 [info] [forwarding][code][127.0.0.1:60083 -> 127.0.0.1:42663][5d621007-57bd-40c3-b036-9652ff6445df] received connection request
2025-08-01 09:35:02.588 [info] (ssh_tunnel) stderr: debug1: Connection to port 60081 forwarding to socks port 0 requested.
debug1: channel 3: new dynamic-tcpip [dynamic-tcpip] (inactive timeout: 0)

2025-08-01 09:35:02.628 [info] [forwarding][code][127.0.0.1:60083 -> 127.0.0.1:60081 -> 127.0.0.1:42663][5d621007-57bd-40c3-b036-9652ff6445df] socks forwarding established
2025-08-01 09:35:02.808 [info] Saved platform linux for remote host login.haicore.berlin
",log,tab
3,97,"sample.py",0,0,"",python,tab
4,151,"extension-output-pdoom-org.crowd-code-#1-crowd-code",0,0,"9:35:08 AM [info] Activating crowd-code
9:35:08 AM [info] Recording started
9:35:08 AM [info] Initializing git provider using file system watchers...
",Log,tab
5,302,"extension-output-pdoom-org.crowd-code-#1-crowd-code",150,0,"9:35:08 AM [info] Git repository found
9:35:08 AM [info] Git provider initialized successfully
9:35:08 AM [info] Initial git state: [object Object]
",Log,content
6,2720,"sample.py",0,0,"",python,tab
7,2722,"TERMINAL",0,0,"",,terminal_focus
8,3466,"TERMINAL",0,0,"source /home/franz.srambical/jafar/.venv/bin/activate",,terminal_command
9,18980,"sample.py",0,0,"",python,tab
10,19036,"sample.py",69,0,"",python,selection_command
11,45818,"genie.py",0,0,"from typing import Dict

import optax
import jax
import jax.numpy as jnp
import flax.nnx as nnx
import orbax.checkpoint as ocp

from models.dynamics import DynamicsMaskGIT, DynamicsCausal
from models.lam import LatentActionModel
from models.tokenizer import TokenizerVQVAE


class Genie(nnx.Module):
    """"""Genie model""""""

    def __init__(
        self,
        in_dim: int,
        tokenizer_dim: int,
        tokenizer_ffn_dim: int,
        latent_patch_dim: int,
        num_patch_latents: int,
        patch_size: int,
        tokenizer_num_blocks: int,
        tokenizer_num_heads: int,
        lam_dim: int,
        lam_ffn_dim: int,
        latent_action_dim: int,
        num_latent_actions: int,
        lam_patch_size: int,
        lam_num_blocks: int,
        lam_num_heads: int,
        lam_co_train: bool,
        dyna_type: str,
        dyna_dim: int,
        dyna_ffn_dim: int,
        dyna_num_blocks: int,
        dyna_num_heads: int,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
        dropout: float = 0.0,
        mask_limit: float = 0.0,
    ):
        # --- Tokenizer ---
        self.in_dim = in_dim
        self.tokenizer_dim = tokenizer_dim
        self.tokenizer_ffn_dim = tokenizer_ffn_dim
        self.latent_patch_dim = latent_patch_dim
        self.num_patch_latents = num_patch_latents
        self.patch_size = patch_size
        self.tokenizer_num_blocks = tokenizer_num_blocks
        self.tokenizer_num_heads = tokenizer_num_heads
        # --- LAM ---
        self.lam_dim = lam_dim
        self.lam_ffn_dim = lam_ffn_dim
        self.latent_action_dim = latent_action_dim
        self.num_latent_actions = num_latent_actions
        self.lam_patch_size = lam_patch_size
        self.lam_num_blocks = lam_num_blocks
        self.lam_num_heads = lam_num_heads
        self.lam_co_train = lam_co_train
        # --- Dynamics ---
        self.dyna_type = dyna_type
        self.dyna_dim = dyna_dim
        self.dyna_ffn_dim = dyna_ffn_dim
        self.dyna_num_blocks = dyna_num_blocks
        self.dyna_num_heads = dyna_num_heads
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention
        self.dropout = dropout
        self.mask_limit = mask_limit

        self.tokenizer = TokenizerVQVAE(
            in_dim=self.in_dim,
            model_dim=self.tokenizer_dim,
            ffn_dim=self.tokenizer_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_patch_latents,
            patch_size=self.patch_size,
            num_blocks=self.tokenizer_num_blocks,
            num_heads=self.tokenizer_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        self.lam = LatentActionModel(
            in_dim=self.in_dim,
            model_dim=self.lam_dim,
            ffn_dim=self.lam_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_latent_actions,
            patch_size=self.lam_patch_size,
            num_blocks=self.lam_num_blocks,
            num_heads=self.lam_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        if self.dyna_type == ""maskgit"":
            self.dynamics = DynamicsMaskGIT(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                mask_limit=self.mask_limit,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                rngs=rngs,
            )
        elif self.dyna_type == ""causal"":
            self.dynamics = DynamicsCausal(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                decode=decode,
                rngs=rngs,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {self.dyna_type}"")

    def __call__(
        self, batch: Dict[str, jax.Array], training: bool = True
    ) -> Dict[str, jax.Array]:
        videos_BTHWC = batch[""videos""]
        tokenizer_outputs = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_indices_BTN = tokenizer_outputs[""indices""]
        lam_outputs = self.lam.vq_encode(videos_BTHWC, training=False)
        z_q_BTm11L = lam_outputs[""z_q""]
        action_indices_E = lam_outputs[""indices""]
        latent_actions_BTm11L = jax.lax.cond(
            self.lam_co_train,
            lambda: z_q_BTm11L,
            lambda: jax.lax.stop_gradient(z_q_BTm11L),
        )
        outputs = dict(
            video_tokens=jax.lax.stop_gradient(token_indices_BTN),
            latent_actions=latent_actions_BTm11L,
        )
        outputs[""mask_rng""] = batch[""mask_rng""]
        dyna_logits_BTNV, dyna_mask = self.dynamics(outputs, training)
        outputs[""token_logits""] = dyna_logits_BTNV
        if dyna_mask is not None:
            outputs[""mask""] = dyna_mask
        mle_indices_BTN = jnp.argmax(outputs[""token_logits""], axis=-1)
        H, W = batch[""videos""].shape[2:4]
        outputs[""recon""] = self.tokenizer.decode(mle_indices_BTN, (H, W))
        outputs[""lam_indices""] = action_indices_E
        return outputs

    # FIXME (f.srambical): sampling should be moved to the dynamics classes
    def sample(
        self,
        batch: Dict[str, jax.Array],
        seq_len: int,
        steps: int = 25,
        temperature: float = 1,
        sample_argmax: bool = False,
    ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)

        def maskgit_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, mask_BSN, action_tokens_EL = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = self.dynamics.patch_embed(token_idxs_BSN)
            mask_token_111M = self.dynamics.mask_token.value
            mask_expanded_BSN1 = mask_BSN[..., None]
            vid_embed_BSNM = jnp.where(mask_expanded_BSN1, mask_token_111M, vid_embed_BSNM)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = self.dynamics.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNM += act_embed_BS1M
            unmasked_ratio = jnp.cos(jnp.pi * (step + 1) / (steps * 2))
            step_temp = temperature * (1.0 - unmasked_ratio)
            final_logits_BSNV = self.dynamics.transformer(vid_embed_BSNM) / step_temp

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_BSN = jnp.argmax(final_logits_BSNV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_BSN = jax.random.categorical(_rng, final_logits_BSNV)
            gather_fn = jax.vmap(jax.vmap(jax.vmap(lambda x, y: x[y])))
            final_token_probs_BSN = gather_fn(
                jax.nn.softmax(final_logits_BSNV), sampled_token_idxs_BSN
            )
            final_token_probs_BSN += ~mask_BSN
            # Update masked tokens only
            token_idxs_BSN = jnp.where(mask_BSN, sampled_token_idxs_BSN, token_idxs_BSN)

            # --- Update mask ---
            num_unmasked_tokens = jnp.round(N * (1.0 - unmasked_ratio)).astype(int)
            idx_mask_N = jnp.arange(final_token_probs_BSN.shape[-1]) > num_unmasked_tokens
            sorted_idxs_BSN = jnp.argsort(final_token_probs_BSN, axis=-1, descending=True)
            mask_update_fn = jax.vmap(lambda msk, ids: msk.at[ids].set(idx_mask_N))
            new_mask_BSN = mask_update_fn(mask_BSN, sorted_idxs_BSN)

            new_carry = (rng, token_idxs_BSN, new_mask_BSN, action_tokens_EL)
            return new_carry, None

        def generation_step_fn(
            carry: tuple[jax.Array, jax.Array], step_t: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array], None]:
            rng, current_token_idxs_BSN = carry
            rng, step_rng = jax.random.split(rng)

            # Mask current and future frames (i.e., t >= step_t)
            mask_S = jnp.arange(seq_len) >= step_t
            mask_BSN = jnp.broadcast_to(mask_S[None, :, None], (B, seq_len, N)).astype(
                bool
            )
            masked_token_idxs_BSN = current_token_idxs_BSN * ~mask_BSN

            # --- Initialize and run MaskGIT loop ---
            init_carry_maskgit = (
                step_rng,
                masked_token_idxs_BSN,
                mask_BSN,
                action_tokens_EL,
            )
            final_carry_maskgit, _ = jax.lax.scan(
                maskgit_step_fn, init_carry_maskgit, jnp.arange(steps)
            )
            updated_token_idxs_BSN = final_carry_maskgit[1]
            new_carry = (rng, updated_token_idxs_BSN)
            return new_carry, None

        # --- Run the autoregressive generation using jax.lax.scan ---
        initial_carry = (batch[""rng""], token_idxs_BSN)
        timesteps_to_scan = jnp.arange(T, seq_len)
        final_carry, _ = jax.lax.scan(
            generation_step_fn, initial_carry, timesteps_to_scan
        )
        final_token_idxs_BSN = final_carry[1]

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def sample_causal(
            self,
            batch: Dict[str, jax.Array],
            seq_len: int,
            temperature: float = 1,
            sample_argmax: bool = False,
        ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # FIXME (f.srambical): reset spatial kv cache after each frame
        assert isinstance(self.dynamics, DynamicsCausal)
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)
        dynamics_causal: DynamicsCausal = self.dynamics

        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.spatial_attention.cached_value[...] = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.temporal_attention.cache_index[...] = jnp.array(0, dtype=jnp.int32)
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            current_token_idxs_BSN = final_carry_causal[1]
        
        final_token_idxs_BSN = current_token_idxs_BSN

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def vq_encode(self, batch: Dict[str, jax.Array], training: bool) -> jax.Array:
        # --- Preprocess videos ---
        video_BTHWC = batch[""videos""]
        lam_output = self.lam.vq_encode(video_BTHWC, training=training)
        lam_indices_E = lam_output[""indices""]
        return lam_indices_E

# FIXME (f.srambical): add conversion script for old checkpoints
def restore_genie_components(
    optimizer: nnx.Optimizer,
    sharding: jax.sharding.NamedSharding,
    rng: jax.Array,
    args,
) -> nnx.Optimizer:
    """"""Restore pre-trained Genie components""""""
    rngs = nnx.Rngs(rng)

    tx = optimizer.tx
    model = optimizer.model
    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )

    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    tokenizer_checkpoint_manager = ocp.CheckpointManager(
        directory=args.tokenizer_checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )
    dummy_tokenizer = TokenizerVQVAE(
        in_dim=args.image_channels,
        model_dim=args.tokenizer_dim,
        ffn_dim=args.tokenizer_ffn_dim,
        latent_dim=args.latent_patch_dim,
        num_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        num_blocks=args.tokenizer_num_blocks,
        num_heads=args.tokenizer_num_heads,
        dropout=args.dropout,
        codebook_dropout=args.dropout,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        rngs=rngs,
    )
    dummy_tokenizer_optimizer = nnx.Optimizer(dummy_tokenizer, tx)
    dummy_tokenizer_optimizer_state = nnx.state(dummy_tokenizer_optimizer)
    abstract_sharded_tokenizer_optimizer_state = _create_abstract_sharded_pytree(
        dummy_tokenizer_optimizer_state, sharding
    )
    restored_tokenizer = tokenizer_checkpoint_manager.restore(
        step=tokenizer_checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(  # type: ignore
                abstract_sharded_tokenizer_optimizer_state  # type: ignore
            ),
        ),
    )[""model_state""]
    nnx.update(dummy_tokenizer_optimizer.model, restored_tokenizer.model)
    model.tokenizer = dummy_tokenizer_optimizer.model
    tokenizer_checkpoint_manager.close()

    if args.lam_checkpoint:
        lam_checkpoint_manager = ocp.CheckpointManager(
            directory=args.lam_checkpoint,
            options=checkpoint_options,
            handler_registry=handler_registry,
        )
        dummy_lam = LatentActionModel(
            in_dim=args.image_channels,
            model_dim=args.lam_dim,
            ffn_dim=args.lam_ffn_dim,
            latent_dim=args.latent_patch_dim,
            num_latents=args.num_latent_actions,
            patch_size=args.lam_patch_size,
            num_blocks=args.lam_num_blocks,
            num_heads=args.lam_num_heads,
            dropout=args.dropout,
            codebook_dropout=args.dropout,
            param_dtype=args.param_dtype,
            dtype=args.dtype,
            use_flash_attention=args.use_flash_attention,
            rngs=rngs,
        )
        dummy_lam_optimizer = nnx.Optimizer(dummy_lam, tx)
        dummy_lam_optimizer_state = nnx.state(dummy_lam_optimizer)
        abstract_sharded_lam_optimizer_state = _create_abstract_sharded_pytree(
            dummy_lam_optimizer_state, sharding
        )
        restored_lam_optimizer = lam_checkpoint_manager.restore(
            step=lam_checkpoint_manager.latest_step(),
            args=ocp.args.Composite(
                model_state=ocp.args.PyTreeRestore(  # type: ignore
                    abstract_sharded_lam_optimizer_state  # type: ignore
                ),
            ),
        )[""model_state""]
        nnx.update(dummy_lam_optimizer.model, restored_lam_optimizer.model)
        model.lam = dummy_lam_optimizer.model
        # Remove the LAM decoder to save memory and avoid unnecessary computation.
        del model.lam.decoder
        lam_checkpoint_manager.close()
    
    # Reinitialize the optimizer states
    optimizer = nnx.Optimizer(model, tx)
    return optimizer


def _create_abstract_sharded_pytree(
    pytree_template: nnx.GraphState, sharding_spec: jax.sharding.NamedSharding
) -> jax.Array:
    """"""Replaces arrays in a pytree with ShapeDtypeStructs having the given sharding.""""""

    def map_fn(leaf_template):
        if hasattr(leaf_template, ""shape"") and hasattr(leaf_template, ""dtype""):
            return jax.ShapeDtypeStruct(
                leaf_template.shape, leaf_template.dtype, sharding=sharding_spec
            )
        return leaf_template

    return jax.tree_util.tree_map(map_fn, pytree_template)
",python,tab
12,45820,"genie.py",14205,0,"",python,selection_command
13,82781,"TERMINAL",0,0,"salloc --gpus=1 --ntasks-per-node=1 --cpus-per-task=1 --mem=100G --time=10:00:00",,terminal_command
14,82833,"TERMINAL",0,0,"salloc: Granted job allocation 14571
",,terminal_output
15,82938,"TERMINAL",0,0,"salloc: Waiting for resource configuration
",,terminal_output
16,83943,"TERMINAL",0,0,"salloc: Nodes hai005 are ready for job
",,terminal_output
17,84310,"TERMINAL",0,0,"Running inside SLURM, Job ID 14571.
",,terminal_output
18,84384,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
19,84875,"TERMINAL",0,0,"
(reverse-i-search)`': ",,terminal_output
20,85125,"TERMINAL",0,0,"s': salloc --gpus=1 --ntasks-per-node=1 --cpus-per-task=1 --mem=100G --time=10:00:00",,terminal_output
21,85221,"TERMINAL",0,0,"
a': salloc --gpus=1 --ntasks-per-node=1 --cpus-per-tas
m': bash experiments/sample.sh ",,terminal_output
22,85281,"TERMINAL",0,0,"p': bash experiments/samp",,terminal_output
23,85477,"TERMINAL",0,0,"l': bash experiments/sampl
failed reverse-i-search)`samplo': bash experiments/sampl",,terminal_output
24,86103,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $

",,terminal_output
25,97608,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
26,107060,"genie.py",0,0,"from typing import Dict

import optax
import jax
import jax.numpy as jnp
import flax.nnx as nnx
import orbax.checkpoint as ocp

from models.dynamics import DynamicsMaskGIT, DynamicsCausal
from models.lam import LatentActionModel
from models.tokenizer import TokenizerVQVAE


class Genie(nnx.Module):
    """"""Genie model""""""

    def __init__(
        self,
        in_dim: int,
        tokenizer_dim: int,
        tokenizer_ffn_dim: int,
        latent_patch_dim: int,
        num_patch_latents: int,
        patch_size: int,
        tokenizer_num_blocks: int,
        tokenizer_num_heads: int,
        lam_dim: int,
        lam_ffn_dim: int,
        latent_action_dim: int,
        num_latent_actions: int,
        lam_patch_size: int,
        lam_num_blocks: int,
        lam_num_heads: int,
        lam_co_train: bool,
        dyna_type: str,
        dyna_dim: int,
        dyna_ffn_dim: int,
        dyna_num_blocks: int,
        dyna_num_heads: int,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
        dropout: float = 0.0,
        mask_limit: float = 0.0,
    ):
        # --- Tokenizer ---
        self.in_dim = in_dim
        self.tokenizer_dim = tokenizer_dim
        self.tokenizer_ffn_dim = tokenizer_ffn_dim
        self.latent_patch_dim = latent_patch_dim
        self.num_patch_latents = num_patch_latents
        self.patch_size = patch_size
        self.tokenizer_num_blocks = tokenizer_num_blocks
        self.tokenizer_num_heads = tokenizer_num_heads
        # --- LAM ---
        self.lam_dim = lam_dim
        self.lam_ffn_dim = lam_ffn_dim
        self.latent_action_dim = latent_action_dim
        self.num_latent_actions = num_latent_actions
        self.lam_patch_size = lam_patch_size
        self.lam_num_blocks = lam_num_blocks
        self.lam_num_heads = lam_num_heads
        self.lam_co_train = lam_co_train
        # --- Dynamics ---
        self.dyna_type = dyna_type
        self.dyna_dim = dyna_dim
        self.dyna_ffn_dim = dyna_ffn_dim
        self.dyna_num_blocks = dyna_num_blocks
        self.dyna_num_heads = dyna_num_heads
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention
        self.dropout = dropout
        self.mask_limit = mask_limit

        self.tokenizer = TokenizerVQVAE(
            in_dim=self.in_dim,
            model_dim=self.tokenizer_dim,
            ffn_dim=self.tokenizer_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_patch_latents,
            patch_size=self.patch_size,
            num_blocks=self.tokenizer_num_blocks,
            num_heads=self.tokenizer_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        self.lam = LatentActionModel(
            in_dim=self.in_dim,
            model_dim=self.lam_dim,
            ffn_dim=self.lam_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_latent_actions,
            patch_size=self.lam_patch_size,
            num_blocks=self.lam_num_blocks,
            num_heads=self.lam_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        if self.dyna_type == ""maskgit"":
            self.dynamics = DynamicsMaskGIT(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                mask_limit=self.mask_limit,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                rngs=rngs,
            )
        elif self.dyna_type == ""causal"":
            self.dynamics = DynamicsCausal(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                decode=decode,
                rngs=rngs,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {self.dyna_type}"")

    def __call__(
        self, batch: Dict[str, jax.Array], training: bool = True
    ) -> Dict[str, jax.Array]:
        videos_BTHWC = batch[""videos""]
        tokenizer_outputs = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_indices_BTN = tokenizer_outputs[""indices""]
        lam_outputs = self.lam.vq_encode(videos_BTHWC, training=False)
        z_q_BTm11L = lam_outputs[""z_q""]
        action_indices_E = lam_outputs[""indices""]
        latent_actions_BTm11L = jax.lax.cond(
            self.lam_co_train,
            lambda: z_q_BTm11L,
            lambda: jax.lax.stop_gradient(z_q_BTm11L),
        )
        outputs = dict(
            video_tokens=jax.lax.stop_gradient(token_indices_BTN),
            latent_actions=latent_actions_BTm11L,
        )
        outputs[""mask_rng""] = batch[""mask_rng""]
        dyna_logits_BTNV, dyna_mask = self.dynamics(outputs, training)
        outputs[""token_logits""] = dyna_logits_BTNV
        if dyna_mask is not None:
            outputs[""mask""] = dyna_mask
        mle_indices_BTN = jnp.argmax(outputs[""token_logits""], axis=-1)
        H, W = batch[""videos""].shape[2:4]
        outputs[""recon""] = self.tokenizer.decode(mle_indices_BTN, (H, W))
        outputs[""lam_indices""] = action_indices_E
        return outputs

    # FIXME (f.srambical): sampling should be moved to the dynamics classes
    def sample(
        self,
        batch: Dict[str, jax.Array],
        seq_len: int,
        steps: int = 25,
        temperature: float = 1,
        sample_argmax: bool = False,
    ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)

        def maskgit_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, mask_BSN, action_tokens_EL = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = self.dynamics.patch_embed(token_idxs_BSN)
            mask_token_111M = self.dynamics.mask_token.value
            mask_expanded_BSN1 = mask_BSN[..., None]
            vid_embed_BSNM = jnp.where(mask_expanded_BSN1, mask_token_111M, vid_embed_BSNM)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = self.dynamics.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNM += act_embed_BS1M
            unmasked_ratio = jnp.cos(jnp.pi * (step + 1) / (steps * 2))
            step_temp = temperature * (1.0 - unmasked_ratio)
            final_logits_BSNV = self.dynamics.transformer(vid_embed_BSNM) / step_temp

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_BSN = jnp.argmax(final_logits_BSNV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_BSN = jax.random.categorical(_rng, final_logits_BSNV)
            gather_fn = jax.vmap(jax.vmap(jax.vmap(lambda x, y: x[y])))
            final_token_probs_BSN = gather_fn(
                jax.nn.softmax(final_logits_BSNV), sampled_token_idxs_BSN
            )
            final_token_probs_BSN += ~mask_BSN
            # Update masked tokens only
            token_idxs_BSN = jnp.where(mask_BSN, sampled_token_idxs_BSN, token_idxs_BSN)

            # --- Update mask ---
            num_unmasked_tokens = jnp.round(N * (1.0 - unmasked_ratio)).astype(int)
            idx_mask_N = jnp.arange(final_token_probs_BSN.shape[-1]) > num_unmasked_tokens
            sorted_idxs_BSN = jnp.argsort(final_token_probs_BSN, axis=-1, descending=True)
            mask_update_fn = jax.vmap(lambda msk, ids: msk.at[ids].set(idx_mask_N))
            new_mask_BSN = mask_update_fn(mask_BSN, sorted_idxs_BSN)

            new_carry = (rng, token_idxs_BSN, new_mask_BSN, action_tokens_EL)
            return new_carry, None

        def generation_step_fn(
            carry: tuple[jax.Array, jax.Array], step_t: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array], None]:
            rng, current_token_idxs_BSN = carry
            rng, step_rng = jax.random.split(rng)

            # Mask current and future frames (i.e., t >= step_t)
            mask_S = jnp.arange(seq_len) >= step_t
            mask_BSN = jnp.broadcast_to(mask_S[None, :, None], (B, seq_len, N)).astype(
                bool
            )
            masked_token_idxs_BSN = current_token_idxs_BSN * ~mask_BSN

            # --- Initialize and run MaskGIT loop ---
            init_carry_maskgit = (
                step_rng,
                masked_token_idxs_BSN,
                mask_BSN,
                action_tokens_EL,
            )
            final_carry_maskgit, _ = jax.lax.scan(
                maskgit_step_fn, init_carry_maskgit, jnp.arange(steps)
            )
            updated_token_idxs_BSN = final_carry_maskgit[1]
            new_carry = (rng, updated_token_idxs_BSN)
            return new_carry, None

        # --- Run the autoregressive generation using jax.lax.scan ---
        initial_carry = (batch[""rng""], token_idxs_BSN)
        timesteps_to_scan = jnp.arange(T, seq_len)
        final_carry, _ = jax.lax.scan(
            generation_step_fn, initial_carry, timesteps_to_scan
        )
        final_token_idxs_BSN = final_carry[1]

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def sample_causal(
            self,
            batch: Dict[str, jax.Array],
            seq_len: int,
            temperature: float = 1,
            sample_argmax: bool = False,
        ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # FIXME (f.srambical): reset spatial kv cache after each frame
        assert isinstance(self.dynamics, DynamicsCausal)
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)
        dynamics_causal: DynamicsCausal = self.dynamics

        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.spatial_attention.cached_value[...] = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.temporal_attention.cache_index[...] = jnp.array(0, dtype=jnp.int32)
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            current_token_idxs_BSN = final_carry_causal[1]
        
        final_token_idxs_BSN = current_token_idxs_BSN

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def vq_encode(self, batch: Dict[str, jax.Array], training: bool) -> jax.Array:
        # --- Preprocess videos ---
        video_BTHWC = batch[""videos""]
        lam_output = self.lam.vq_encode(video_BTHWC, training=training)
        lam_indices_E = lam_output[""indices""]
        return lam_indices_E

# FIXME (f.srambical): add conversion script for old checkpoints
def restore_genie_components(
    optimizer: nnx.Optimizer,
    sharding: jax.sharding.NamedSharding,
    rng: jax.Array,
    args,
) -> nnx.Optimizer:
    """"""Restore pre-trained Genie components""""""
    rngs = nnx.Rngs(rng)

    tx = optimizer.tx
    model = optimizer.model
    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )

    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    tokenizer_checkpoint_manager = ocp.CheckpointManager(
        directory=args.tokenizer_checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )
    dummy_tokenizer = TokenizerVQVAE(
        in_dim=args.image_channels,
        model_dim=args.tokenizer_dim,
        ffn_dim=args.tokenizer_ffn_dim,
        latent_dim=args.latent_patch_dim,
        num_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        num_blocks=args.tokenizer_num_blocks,
        num_heads=args.tokenizer_num_heads,
        dropout=args.dropout,
        codebook_dropout=args.dropout,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        rngs=rngs,
    )
    dummy_tokenizer_optimizer = nnx.Optimizer(dummy_tokenizer, tx)
    dummy_tokenizer_optimizer_state = nnx.state(dummy_tokenizer_optimizer)
    abstract_sharded_tokenizer_optimizer_state = _create_abstract_sharded_pytree(
        dummy_tokenizer_optimizer_state, sharding
    )
    restored_tokenizer = tokenizer_checkpoint_manager.restore(
        step=tokenizer_checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(  # type: ignore
                abstract_sharded_tokenizer_optimizer_state  # type: ignore
            ),
        ),
    )[""model_state""]
    nnx.update(dummy_tokenizer_optimizer.model, restored_tokenizer.model)
    model.tokenizer = dummy_tokenizer_optimizer.model
    tokenizer_checkpoint_manager.close()

    if args.lam_checkpoint:
        lam_checkpoint_manager = ocp.CheckpointManager(
            directory=args.lam_checkpoint,
            options=checkpoint_options,
            handler_registry=handler_registry,
        )
        dummy_lam = LatentActionModel(
            in_dim=args.image_channels,
            model_dim=args.lam_dim,
            ffn_dim=args.lam_ffn_dim,
            latent_dim=args.latent_patch_dim,
            num_latents=args.num_latent_actions,
            patch_size=args.lam_patch_size,
            num_blocks=args.lam_num_blocks,
            num_heads=args.lam_num_heads,
            dropout=args.dropout,
            codebook_dropout=args.dropout,
            param_dtype=args.param_dtype,
            dtype=args.dtype,
            use_flash_attention=args.use_flash_attention,
            rngs=rngs,
        )
        dummy_lam_optimizer = nnx.Optimizer(dummy_lam, tx)
        dummy_lam_optimizer_state = nnx.state(dummy_lam_optimizer)
        abstract_sharded_lam_optimizer_state = _create_abstract_sharded_pytree(
            dummy_lam_optimizer_state, sharding
        )
        restored_lam_optimizer = lam_checkpoint_manager.restore(
            step=lam_checkpoint_manager.latest_step(),
            args=ocp.args.Composite(
                model_state=ocp.args.PyTreeRestore(  # type: ignore
                    abstract_sharded_lam_optimizer_state  # type: ignore
                ),
            ),
        )[""model_state""]
        nnx.update(dummy_lam_optimizer.model, restored_lam_optimizer.model)
        model.lam = dummy_lam_optimizer.model
        # Remove the LAM decoder to save memory and avoid unnecessary computation.
        del model.lam.decoder
        lam_checkpoint_manager.close()
    
    # Reinitialize the optimizer states
    optimizer = nnx.Optimizer(model, tx)
    return optimizer


def _create_abstract_sharded_pytree(
    pytree_template: nnx.GraphState, sharding_spec: jax.sharding.NamedSharding
) -> jax.Array:
    """"""Replaces arrays in a pytree with ShapeDtypeStructs having the given sharding.""""""

    def map_fn(leaf_template):
        if hasattr(leaf_template, ""shape"") and hasattr(leaf_template, ""dtype""):
            return jax.ShapeDtypeStruct(
                leaf_template.shape, leaf_template.dtype, sharding=sharding_spec
            )
        return leaf_template

    return jax.tree_util.tree_map(map_fn, pytree_template)
",python,tab
27,107060,"genie.py",14205,0,"",python,selection_command
28,114943,"genie.py",16295,0,"",python,selection_mouse
29,118095,"utils/nn.py",0,0,"import math
from typing import Tuple, Callable, List

from flax import nnx
import jax
import jax.numpy as jnp
import einops


class PositionalEncoding(nnx.Module):
    """"""https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/JAX/tutorial6/Transformers_and_MHAttention.html""""""

    def __init__(self, d_model: int, max_len: int = 5000):
        self.d_model = d_model
        self.max_len = max_len

        pe = jnp.zeros((self.max_len, self.d_model))
        position = jnp.arange(0, self.max_len, dtype=jnp.float32)[:, None]
        div_term = jnp.exp(
            jnp.arange(0, self.d_model, 2) * (-math.log(10000.0) / self.d_model)
        )
        pe = pe.at[:, 0::2].set(jnp.sin(position * div_term))
        pe = pe.at[:, 1::2].set(jnp.cos(position * div_term))
        self.pe = nnx.Variable(pe)

    def __call__(self, x: jax.Array) -> jax.Array:
        x = x + self.pe[: x.shape[2]]
        return x


class STBlock(nnx.Module):
    def __init__(
        self,
        dim: int,
        ffn_dim: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        rngs: nnx.Rngs,
    ):
        self.dim = dim
        self.ffn_dim = ffn_dim
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.spatial_pos_enc = PositionalEncoding(self.dim)
        self.spatial_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.spatial_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.dim,
            qkv_features=self.dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=False
            ),
            rngs=rngs,
            decode=False,
        )

        self.temporal_pos_enc = PositionalEncoding(self.dim)
        self.temporal_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.temporal_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.dim,
            qkv_features=self.dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=False,
        )

        self.ffn_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense1 = nnx.Linear(
            in_features=self.dim,
            out_features=self.ffn_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense2 = nnx.Linear(
            in_features=self.ffn_dim,
            out_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    @nnx.remat
    def __call__(self, x_BTNM: jax.Array) -> jax.Array:
        # --- Spatial attention ---
        z_BTNM = self.spatial_pos_enc(x_BTNM)
        z_BTNM = self.spatial_norm(z_BTNM)
        z_BTNM = self.spatial_attention(z_BTNM)
        x_BTNM = x_BTNM + z_BTNM

        # --- Temporal attention ---
        x_BNTM = x_BTNM.swapaxes(1, 2)
        z_BNTM = self.temporal_pos_enc(x_BNTM)
        z_BNTM = self.temporal_norm(z_BNTM)
        z_BNTM = self.temporal_attention(z_BNTM)
        x_BNTM = x_BNTM + z_BNTM
        x_BTNM = x_BNTM.swapaxes(1, 2)

        # --- Feedforward ---
        z_BTNM = self.ffn_norm(x_BTNM)
        z_BTND = self.ffn_dense1(z_BTNM)
        z_BTND = jax.nn.gelu(z_BTND)
        z_BTNM = self.ffn_dense2(z_BTND)
        x_BTNM = x_BTNM + z_BTNM

        return x_BTNM


class STTransformer(nnx.Module):
    """"""
    Dimension keys:
        B: batch size
        T: number of frames
        N: number of patches per frame
        I: number of input features
        M: model dimension
        D: FFN dimension
        O: number of output features
    """"""
    def __init__(
        self,
        input_dim: int,
        model_dim: int,
        ffn_dim: int,
        out_dim: int,
        num_blocks: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        rngs: nnx.Rngs,
    ):
        self.input_dim = input_dim
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.out_dim = out_dim
        self.num_blocks = num_blocks
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.input_norm1 = nnx.LayerNorm(
            num_features=self.input_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_dense = nnx.Linear(
            in_features=self.input_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_norm2 = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

        self.blocks = []
        for _ in range(self.num_blocks):
            self.blocks.append(
                STBlock(
                    dim=self.model_dim,
                    ffn_dim=self.ffn_dim,
                    num_heads=self.num_heads,
                    dropout=self.dropout,
                    param_dtype=self.param_dtype,
                    dtype=self.dtype,
                    use_flash_attention=self.use_flash_attention,
                    rngs=rngs,
                )
            )

        self.output_dense = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.out_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    def __call__(self, x_BTNI: jax.Array) -> jax.Array:
        x_BTNI = self.input_norm1(x_BTNI)
        x_BTNM = self.input_dense(x_BTNI)
        x_BTNM = self.input_norm2(x_BTNM)

        for block in self.blocks:
            x_BTNM = block(x_BTNM)

        x_BTNO = self.output_dense(x_BTNM)
        return x_BTNO

class TransformerBlock(nnx.Module):
    def __init__(
        self,
        model_dim: int,
        ffn_dim: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
    ):
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention
        self.decode = decode

        self.temporal_pos_enc = PositionalEncoding(self.model_dim)
        self.spatial_pos_enc = PositionalEncoding(self.model_dim)
        self.temporal_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.spatial_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.temporal_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.model_dim,
            qkv_features=self.model_dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=self.decode,
        )
        self.spatial_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.model_dim,
            qkv_features=self.model_dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=self.decode,
        )
        self.ffn_dense1 = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.ffn_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense2 = nnx.Linear(
            in_features=self.ffn_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    @nnx.remat
    def __call__(self, x_BTNM: jax.Array, pos_index: Tuple[jax.Array, jax.Array] | None = None) -> jax.Array:
        # --- Spatial attention ---
        B, T, N, M = x_BTNM.shape
        z_FNM = einops.rearrange(x_BTNM, ""b t n m -> (b t) n m"")
        z_FNM = self.spatial_norm(z_FNM)
        if self.decode:
            assert pos_index is not None
            z_FM = z_FNM[:, pos_index[1]]
            z_F1M = jnp.reshape(z_FM, (B * T, 1, M))
            z_F1M = self.spatial_attention(z_F1M)
            z_FM = jnp.reshape(z_F1M, (B * T, M))
            z_FNM = z_FNM.at[:, pos_index[1], :].set(z_FM)
        else:
            z_FNM = self.spatial_attention(z_FNM)
        z_BTNM = einops.rearrange(z_FNM, ""(b t) n m -> b t n m"", t=T)
        x_BTNM = x_BTNM + z_BTNM
        # --- Temporal attention ---
        z_PTM = einops.rearrange(x_BTNM, ""b t n m -> (b n) t m"")
        z_PTM = self.temporal_norm(z_PTM)
        if self.decode:
            assert pos_index is not None
            z_PM = z_PTM[:, pos_index[0]]
            z_P1M = jnp.reshape(z_PM, (B * N, 1, M))
            z_P1M = self.temporal_attention(z_P1M)
            z_PM = jnp.reshape(z_P1M, (B * N, M))
            z_PTM = z_PTM.at[:, pos_index[0], :].set(z_PM)
        else:
            z_PTM = self.temporal_attention(z_PTM)
        z_BTNM = einops.rearrange(z_PTM, ""(b n) t m -> b t n m"", n=N)
        x_BTNM = x_BTNM + z_BTNM
        # --- Feedforward ---
        z_BTNM = self.ffn_norm(x_BTNM)
        z_BTND = self.ffn_dense1(z_BTNM)
        z_BTND = jax.nn.gelu(z_BTND)
        z_BTNM = self.ffn_dense2(z_BTND)
        x_BTNM = x_BTNM + z_BTNM

        return x_BTNM

class Transformer(nnx.Module):
    """"""
    Dimension keys:
        B: batch size
        T: number of frames
        N: number of patches per frame
        I: number of input features
        M: model dimension
        D: FFN dimension
        O: number of output features
        F: number of frames in batch
        P: number of patch positions in batch
    """"""
    def __init__(
        self,
        input_dim: int,
        model_dim: int,
        ffn_dim: int,
        out_dim: int,
        num_blocks: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
    ):
        self.input_dim = input_dim
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.out_dim = out_dim
        self.num_blocks = num_blocks
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.pos_enc = PositionalEncoding(self.model_dim)
        self.input_norm1 = nnx.LayerNorm(
            num_features=self.input_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_dense = nnx.Linear(
            in_features=self.input_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_norm2 = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

        self.blocks: List[TransformerBlock] = []
        for _ in range(self.num_blocks):
            self.blocks.append(
                TransformerBlock(
                    model_dim=self.model_dim,
                    ffn_dim=self.ffn_dim,
                    num_heads=self.num_heads,
                    dropout=self.dropout,
                    param_dtype=self.param_dtype,
                    dtype=self.dtype,
                    use_flash_attention=self.use_flash_attention,
                    decode=decode,
                    rngs=rngs,
                )
            )
        self.output_dense = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.out_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    def __call__(self, x_BTNI: jax.Array, pos_index: Tuple[jax.Array, jax.Array] | None = None) -> jax.Array:
        x_BTNI = self.input_norm1(x_BTNI)
        x_BTNM = self.input_dense(x_BTNI)
        x_BTNM = self.input_norm2(x_BTNM)
        x_BTNM = self.pos_enc(x_BTNM)

        for block in self.blocks:
            x_BTNM = block(x_BTNM, pos_index)

        x_BTNV = self.output_dense(x_BTNM)
        return x_BTNV

def normalize(x: jax.Array) -> jax.Array:
    return x / (jnp.linalg.norm(x, ord=2, axis=-1, keepdims=True) + 1e-8)


class VectorQuantizer(nnx.Module):
    """"""
    Dimension keys:
        D: B * T * N
        K: number of latents
        L: latent dimension
    """"""
    def __init__(
        self, latent_dim: int, num_latents: int, dropout: float, rngs: nnx.Rngs
    ):
        self.latent_dim = latent_dim
        self.num_latents = num_latents
        self.dropout = dropout

        self.codebook = nnx.Param(
            normalize(
                nnx.initializers.lecun_uniform()(
                    rngs.params(), (self.num_latents, self.latent_dim)
                )
            )
        )
        self.drop = nnx.Dropout(self.dropout, rngs=rngs)

    def __call__(
        self, x_DL: jax.Array, training: bool
    ) -> Tuple[jax.Array, jax.Array, jax.Array, jax.Array]:
        # --- Compute distances ---
        x_DL = normalize(x_DL)
        normalized_codebook_KL = normalize(self.codebook.value)
        distance_DK = -jnp.matmul(x_DL, normalized_codebook_KL.T)
        if training:
            distance_DK = self.drop(distance_DK)

        # --- Get indices and embeddings ---
        indices_D = jnp.argmin(distance_DK, axis=-1)
        z_DL = self.codebook[indices_D]

        # --- Straight through estimator ---
        z_q_DL = x_DL + jax.lax.stop_gradient(z_DL - x_DL)
        return z_q_DL, z_DL, x_DL, indices_D

    def get_codes(self, indices_E: jax.Array) -> jax.Array:
        return self.codebook[indices_E]


def _create_flash_attention_fn(use_flash_attention: bool, is_causal: bool) -> Callable:
    """"""
    Create an attention function that uses flash attention if enabled.

    flax.nnx.MultiHeadAttention provides tensors with shape (batch..., length, num_heads, head_dim),
    but jax.nn.dot_product_attention expects (batch, length, num_heads, head_dim). We reshape to
    ensure compatibility. cuDNN's flash attention additionally requires a sequence length that
    is a multiple of 4. We pad the sequence length to the nearest multiple of 4 and mask
    accordingly. Note that cuDNN requires the mask to be broadcast before calling the attention
    function due to strict shape checking.
    """"""

    # FIXME (f.srambical): keys and values could have different dimensionalities
    def attention_fn(query_BSHD, key_BSHD, value_BSHD, bias=None, mask_B111=None, **kwargs):
        implementation = ""cudnn"" if use_flash_attention else None

        def _merge_batch_dims(x):
            return einops.rearrange(x, ""... l h k -> (...) l h k"")

        def _pad(x):
            return jnp.pad(x, ((0, 0), (0, pad_size), (0, 0), (0, 0)))

        original_shape = query_BSHD.shape
        original_seq_len = query_BSHD.shape[-3]

        # Pad to nearest multiple of 4
        T = ((original_seq_len + 3) // 4) * 4
        pad_size = T - original_seq_len

        query_BTHD = _pad(_merge_batch_dims(query_BSHD))
        key_BTHD = _pad(_merge_batch_dims(key_BSHD))
        value_BTHD = _pad(_merge_batch_dims(value_BSHD))
        B = query_BTHD.shape[0]

        attention_mask = jnp.ones((T, T), dtype=jnp.bool_)
        attention_mask = attention_mask.at[original_seq_len:, :].set(False)
        attention_mask = attention_mask.at[:, original_seq_len:].set(False)

        # Handle causal mask for cached decoder self-attention (from nnx.MultiHeadAttention)
        if mask_B111 is not None:
            mask_B111 = _merge_batch_dims(mask_B111)
            # We need to broadcast T and S dimensions to target_seq_len since cudnn attention strictly checks the mask shape
            # https://github.com/jax-ml/jax/issues/28974
            # https://github.com/jax-ml/jax/blob/08c7677393672ccb85c10f1ed0bd506905c3c994/jax/_src/cudnn/fused_attention_stablehlo.py#L1830
            # https://github.com/jax-ml/jax/blob/08c7677393672ccb85c10f1ed0bd506905c3c994/jax/_src/cudnn/fused_attention_stablehlo.py#L337
            mask_B1QK = einops.repeat(mask_B111, ""... 1 1 -> ... t s"", t=T, s=T)
            mask_B1QK = mask_B111.astype(jnp.bool)
        else:
            mask_11QK = attention_mask[jnp.newaxis, jnp.newaxis, :, :]
            mask_B1QK = jnp.broadcast_to(mask_11QK, (B, 1, T, T))

        bias_4d = _pad(_merge_batch_dims(bias)) if bias is not None else None

        # NOTE: jax.nn.dot_product_attention does not support dropout
        output_4d = jax.nn.dot_product_attention(
            query=query_BTHD,
            key=key_BTHD,
            value=value_BTHD,
            bias=bias_4d,
            mask=mask_B1QK,
            implementation=implementation,
            is_causal=is_causal,
        )
        return output_4d[..., :original_seq_len, :, :].reshape(original_shape)

    return attention_fn
",python,tab
30,120865,"TERMINAL",0,0,"2025-08-01 09:37:09.588292: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
31,121093,"genie.py",0,0,"",python,tab
32,122210,"genie.py",16726,0,"",python,selection_command
33,122355,"genie.py",16609,0,"",python,selection_command
34,122408,"TERMINAL",0,0,"2025-08-01 09:37:11.157786: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
35,122490,"genie.py",16499,0,"",python,selection_command
36,122637,"genie.py",16426,0,"",python,selection_command
37,122771,"genie.py",16352,0,"",python,selection_command
38,122932,"genie.py",16280,0,"",python,selection_command
39,123631,"genie.py",16264,71,"                # assert block.spatial_attention.cached_key is not None",python,selection_command
40,124182,"genie.py",16264,145,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None",python,selection_command
41,124349,"genie.py",16264,218,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None",python,selection_command
42,124490,"genie.py",16264,328,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
43,124632,"genie.py",16264,445,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.spatial_attention.cached_value[...] = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
44,125157,"genie.py",16264,537,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.spatial_attention.cached_value[...] = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.temporal_attention.cache_index[...] = jnp.array(0, dtype=jnp.int32)",python,selection_command
45,125409,"genie.py",16264,538,"",python,content
46,125428,"genie.py",16280,0,"",python,selection_command
47,126185,"TERMINAL",0,0,"2025-08-01 09:37:14.928589: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
48,128889,"genie.py",0,0,"",python,tab
49,128890,"genie.py",16264,0,"",python,selection_command
50,130438,"genie.py",0,0,"",python,tab
51,130439,"genie.py",14205,0,"",python,selection_command
52,133487,"TERMINAL",0,0,"2025-08-01 09:37:22.231109: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
53,138481,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
54,138574,"sample.py",0,0,"from dataclasses import dataclass
import time
import os
import optax
import math

import dm_pix as pix
import einops
import jax
import jax.numpy as jnp
import flax.linen as nn
import numpy as np
import orbax.checkpoint as ocp
from PIL import Image, ImageDraw
import tyro
from flax import nnx

from genie import Genie
from utils.dataloader import get_dataloader


@dataclass
class Args:
    # Experiment
    seed: int = 0
    seq_len: int = 16
    image_channels: int = 3
    image_height: int = 90
    image_width: int = 160
    data_dir: str = ""data/coinrun_episodes""
    checkpoint: str = """"
    # Sampling
    batch_size: int = 1
    maskgit_steps: int = 25
    temperature: float = 1.0
    sample_argmax: bool = True
    start_frame: int = 0
    # Tokenizer checkpoint
    tokenizer_dim: int = 512
    tokenizer_ffn_dim: int = 2048
    latent_patch_dim: int = 32
    num_patch_latents: int = 1024
    patch_size: int = 4
    tokenizer_num_blocks: int = 4
    tokenizer_num_heads: int = 8
    # LAM checkpoint
    lam_dim: int = 512
    lam_ffn_dim: int = 2048
    latent_action_dim: int = 32
    num_latent_actions: int = 6
    lam_patch_size: int = 16
    lam_num_blocks: int = 4
    lam_num_heads: int = 8
    # Dynamics checkpoint
    dyna_type: str = ""maskgit""
    dyna_dim: int = 512
    dyna_ffn_dim: int = 2048
    dyna_num_blocks: int = 6
    dyna_num_heads: int = 8
    param_dtype = jnp.float32
    dtype = jnp.bfloat16
    use_flash_attention: bool = True


args = tyro.cli(Args)

if __name__ == ""__main__"":
    """"""
    Dimension keys:
        B: batch size
        T: number of input (conditioning) frames
        N: number of patches per frame
        S: sequence length
        H: height
        W: width
        E: B * (S - 1)
    """"""
    jax.distributed.initialize()

    rng = jax.random.key(args.seed)

    # --- Load Genie checkpoint ---
    rngs = nnx.Rngs(rng)
    genie = Genie(
        # Tokenizer
        in_dim=args.image_channels,
        tokenizer_dim=args.tokenizer_dim,
        tokenizer_ffn_dim=args.tokenizer_ffn_dim,
        latent_patch_dim=args.latent_patch_dim,
        num_patch_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        tokenizer_num_blocks=args.tokenizer_num_blocks,
        tokenizer_num_heads=args.tokenizer_num_heads,
        # LAM
        lam_dim=args.lam_dim,
        lam_ffn_dim=args.lam_ffn_dim,
        latent_action_dim=args.latent_action_dim,
        num_latent_actions=args.num_latent_actions,
        lam_patch_size=args.lam_patch_size,
        lam_num_blocks=args.lam_num_blocks,
        lam_num_heads=args.lam_num_heads,
        lam_co_train=False,
        # Dynamics
        dyna_type=args.dyna_type,
        dyna_dim=args.dyna_dim,
        dyna_ffn_dim=args.dyna_ffn_dim,
        dyna_num_blocks=args.dyna_num_blocks,
        dyna_num_heads=args.dyna_num_heads,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        decode=True,
        rngs=rngs,
    )

    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeSave, ocp.handlers.PyTreeCheckpointHandler
    )
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )
    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    checkpoint_manager = ocp.CheckpointManager(
        args.checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )

    dummy_tx = optax.adamw(
        learning_rate=optax.linear_schedule(0.0001, 0.0001, 10000),
        b1=0.9,
        b2=0.9,
        weight_decay=1e-4,
        mu_dtype=args.dtype,
    )
    dummy_optimizer = nnx.Optimizer(genie, dummy_tx)

    abstract_optimizer = nnx.eval_shape(lambda: dummy_optimizer)
    abstract_optimizer_state = nnx.state(abstract_optimizer)
    restored = checkpoint_manager.restore(
        checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(abstract_optimizer_state),  # type: ignore
        ),
    )
    restored_optimizer_state = restored[""model_state""]
    nnx.update(dummy_optimizer, restored_optimizer_state)

    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame + 1]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)

    # --- Sample + evaluate video ---
    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)

    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
    recon_video_BSHWC = recon_video_BSHWC.astype(jnp.float32)
    gt = gt_video[:, : recon_video_BSHWC.shape[1]].clip(0, 1).reshape(-1, *gt_video.shape[2:])
    recon = recon_video_BSHWC.clip(0, 1).reshape(-1, *recon_video_BSHWC.shape[2:])
    ssim = jnp.asarray(
        pix.ssim(gt[:, args.start_frame + 1 :], recon[:, args.start_frame + 1 :])
    ).mean()
    print(f""SSIM: {ssim}"")

    # --- Construct video ---
    true_videos = (gt_video * 255).astype(np.uint8)
    pred_videos = (recon_video_BSHWC * 255).astype(np.uint8)
    video_comparison = np.zeros((2, *recon_video_BSHWC.shape), dtype=np.uint8)
    video_comparison[0] = true_videos[:, : args.seq_len]
    video_comparison[1] = pred_videos
    frames = einops.rearrange(video_comparison, ""n b t h w c -> t (b h) (n w) c"")

    # --- Save video ---
    imgs = [Image.fromarray(img) for img in frames]
    # Write actions on each frame, on each row (i.e., for each video in the batch, on the GT row)
    B, S, _, _, _ = video_batch_BSHWC.shape
    action_batch_BSm11 = jnp.reshape(action_batch_E, (B, S-1, 1))
    for t, img in enumerate(imgs[1:]):
        d = ImageDraw.Draw(img)
        for row in range(action_batch_BSm11.shape[0]):
            action = action_batch_BSm11[row, t, 0]
            y_offset = row * video_batch_BSHWC.shape[2] + 2
            d.text((2, y_offset), f""{action}"", fill=255)
    imgs[0].save(
        f""generation_{time.time()}.gif"",
        save_all=True,
        append_images=imgs[1:],
        duration=250,
        loop=0,
    )
",python,tab
55,138575,"sample.py",69,0,"",python,selection_command
56,140438,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
57,146612,"genie.py",0,0,"",python,tab
58,146612,"genie.py",14205,0,"",python,selection_command
59,152831,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
60,153300,"TERMINAL",0,0,"

",,terminal_output
61,164665,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
62,252897,"TERMINAL",0,0,"2025-08-01 09:39:21.642595: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
63,254492,"TERMINAL",0,0,"2025-08-01 09:39:23.220869: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
64,258281,"TERMINAL",0,0,"2025-08-01 09:39:27.025481: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
65,265592,"TERMINAL",0,0,"2025-08-01 09:39:34.336688: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
66,270636,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
67,272217,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
68,341006,"genie.py",0,0,"",python,tab
69,342594,"genie.py",16366,0,"",python,selection_command
70,343214,"genie.py",16383,0,"",python,selection_command
71,343264,"genie.py",16436,0,"",python,selection_command
72,343528,"genie.py",16470,0,"",python,selection_command
73,343548,"genie.py",16496,0,"",python,selection_command
74,343587,"genie.py",16536,0,"",python,selection_command
75,343618,"genie.py",16570,0,"",python,selection_command
76,343647,"genie.py",16618,0,"",python,selection_command
77,343679,"genie.py",16636,0,"",python,selection_command
78,343713,"genie.py",16686,0,"",python,selection_command
79,343746,"genie.py",16747,0,"",python,selection_command
80,343779,"genie.py",16765,0,"",python,selection_command
81,343818,"genie.py",16815,0,"",python,selection_command
82,343992,"genie.py",16765,0,"",python,selection_command
83,344704,"genie.py",16747,0,"",python,selection_command
84,345435,"genie.py",16686,0,"",python,selection_command
85,345487,"genie.py",16636,0,"",python,selection_command
86,346265,"genie.py",16618,0,"",python,selection_command
87,346323,"genie.py",16570,0,"",python,selection_command
88,346576,"genie.py",16536,0,"",python,selection_command
89,346605,"genie.py",16496,0,"",python,selection_command
90,346640,"genie.py",16470,0,"",python,selection_command
91,346666,"genie.py",16436,0,"",python,selection_command
92,346698,"genie.py",16383,0,"",python,selection_command
93,346734,"genie.py",16366,0,"",python,selection_command
94,346768,"genie.py",16280,0,"",python,selection_command
95,346799,"genie.py",16219,0,"",python,selection_command
96,346833,"genie.py",16155,0,"",python,selection_command
97,347097,"genie.py",16219,0,"",python,selection_command
98,347733,"genie.py",16280,0,"",python,selection_command
99,348582,"genie.py",16366,0,"",python,selection_command
100,350690,"genie.py",16280,0,"",python,selection_command
101,390605,"genie.py",0,0,"",python,tab
102,394549,"genie.py",14204,0,"",python,selection_mouse
103,394550,"genie.py",14203,0,"",python,selection_command
104,394919,"genie.py",14203,2,"s
",python,selection_mouse
105,394920,"genie.py",14204,1,"
",python,selection_command
106,395439,"genie.py",14205,0,"",python,selection_mouse
107,395848,"genie.py",14204,1,"
",python,selection_mouse
108,396921,"genie.py",14204,0,"",python,selection_mouse
109,396922,"genie.py",14203,0,"",python,selection_command
110,397293,"genie.py",14203,2,"s
",python,selection_mouse
111,397294,"genie.py",14204,1,"
",python,selection_command
112,397570,"genie.py",14205,0,"",python,selection_mouse
113,402117,"genie.py",0,0,"",python,tab
114,403845,"genie.py",16237,0,"",python,selection_command
115,404654,"genie.py",15174,0,"",python,selection_command
116,404892,"genie.py",14833,0,"",python,selection_command
117,405084,"genie.py",14638,0,"",python,selection_command
118,405241,"genie.py",14218,0,"",python,selection_command
119,407901,"genie.py",14205,0,"",python,selection_command
120,408482,"genie.py",14205,0,"                # assert block.spatial_attention.cached_key is not None
                # assert block.spatial_attention.cached_value is not None
                # assert block.spatial_attention.cache_index is not None
                # block.spatial_attention.cached_key = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.spatial_attention.cached_value[...] = jnp.zeros((B * T, 1, self.dyna_dim), dtype=self.dtype)
                # block.temporal_attention.cache_index[...] = jnp.array(0, dtype=jnp.int32)
",python,content
121,408493,"genie.py",14221,0,"",python,selection_command
122,409792,"genie.py",14205,538,"",python,content
123,410559,"genie.py",14149,0,"",python,selection_command
124,410643,"genie.py",14204,0,"
        ",python,content
125,411234,"genie.py",14213,0,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * T, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * N, 1, self.dyna_dim), dtype=self.dtype)
",python,content
126,412620,"genie.py",14362,0,"",python,selection_command
127,412694,"genie.py",14270,0,"",python,selection_command
128,412827,"genie.py",14205,0,"",python,selection_command
129,412962,"genie.py",14149,0,"",python,selection_command
130,413448,"genie.py",14205,0,"",python,selection_command
131,413851,"genie.py",14205,64,"                for block in dynamics_causal.transformer.blocks:",python,selection_command
132,414133,"genie.py",14205,1," ",python,selection_command
133,415368,"genie.py",14205,64,"                for block in dynamics_causal.transformer.blocks:",python,selection_command
134,415956,"genie.py",14205,16,"            ",python,content
135,415958,"genie.py",14217,0,"",python,selection_command
136,416755,"genie.py",14205,60,"            for block in dynamics_causal.transformer.blocks:",python,selection_command
137,417249,"genie.py",14205,12,"        ",python,content
138,417251,"genie.py",14213,0,"",python,selection_command
139,419137,"genie.py",14270,0,"",python,selection_command
140,419221,"genie.py",14362,0,"",python,selection_command
141,419588,"genie.py",14447,0,"",python,selection_command
142,420042,"genie.py",14447,1,"",python,content
143,421626,"genie.py",14354,0,"",python,selection_command
144,421688,"genie.py",14262,0,"",python,selection_command
145,421842,"genie.py",14205,0,"",python,selection_command
146,422443,"genie.py",14204,0,"
        ",python,content
147,422847,"genie.py",14205,8,"",python,content
148,424040,"genie.py",14206,0,"",python,selection_command
149,430314,"sample.py",0,0,"",python,tab
150,431530,"sample.py",6763,0,"",python,selection_command
151,431782,"sample.py",6675,0,"",python,selection_command
152,432044,"sample.py",6763,0,"",python,selection_command
153,432307,"sample.py",6675,0,"",python,selection_command
154,432547,"sample.py",6623,0,"",python,selection_command
155,433517,"sample.py",6619,0,"",python,selection_command
156,434954,"sample.py",6758,0," #",python,content
157,434955,"sample.py",6670,0," #",python,content
158,434955,"sample.py",6619,0,"# ",python,content
159,437916,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
160,438132,"TERMINAL",0,0,"

",,terminal_output
161,449394,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
162,457621,"TERMINAL",0,0,"2025-08-01 09:42:46.364099: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
163,459125,"TERMINAL",0,0,"2025-08-01 09:42:47.869151: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
164,462900,"TERMINAL",0,0,"2025-08-01 09:42:51.636749: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
165,463650,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 405, in sample_causal
    final_carry_causal, _ = jax.lax.scan(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 371, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 337, in __call__
    z_P1M = self.temporal_attention(z_P1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 556, in __call__
    raise ValueError(
ValueError: Autoregressive cache shape error, expected query shape (920, 1, 8, 64) instead got (921, 1, 8, 64).
",,terminal_output
166,464788,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
167,470175,"genie.py",0,0,"",python,tab
168,471999,"genie.py",14263,0,"",python,selection_command
169,472044,"genie.py",14275,0,"",python,selection_command
170,472296,"genie.py",14279,0,"",python,selection_command
171,472471,"genie.py",14280,0,"",python,selection_command
172,472633,"genie.py",14297,0,"",python,selection_command
173,472880,"genie.py",14298,0,"",python,selection_command
174,473015,"genie.py",14308,0,"",python,selection_command
175,473154,"genie.py",14310,0,"",python,selection_command
176,473297,"genie.py",14311,0,"",python,selection_command
177,473488,"genie.py",14313,0,"",python,selection_command
178,473622,"genie.py",14315,0,"",python,selection_command
179,473750,"genie.py",14316,0,"",python,selection_command
180,474903,"genie.py",14315,0,"",python,selection_command
181,474903,"genie.py",14314,0,"",python,selection_command
182,475515,"genie.py",14406,0,"",python,selection_command
183,475814,"genie.py",14407,0,"",python,selection_command
184,475951,"genie.py",14408,0,"",python,selection_command
185,477206,"genie.py",14316,0,"",python,selection_command
186,477220,"genie.py",14315,0,"",python,selection_command
187,479183,"sample.py",0,0,"",python,tab
188,480780,"genie.py",0,0,"",python,tab
189,482020,"genie.py",14315,1,"S",python,content
190,482936,"sample.py",0,0,"",python,tab
191,484096,"genie.py",0,0,"",python,tab
192,484538,"genie.py",14407,0,"",python,selection_command
193,485150,"genie.py",14408,0,"",python,selection_command
194,485276,"genie.py",14316,0,"",python,selection_command
195,485444,"genie.py",14315,0,"",python,selection_command
196,487877,"genie.py",13115,0,"",python,selection_command
197,488464,"genie.py",12019,0,"",python,selection_command
198,489399,"genie.py",10976,0,"",python,selection_command
199,489982,"genie.py",9571,0,"",python,selection_command
200,491242,"genie.py",8058,0,"",python,selection_command
201,492622,"genie.py",9571,0,"",python,selection_command
202,492775,"genie.py",10976,0,"",python,selection_command
203,492975,"genie.py",12019,0,"",python,selection_command
204,493211,"genie.py",13115,0,"",python,selection_command
205,494610,"genie.py",14275,0,"",python,selection_command
206,499321,"genie.py",14218,0,"",python,selection_command
207,499579,"genie.py",14205,0,"",python,selection_command
208,499592,"genie.py",14161,0,"",python,selection_command
209,499624,"genie.py",14094,0,"",python,selection_command
210,499796,"genie.py",14022,0,"",python,selection_command
211,499983,"genie.py",13959,0,"",python,selection_command
212,500298,"genie.py",14022,0,"",python,selection_command
213,500759,"genie.py",13959,0,"",python,selection_command
214,500919,"genie.py",13919,0,"",python,selection_command
215,501070,"genie.py",13880,0,"",python,selection_command
216,502052,"genie.py",13919,0,"",python,selection_command
217,502660,"genie.py",13959,0,"",python,selection_command
218,505201,"genie.py",14022,0,"",python,selection_command
219,505633,"genie.py",14081,0,"
        ",python,content
220,506287,"genie.py",14090,0,"S",python,content
221,506287,"genie.py",14091,0,"",python,selection_keyboard
222,506816,"genie.py",14091,0," = seq_len",python,content
223,507251,"genie.py",14100,0,"",python,selection_command
224,508249,"genie.py",14082,20,"",python,content
225,508261,"genie.py",14090,0,"",python,selection_command
226,509377,"genie.py",14157,0,"",python,selection_command
227,509625,"genie.py",14205,0,"",python,selection_command
228,509659,"genie.py",14214,0,"",python,selection_command
229,509690,"genie.py",14271,0,"",python,selection_command
230,509727,"genie.py",14363,0,"",python,selection_command
231,509755,"genie.py",14448,0,"",python,selection_command
232,509789,"genie.py",14457,0,"",python,selection_command
233,509822,"genie.py",14485,0,"",python,selection_command
234,509856,"genie.py",14573,0,"",python,selection_command
235,509888,"genie.py",14650,0,"",python,selection_command
236,509932,"genie.py",14716,0,"",python,selection_command
237,509965,"genie.py",14760,0,"",python,selection_command
238,509988,"genie.py",14795,0,"",python,selection_command
239,510022,"genie.py",14804,0,"",python,selection_command
240,510055,"genie.py",14851,0,"",python,selection_command
241,510089,"genie.py",14916,0,"",python,selection_command
242,510123,"genie.py",14925,0,"",python,selection_command
243,510158,"genie.py",14966,0,"",python,selection_command
244,510191,"genie.py",15045,0,"",python,selection_command
245,510230,"genie.py",15122,0,"",python,selection_command
246,510291,"genie.py",15045,0,"",python,selection_command
247,510567,"genie.py",14966,0,"",python,selection_command
248,510578,"genie.py",14925,0,"",python,selection_command
249,510612,"genie.py",14916,0,"",python,selection_command
250,510645,"genie.py",14851,0,"",python,selection_command
251,510678,"genie.py",14804,0,"",python,selection_command
252,510717,"genie.py",14795,0,"",python,selection_command
253,510863,"genie.py",14804,0,"",python,selection_command
254,511092,"genie.py",14795,0,"",python,selection_command
255,511345,"genie.py",14760,0,"",python,selection_command
256,511368,"genie.py",14716,0,"",python,selection_command
257,511403,"genie.py",14650,0,"",python,selection_command
258,511435,"genie.py",14573,0,"",python,selection_command
259,511468,"genie.py",14485,0,"",python,selection_command
260,511501,"genie.py",14457,0,"",python,selection_command
261,511535,"genie.py",14448,0,"",python,selection_command
262,511694,"genie.py",14457,0,"",python,selection_command
263,511912,"genie.py",14448,0,"",python,selection_command
264,512076,"genie.py",14363,0,"",python,selection_command
265,512222,"genie.py",14271,0,"",python,selection_command
266,512562,"genie.py",14275,0,"",python,selection_command
267,512819,"genie.py",14280,0,"",python,selection_command
268,512844,"genie.py",14281,0,"",python,selection_command
269,512880,"genie.py",14298,0,"",python,selection_command
270,512913,"genie.py",14299,0,"",python,selection_command
271,512944,"genie.py",14309,0,"",python,selection_command
272,513000,"genie.py",14311,0,"",python,selection_command
273,513021,"genie.py",14313,0,"",python,selection_command
274,513049,"genie.py",14315,0,"",python,selection_command
275,513720,"genie.py",14315,1,"",python,content
276,514065,"genie.py",14315,0,"s",python,content
277,514065,"genie.py",14316,0,"",python,selection_keyboard
278,514164,"genie.py",14316,0,"e",python,content
279,514165,"genie.py",14317,0,"",python,selection_keyboard
280,514247,"genie.py",14317,0,"q",python,content
281,514247,"genie.py",14318,0,"",python,selection_keyboard
282,514751,"genie.py",14318,0,"_",python,content
283,514751,"genie.py",14319,0,"",python,selection_keyboard
284,515137,"genie.py",14319,0,"l",python,content
285,515138,"genie.py",14320,0,"",python,selection_keyboard
286,515151,"genie.py",14320,0,"e",python,content
287,515152,"genie.py",14321,0,"",python,selection_keyboard
288,515278,"genie.py",14321,0,"n",python,content
289,515278,"genie.py",14322,0,"",python,selection_keyboard
290,515550,"genie.py",14321,0,"",python,selection_command
291,516063,"genie.py",14419,0,"",python,selection_command
292,516216,"genie.py",14418,0,"",python,selection_command
293,516464,"genie.py",14417,0,"",python,selection_command
294,516497,"genie.py",14416,0,"",python,selection_command
295,516529,"genie.py",14415,0,"",python,selection_command
296,516689,"genie.py",14414,0,"",python,selection_command
297,518291,"genie.py",14414,0,"(",python,content
298,518291,"genie.py",14415,0,"",python,selection_keyboard
299,519386,"genie.py",14416,0,"",python,selection_command
300,520271,"genie.py",14416,0,")",python,content
301,520271,"genie.py",14417,0,"",python,selection_keyboard
302,521072,"genie.py",14416,0,"",python,selection_command
303,521704,"genie.py",14416,0," ",python,content
304,521704,"genie.py",14417,0,"",python,selection_keyboard
305,522743,"genie.py",14417,0,"=",python,content
306,522743,"genie.py",14418,0,"",python,selection_keyboard
307,523005,"genie.py",14417,1,"",python,content
308,523340,"genie.py",14417,0,"+",python,content
309,523340,"genie.py",14418,0,"",python,selection_keyboard
310,523656,"genie.py",14418,0," ",python,content
311,523656,"genie.py",14419,0,"",python,selection_keyboard
312,523891,"genie.py",14419,0,"1",python,content
313,523891,"genie.py",14420,0,"",python,selection_keyboard
314,524223,"genie.py",14419,0,"",python,selection_command
315,525952,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
316,526141,"TERMINAL",0,0,"

",,terminal_output
317,537576,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
318,550910,"TERMINAL",0,0,"2025-08-01 09:44:19.650430: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
319,552427,"TERMINAL",0,0,"2025-08-01 09:44:21.171110: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
320,556234,"TERMINAL",0,0,"2025-08-01 09:44:24.980448: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
321,563576,"TERMINAL",0,0,"2025-08-01 09:44:32.323021: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
322,568536,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
323,570209,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
324,597569,"genie.py",14488,0,"",python,selection_mouse
325,597570,"genie.py",14487,0,"",python,selection_command
326,620986,"sample.py",0,0,"",python,tab
327,620986,"sample.py",69,0,"",python,selection_command
328,622301,"genie.py",0,0,"",python,tab
329,622301,"genie.py",14205,0,"",python,selection_command
330,623641,"genie.py",0,0,"",python,tab
331,623642,"genie.py",14206,0,"",python,selection_command
332,625995,"sample.py",0,0,"",python,tab
333,625995,"sample.py",6615,0,"",python,selection_command
334,628323,"sample.py",6624,0,"",python,selection_mouse
335,636276,"sample.py",6615,54,"",python,content
336,636300,"sample.py",6619,0,"",python,selection_command
337,636804,"sample.py",6615,90,"",python,content
338,636807,"sample.py",6619,0,"",python,selection_command
339,637396,"sample.py",6615,97,"",python,content
340,644777,"genie.py",0,0,"",python,tab
341,644777,"genie.py",14206,0,"",python,selection_command
342,647217,"sample.py",0,0,"",python,tab
343,647218,"sample.py",69,0,"",python,selection_command
344,655479,"sample.py",6503,0,"",python,selection_mouse
345,658502,"sample.py",7458,0,"",python,selection_command
346,659968,"sample.py",7474,0,"",python,selection_command
347,660378,"sample.py",7669,0,"",python,selection_command
348,662472,"sample.py",7474,0,"",python,selection_command
349,662669,"sample.py",7458,0,"",python,selection_command
350,662973,"sample.py",6503,0,"",python,selection_command
351,668770,"sample.py",0,0,"",python,tab
352,670168,"sample.py",6543,0,"",python,selection_command
353,670226,"sample.py",6499,0,"",python,selection_command
354,670903,"sample.py",6499,44,"",python,content
355,670915,"sample.py",6503,0,"",python,selection_command
356,671327,"sample.py",6499,72,"",python,content
357,671648,"sample.py",6499,1,"",python,content
358,671651,"sample.py",6503,0,"",python,selection_command
359,678326,"sample.py",0,0,"",python,tab
360,678328,"sample.py",6499,0,"",python,selection_command
361,680230,"sample.py",0,0,"",python,tab
362,680232,"sample.py",69,0,"",python,selection_command
363,680372,"sample.py",6499,117,"",python,content
364,684329,"genie.py",0,0,"",python,tab
365,684329,"genie.py",14205,0,"",python,selection_command
366,871225,"experiments/sample.sh",0,0,"source .venv/bin/activate

data_dir=""$PWD/data_arrayrecord/dummy""
ckpt_dir=""$PWD/checkpoints/causal_dynamics_openai_grain_tok_restore""

export XLA_FLAGS='--xla_gpu_autotune_level=0'
export PYTHONUNBUFFERED=1
srun python sample.py \
    --dyna_type ""causal"" \
    --batch_size 1 \
    --seq_len 3 \
    --start_frame 1 \
    --checkpoint $ckpt_dir \
    --data_dir $data_dir",shellscript,tab
367,872611,"experiments/sample.sh",294,1,"r",shellscript,content
368,874394,"experiments/sample.sh",294,1,"2",shellscript,content
369,877009,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
370,877282,"TERMINAL",0,0,"

",,terminal_output
371,888527,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
372,896769,"TERMINAL",0,0,"2025-08-01 09:50:05.511569: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
373,898208,"TERMINAL",0,0,"2025-08-01 09:50:06.954445: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
374,901795,"TERMINAL",0,0,"2025-08-01 09:50:10.539964: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
375,902717,"TERMINAL",0,0,"2025-08-01 09:50:11.458306: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
376,905111,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
377,906513,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
378,910979,"experiments/sample.sh",294,1,"4",shellscript,content
379,915406,"experiments/sample.sh",294,1,"3",shellscript,content
380,917024,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
381,917476,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
382,917729,"TERMINAL",0,0,"

",,terminal_output
383,928674,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
384,936956,"TERMINAL",0,0,"2025-08-01 09:50:45.652567: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
385,938424,"TERMINAL",0,0,"2025-08-01 09:50:47.170728: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
386,942232,"TERMINAL",0,0,"2025-08-01 09:50:50.975595: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
387,949583,"TERMINAL",0,0,"2025-08-01 09:50:58.328808: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
388,954570,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
389,956213,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
390,975409,"experiments/sample.sh",273,0,"",shellscript,selection_command
391,975457,"experiments/sample.sh",246,0,"",shellscript,selection_command
392,975690,"experiments/sample.sh",222,0,"",shellscript,selection_command
393,975852,"experiments/sample.sh",196,0,"",shellscript,selection_command
394,976083,"experiments/sample.sh",150,0,"",shellscript,selection_command
395,977513,"experiments/sample.sh",136,46,"",shellscript,content
396,979565,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
397,979789,"TERMINAL",0,0,"

",,terminal_output
398,991141,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
399,999955,"TERMINAL",0,0,"2025-08-01 09:51:48.697941: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
400,1004443,"TERMINAL",0,0,"2025-08-01 09:51:53.185892: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
401,1013497,"TERMINAL",0,0,"2025-08-01 09:52:02.239293: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
402,1027486,"TERMINAL",0,0,"2025-08-01 09:52:16.231965: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
403,1040447,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
404,1042144,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
405,1047773,"sample.py",0,0,"",python,tab
406,1049108,"sample.py",6465,0,"",python,selection_command
407,1049442,"sample.py",6503,0,"",python,selection_command
408,1052616,"sample.py",6498,0,"
    ",python,content
409,1053790,"sample.py",6503,0,"j",python,content
410,1053791,"sample.py",6504,0,"",python,selection_keyboard
411,1053813,"sample.py",6504,0,"a",python,content
412,1053813,"sample.py",6505,0,"",python,selection_keyboard
413,1053945,"sample.py",6505,0,"x",python,content
414,1053945,"sample.py",6506,0,"",python,selection_keyboard
415,1054031,"sample.py",6506,0,".",python,content
416,1054031,"sample.py",6507,0,"",python,selection_keyboard
417,1054824,"sample.py",6507,0,"b",python,content
418,1054824,"sample.py",6508,0,"",python,selection_keyboard
419,1054979,"sample.py",6508,0,"l",python,content
420,1054979,"sample.py",6509,0,"",python,selection_keyboard
421,1055074,"sample.py",6509,0,"o",python,content
422,1055074,"sample.py",6510,0,"",python,selection_keyboard
423,1055499,"sample.py",6510,0,"c",python,content
424,1055499,"sample.py",6511,0,"",python,selection_keyboard
425,1055699,"sample.py",6511,0,"k",python,content
426,1055699,"sample.py",6512,0,"",python,selection_keyboard
427,1058004,"sample.py",6511,0,"",python,selection_command
428,1058647,"sample.py",6499,13,"    ",python,content
429,1189341,"sample.py",6503,0,"t",python,content
430,1189341,"sample.py",6504,0,"",python,selection_keyboard
431,1189379,"sample.py",6504,0,"i",python,content
432,1189379,"sample.py",6505,0,"",python,selection_keyboard
433,1189453,"sample.py",6505,0,"m",python,content
434,1189453,"sample.py",6506,0,"",python,selection_keyboard
435,1189503,"sample.py",6506,0,"e",python,content
436,1189503,"sample.py",6507,0,"",python,selection_keyboard
437,1193759,"sample.py",6507,0,"i",python,content
438,1193759,"sample.py",6508,0,"",python,selection_keyboard
439,1193905,"sample.py",6508,0,"t",python,content
440,1193905,"sample.py",6509,0,"",python,selection_keyboard
441,1194612,"sample.py",6508,0,"",python,selection_command
442,1195139,"sample.py",0,0,"",python,selection_command
443,1197186,"sample.py",6503,6,"jax.block",python,content
444,1197192,"sample.py",6511,0,"",python,selection_command
445,1198637,"sample.py",6503,14,"",python,content
446,1201416,"sample.py",6498,0,"
    ",python,content
447,1201561,"sample.py",6503,0,"t",python,content
448,1201561,"sample.py",6504,0,"",python,selection_keyboard
449,1201627,"sample.py",6504,0,"i",python,content
450,1201627,"sample.py",6505,0,"",python,selection_keyboard
451,1201703,"sample.py",6505,0,"m",python,content
452,1201703,"sample.py",6506,0,"",python,selection_keyboard
453,1201750,"sample.py",6506,0,"e",python,content
454,1201751,"sample.py",6507,0,"",python,selection_keyboard
455,1201952,"sample.py",6507,0,".",python,content
456,1201953,"sample.py",6508,0,"",python,selection_keyboard
457,1202320,"sample.py",6508,0,"i",python,content
458,1202321,"sample.py",6509,0,"",python,selection_keyboard
459,1202679,"sample.py",6508,1,"",python,content
460,1202828,"sample.py",6507,1,"",python,content
461,1202861,"sample.py",6507,0,"i",python,content
462,1202861,"sample.py",6508,0,"",python,selection_keyboard
463,1203048,"sample.py",6508,0,"t",python,content
464,1203049,"sample.py",6509,0,"",python,selection_keyboard
465,1203204,"sample.py",6509,0,".",python,content
466,1203204,"sample.py",6510,0,"",python,selection_keyboard
467,1204010,"sample.py",6510,0,"t",python,content
468,1204010,"sample.py",6511,0,"",python,selection_keyboard
469,1204056,"sample.py",6511,0,"i",python,content
470,1204056,"sample.py",6512,0,"",python,selection_keyboard
471,1204095,"sample.py",6512,0,"m",python,content
472,1204095,"sample.py",6513,0,"",python,selection_keyboard
473,1204173,"sample.py",6513,0,"e",python,content
474,1204173,"sample.py",6514,0,"",python,selection_keyboard
475,1205397,"sample.py",6514,0,"i",python,content
476,1205397,"sample.py",6515,0,"",python,selection_keyboard
477,1205562,"sample.py",6515,0,"t(lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E), number=1)",python,content
478,1206633,"sample.py",6597,0,"",python,selection_command
479,1207596,"sample.py",0,0,"",python,selection_command
480,1208848,"sample.py",80,0,"
import timeit",python,content
481,1209776,"sample.py",81,0,"",python,selection_command
482,1209852,"sample.py",69,0,"",python,selection_command
483,1210271,"sample.py",69,12,"",python,content
484,1212321,"sample.py",8212,0,"",python,selection_command
485,1212909,"sample.py",8206,0,"",python,selection_command
486,1213160,"sample.py",8190,0,"",python,selection_command
487,1213193,"sample.py",8168,0,"",python,selection_command
488,1213224,"sample.py",8136,0,"",python,selection_command
489,1213259,"sample.py",8113,0,"",python,selection_command
490,1213300,"sample.py",8072,0,"",python,selection_command
491,1213326,"sample.py",8054,0,"",python,selection_command
492,1213363,"sample.py",7997,0,"",python,selection_command
493,1213397,"sample.py",7937,0,"",python,selection_command
494,1213430,"sample.py",7886,0,"",python,selection_command
495,1213464,"sample.py",7831,0,"",python,selection_command
496,1213499,"sample.py",7799,0,"",python,selection_command
497,1213532,"sample.py",7760,0,"",python,selection_command
498,1213566,"sample.py",7694,0,"",python,selection_command
499,1213599,"sample.py",7650,0,"",python,selection_command
500,1213633,"sample.py",7552,0,"",python,selection_command
501,1213666,"sample.py",7500,0,"",python,selection_command
502,1213698,"sample.py",7475,0,"",python,selection_command
503,1213732,"sample.py",7474,0,"",python,selection_command
504,1213764,"sample.py",7392,0,"",python,selection_command
505,1213799,"sample.py",7354,0,"",python,selection_command
506,1217632,"sample.py",76,0,"",python,selection_command
507,1218296,"sample.py",6505,0,"",python,selection_command
508,1220149,"sample.py",6511,0,"",python,selection_command
509,1220337,"sample.py",6512,0,"",python,selection_command
510,1275941,"sample.py",6518,0,"",python,selection_command
511,1276117,"sample.py",6519,0,"",python,selection_command
512,1277133,"sample.py",6518,0,"",python,selection_command
513,1277904,"sample.py",6519,0,"",python,selection_command
514,1278997,"sample.py",6519,80,"",python,content
515,1279894,"sample.py",6518,0,"",python,selection_command
516,1280664,"experiments/sample.sh",0,0,"",shellscript,tab
517,1281945,"sample.py",0,0,"",python,tab
518,1282533,"sample.py",6519,0,"",python,selection_command
519,1283094,"experiments/sample.sh",0,0,"",shellscript,tab
520,1284024,"sample.py",0,0,"",python,tab
521,1301286,"sample.py",6518,0,"",python,selection_command
522,1301711,"sample.py",6538,0,"",python,selection_command
523,1301815,"sample.py",6625,0,"",python,selection_command
524,1302188,"sample.py",6538,0,"",python,selection_command
525,1302518,"sample.py",6543,0,"",python,selection_command
526,1302680,"sample.py",6545,0,"",python,selection_command
527,1303413,"sample.py",6545,1,"_",python,selection_command
528,1303535,"sample.py",6545,15,"_autoreg_sample",python,selection_command
529,1303784,"sample.py",6545,16,"_autoreg_sample(",python,selection_command
530,1303937,"sample.py",6545,21,"_autoreg_sample(genie",python,selection_command
531,1304185,"sample.py",6545,22,"_autoreg_sample(genie,",python,selection_command
532,1304215,"sample.py",6545,26,"_autoreg_sample(genie, rng",python,selection_command
533,1304239,"sample.py",6545,27,"_autoreg_sample(genie, rng,",python,selection_command
534,1304274,"sample.py",6545,45,"_autoreg_sample(genie, rng, video_batch_BSHWC",python,selection_command
535,1304495,"sample.py",6545,46,"_autoreg_sample(genie, rng, video_batch_BSHWC,",python,selection_command
536,1304659,"sample.py",6545,61,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E",python,selection_command
537,1304804,"sample.py",6545,62,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)",python,selection_command
538,1305589,"sample.py",6545,0,"",python,selection_command
539,1306053,"sample.py",6519,0,"",python,selection_command
540,1306119,"sample.py",6518,0,"",python,selection_command
541,1306278,"sample.py",6517,0,"",python,selection_command
542,1306699,"sample.py",6518,0,"",python,selection_command
543,1307178,"sample.py",6519,0,"",python,selection_command
544,1307327,"sample.py",6519,0,"''",python,content
545,1307327,"sample.py",6520,0,"",python,selection_keyboard
546,1307532,"sample.py",6520,1,"'",python,content
547,1307532,"sample.py",6521,0,"",python,selection_keyboard
548,1307741,"sample.py",6520,0,"",python,selection_command
549,1308257,"sample.py",6519,0,"",python,selection_command
550,1308757,"sample.py",6520,0,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)",python,content
551,1308762,"sample.py",6581,0,"",python,selection_command
552,1319699,"sample.py",6582,0,"",python,selection_command
553,1319711,"sample.py",6582,0,".",python,content
554,1319711,"sample.py",6583,0,"",python,selection_keyboard
555,1320080,"sample.py",6583,0,"b",python,content
556,1320081,"sample.py",6584,0,"",python,selection_keyboard
557,1320867,"sample.py",6585,0,", number=1",python,content
558,1320867,"sample.py",6584,0,"lock_until_ready()",python,content
559,1320869,"sample.py",6613,0,"",python,selection_command
560,1321282,"sample.py",6612,0,"",python,selection_command
561,1322070,"sample.py",6612,1,"1",python,selection_command
562,1322413,"sample.py",6611,2,"=1",python,selection_command
563,1322618,"sample.py",6605,8,"number=1",python,selection_command
564,1322804,"sample.py",6600,13,"()', number=1",python,selection_command
565,1323297,"sample.py",6601,12,")', number=1",python,selection_command
566,1323470,"sample.py",6602,11,"', number=1",python,selection_command
567,1323622,"sample.py",6603,10,", number=1",python,selection_command
568,1323731,"sample.py",6603,10,"",python,content
569,1324262,"sample.py",6602,0,"",python,selection_command
570,1324486,"sample.py",6601,0,"",python,selection_command
571,1324998,"sample.py",6600,0,"",python,selection_command
572,1325552,"sample.py",6690,0,"",python,selection_command
573,1326001,"sample.py",6676,0,"",python,selection_command
574,1326772,"sample.py",6689,0,"",python,selection_command
575,1327436,"sample.py",6690,0,"",python,selection_command
576,1327472,"sample.py",6690,1,")",python,selection_command
577,1327728,"sample.py",6676,15,"action_batch_E)",python,selection_command
578,1327836,"sample.py",6674,17,", action_batch_E)",python,selection_command
579,1328190,"sample.py",6657,34,"video_batch_BSHWC, action_batch_E)",python,selection_command
580,1328341,"sample.py",6655,36,", video_batch_BSHWC, action_batch_E)",python,selection_command
581,1328490,"sample.py",6652,39,"rng, video_batch_BSHWC, action_batch_E)",python,selection_command
582,1328626,"sample.py",6650,41,", rng, video_batch_BSHWC, action_batch_E)",python,selection_command
583,1328757,"sample.py",6645,46,"genie, rng, video_batch_BSHWC, action_batch_E)",python,selection_command
584,1328959,"sample.py",6644,47,"(genie, rng, video_batch_BSHWC, action_batch_E)",python,selection_command
585,1329140,"sample.py",6629,62,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)",python,selection_command
586,1329743,"sample.py",6629,0,"",python,selection_command
587,1330127,"sample.py",6525,0,"",python,selection_command
588,1330303,"sample.py",6487,0,"",python,selection_command
589,1330709,"sample.py",6500,0,"
    ",python,content
590,1331037,"sample.py",6501,4,"",python,content
591,1331054,"sample.py",6501,0,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)",python,content
592,1331060,"sample.py",6562,0,"",python,selection_command
593,1332365,"sample.py",6501,0,"",python,selection_command
594,1332633,"sample.py",6501,0,"    ",python,content
595,1333040,"sample.py",6504,0,"",python,selection_command
596,1335500,"sample.py",6567,0,".block_until_ready()",python,content
597,1335501,"sample.py",6587,0,"",python,selection_command
598,1336388,"sample.py",6586,0,"",python,selection_command
599,1355332,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
600,1355628,"TERMINAL",0,0,"

",,terminal_output
601,1357819,"experiments/sample.sh",0,0,"",shellscript,tab
602,1360939,"experiments/sample.sh",162,0,"",shellscript,selection_command
603,1361015,"experiments/sample.sh",186,0,"",shellscript,selection_command
604,1361162,"experiments/sample.sh",213,0,"",shellscript,selection_command
605,1361298,"experiments/sample.sh",234,0,"",shellscript,selection_command
606,1361752,"experiments/sample.sh",238,0,"",shellscript,selection_command
607,1361937,"experiments/sample.sh",240,0,"",shellscript,selection_command
608,1362113,"experiments/sample.sh",248,0,"",shellscript,selection_command
609,1362239,"experiments/sample.sh",250,0,"",shellscript,selection_command
610,1362797,"experiments/sample.sh",248,0,"",shellscript,selection_command
611,1363502,"experiments/sample.sh",248,1,"2",shellscript,content
612,1366871,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
613,1367174,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.7 task 0: running
",,terminal_output
614,1367261,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.7
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T09:57:56.001] error: *** STEP 14571.7 ON hai005 CANCELLED AT 2025-08-01T09:57:56 DUE to SIGNAL Killed ***
",,terminal_output
615,1367405,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.7
srun: job abort in progress
",,terminal_output
616,1367558,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
617,1367962,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
618,1368335,"TERMINAL",0,0,"

",,terminal_output
619,1369639,"sample.py",0,0,"",python,tab
620,1379229,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
621,1388271,"TERMINAL",0,0,"2025-08-01 09:58:17.018850: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
622,1392541,"TERMINAL",0,0,"2025-08-01 09:58:21.263885: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
623,1401318,"TERMINAL",0,0,"2025-08-01 09:58:30.061955: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
624,1405213,"TERMINAL",0,0,"2025-08-01 09:58:33.958141: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
625,1407594,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 211, in <module>
    timeit.timeit('_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()')
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
NameError: name '_autoreg_sample' is not defined
",,terminal_output
626,1408717,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
627,1430590,"sample.py",6618,0,"",python,selection_mouse
628,1437019,"sample.py",6588,103,"    timeit.timeit('_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()')",python,selection_command
629,1437676,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
630,1475066,"sample.py",6588,103,"    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,content
631,1503817,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
632,1504027,"TERMINAL",0,0,"

",,terminal_output
633,1511423,"sample.py",6676,0,"",python,selection_mouse
634,1515416,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
635,1516708,"/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py",0,0,"#! /usr/bin/env python3

""""""Tool for measuring execution time of small code snippets.

This module avoids a number of common traps for measuring execution
times.  See also Tim Peters' introduction to the Algorithms chapter in
the Python Cookbook, published by O'Reilly.

Library usage: see the Timer class.

Command line usage:
    python timeit.py [-n N] [-r N] [-s S] [-p] [-h] [--] [statement]

Options:
  -n/--number N: how many times to execute 'statement' (default: see below)
  -r/--repeat N: how many times to repeat the timer (default 5)
  -s/--setup S: statement to be executed once initially (default 'pass').
                Execution time of this setup statement is NOT timed.
  -p/--process: use time.process_time() (default is time.perf_counter())
  -v/--verbose: print raw timing results; repeat for more digits precision
  -u/--unit: set the output time unit (nsec, usec, msec, or sec)
  -h/--help: print this usage message and exit
  --: separate options from statement, use when statement starts with -
  statement: statement to be timed (default 'pass')

A multi-line statement may be given by specifying each line as a
separate argument; indented lines are possible by enclosing an
argument in quotes and using leading spaces.  Multiple -s options are
treated similarly.

If -n is not given, a suitable number of loops is calculated by trying
increasing numbers from the sequence 1, 2, 5, 10, 20, 50, ... until the
total time is at least 0.2 seconds.

Note: there is a certain baseline overhead associated with executing a
pass statement.  It differs between versions.  The code here doesn't try
to hide it, but you should be aware of it.  The baseline overhead can be
measured by invoking the program without arguments.

Classes:

    Timer

Functions:

    timeit(string, string) -> float
    repeat(string, string) -> list
    default_timer() -> float

""""""

import gc
import sys
import time
import itertools

__all__ = [""Timer"", ""timeit"", ""repeat"", ""default_timer""]

dummy_src_name = ""<timeit-src>""
default_number = 1000000
default_repeat = 5
default_timer = time.perf_counter

_globals = globals

# Don't change the indentation of the template; the reindent() calls
# in Timer.__init__() depend on setup being indented 4 spaces and stmt
# being indented 8 spaces.
template = """"""
def inner(_it, _timer{init}):
    {setup}
    _t0 = _timer()
    for _i in _it:
        {stmt}
        pass
    _t1 = _timer()
    return _t1 - _t0
""""""

def reindent(src, indent):
    """"""Helper to reindent a multi-line statement.""""""
    return src.replace(""
"", ""
"" + "" ""*indent)

class Timer:
    """"""Class for timing execution speed of small code snippets.

    The constructor takes a statement to be timed, an additional
    statement used for setup, and a timer function.  Both statements
    default to 'pass'; the timer function is platform-dependent (see
    module doc string).  If 'globals' is specified, the code will be
    executed within that namespace (as opposed to inside timeit's
    namespace).

    To measure the execution time of the first statement, use the
    timeit() method.  The repeat() method is a convenience to call
    timeit() multiple times and return a list of results.

    The statements may contain newlines, as long as they don't contain
    multi-line string literals.
    """"""

    def __init__(self, stmt=""pass"", setup=""pass"", timer=default_timer,
                 globals=None):
        """"""Constructor.  See class doc string.""""""
        self.timer = timer
        local_ns = {}
        global_ns = _globals() if globals is None else globals
        init = ''
        if isinstance(setup, str):
            # Check that the code can be compiled outside a function
            compile(setup, dummy_src_name, ""exec"")
            stmtprefix = setup + '
'
            setup = reindent(setup, 4)
        elif callable(setup):
            local_ns['_setup'] = setup
            init += ', _setup=_setup'
            stmtprefix = ''
            setup = '_setup()'
        else:
            raise ValueError(""setup is neither a string nor callable"")
        if isinstance(stmt, str):
            # Check that the code can be compiled outside a function
            compile(stmtprefix + stmt, dummy_src_name, ""exec"")
            stmt = reindent(stmt, 8)
        elif callable(stmt):
            local_ns['_stmt'] = stmt
            init += ', _stmt=_stmt'
            stmt = '_stmt()'
        else:
            raise ValueError(""stmt is neither a string nor callable"")
        src = template.format(stmt=stmt, setup=setup, init=init)
        self.src = src  # Save for traceback display
        code = compile(src, dummy_src_name, ""exec"")
        exec(code, global_ns, local_ns)
        self.inner = local_ns[""inner""]

    def print_exc(self, file=None):
        """"""Helper to print a traceback from the timed code.

        Typical use:

            t = Timer(...)       # outside the try/except
            try:
                t.timeit(...)    # or t.repeat(...)
            except:
                t.print_exc()

        The advantage over the standard traceback is that source lines
        in the compiled template will be displayed.

        The optional file argument directs where the traceback is
        sent; it defaults to sys.stderr.
        """"""
        import linecache, traceback
        if self.src is not None:
            linecache.cache[dummy_src_name] = (len(self.src),
                                               None,
                                               self.src.split(""
""),
                                               dummy_src_name)
        # else the source is already stored somewhere else

        traceback.print_exc(file=file)

    def timeit(self, number=default_number):
        """"""Time 'number' executions of the main statement.

        To be precise, this executes the setup statement once, and
        then returns the time it takes to execute the main statement
        a number of times, as a float measured in seconds.  The
        argument is the number of times through the loop, defaulting
        to one million.  The main statement, the setup statement and
        the timer function to be used are passed to the constructor.
        """"""
        it = itertools.repeat(None, number)
        gcold = gc.isenabled()
        gc.disable()
        try:
            timing = self.inner(it, self.timer)
        finally:
            if gcold:
                gc.enable()
        return timing

    def repeat(self, repeat=default_repeat, number=default_number):
        """"""Call timeit() a few times.

        This is a convenience function that calls the timeit()
        repeatedly, returning a list of results.  The first argument
        specifies how many times to call timeit(), defaulting to 5;
        the second argument specifies the timer argument, defaulting
        to one million.

        Note: it's tempting to calculate mean and standard deviation
        from the result vector and report these.  However, this is not
        very useful.  In a typical case, the lowest value gives a
        lower bound for how fast your machine can run the given code
        snippet; higher values in the result vector are typically not
        caused by variability in Python's speed, but by other
        processes interfering with your timing accuracy.  So the min()
        of the result is probably the only number you should be
        interested in.  After that, you should look at the entire
        vector and apply common sense rather than statistics.
        """"""
        r = []
        for i in range(repeat):
            t = self.timeit(number)
            r.append(t)
        return r

    def autorange(self, callback=None):
        """"""Return the number of loops and time taken so that total time >= 0.2.

        Calls the timeit method with increasing numbers from the sequence
        1, 2, 5, 10, 20, 50, ... until the time taken is at least 0.2
        second.  Returns (number, time_taken).

        If *callback* is given and is not None, it will be called after
        each trial with two arguments: ``callback(number, time_taken)``.
        """"""
        i = 1
        while True:
            for j in 1, 2, 5:
                number = i * j
                time_taken = self.timeit(number)
                if callback:
                    callback(number, time_taken)
                if time_taken >= 0.2:
                    return (number, time_taken)
            i *= 10

def timeit(stmt=""pass"", setup=""pass"", timer=default_timer,
           number=default_number, globals=None):
    """"""Convenience function to create Timer object and call timeit method.""""""
    return Timer(stmt, setup, timer, globals).timeit(number)

def repeat(stmt=""pass"", setup=""pass"", timer=default_timer,
           repeat=default_repeat, number=default_number, globals=None):
    """"""Convenience function to create Timer object and call repeat method.""""""
    return Timer(stmt, setup, timer, globals).repeat(repeat, number)

def main(args=None, *, _wrap_timer=None):
    """"""Main program, used when run as a script.

    The optional 'args' argument specifies the command line to be parsed,
    defaulting to sys.argv[1:].

    The return value is an exit code to be passed to sys.exit(); it
    may be None to indicate success.

    When an exception happens during timing, a traceback is printed to
    stderr and the return value is 1.  Exceptions at other times
    (including the template compilation) are not caught.

    '_wrap_timer' is an internal interface used for unit testing.  If it
    is not None, it must be a callable that accepts a timer function
    and returns another timer function (used for unit testing).
    """"""
    if args is None:
        args = sys.argv[1:]
    import getopt
    try:
        opts, args = getopt.getopt(args, ""n:u:s:r:tcpvh"",
                                   [""number="", ""setup="", ""repeat="",
                                    ""time"", ""clock"", ""process"",
                                    ""verbose"", ""unit="", ""help""])
    except getopt.error as err:
        print(err)
        print(""use -h/--help for command line help"")
        return 2

    timer = default_timer
    stmt = ""
"".join(args) or ""pass""
    number = 0 # auto-determine
    setup = []
    repeat = default_repeat
    verbose = 0
    time_unit = None
    units = {""nsec"": 1e-9, ""usec"": 1e-6, ""msec"": 1e-3, ""sec"": 1.0}
    precision = 3
    for o, a in opts:
        if o in (""-n"", ""--number""):
            number = int(a)
        if o in (""-s"", ""--setup""):
            setup.append(a)
        if o in (""-u"", ""--unit""):
            if a in units:
                time_unit = a
            else:
                print(""Unrecognized unit. Please select nsec, usec, msec, or sec."",
                    file=sys.stderr)
                return 2
        if o in (""-r"", ""--repeat""):
            repeat = int(a)
            if repeat <= 0:
                repeat = 1
        if o in (""-p"", ""--process""):
            timer = time.process_time
        if o in (""-v"", ""--verbose""):
            if verbose:
                precision += 1
            verbose += 1
        if o in (""-h"", ""--help""):
            print(__doc__, end=' ')
            return 0
    setup = ""
"".join(setup) or ""pass""

    # Include the current directory, so that local imports work (sys.path
    # contains the directory of this script, rather than the current
    # directory)
    import os
    sys.path.insert(0, os.curdir)
    if _wrap_timer is not None:
        timer = _wrap_timer(timer)

    t = Timer(stmt, setup, timer)
    if number == 0:
        # determine number so that 0.2 <= total time < 2.0
        callback = None
        if verbose:
            def callback(number, time_taken):
                msg = ""{num} loop{s} -> {secs:.{prec}g} secs""
                plural = (number != 1)
                print(msg.format(num=number, s='s' if plural else '',
                                  secs=time_taken, prec=precision))
        try:
            number, _ = t.autorange(callback)
        except:
            t.print_exc()
            return 1

        if verbose:
            print()

    try:
        raw_timings = t.repeat(repeat, number)
    except:
        t.print_exc()
        return 1

    def format_time(dt):
        unit = time_unit

        if unit is not None:
            scale = units[unit]
        else:
            scales = [(scale, unit) for unit, scale in units.items()]
            scales.sort(reverse=True)
            for scale, unit in scales:
                if dt >= scale:
                    break

        return ""%.*g %s"" % (precision, dt / scale, unit)

    if verbose:
        print(""raw times: %s"" % "", "".join(map(format_time, raw_timings)))
        print()
    timings = [dt / number for dt in raw_timings]

    best = min(timings)
    print(""%d loop%s, best of %d: %s per loop""
          % (number, 's' if number != 1 else '',
             repeat, format_time(best)))

    best = min(timings)
    worst = max(timings)
    if worst >= best * 4:
        import warnings
        warnings.warn_explicit(""The test results are likely unreliable. ""
                               ""The worst time (%s) was more than four times ""
                               ""slower than the best time (%s).""
                               % (format_time(worst), format_time(best)),
                               UserWarning, '', 0)
    return None

if __name__ == ""__main__"":
    sys.exit(main())
",python,tab
636,1516708,"/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py",8495,0,"",python,selection_command
637,1522666,"/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py",8726,0,"",python,selection_mouse
638,1524017,"/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py",5721,0,"",python,selection_command
639,1529478,"TERMINAL",0,0,"2025-08-01 10:00:38.217843: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
640,1532355,"/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py",8726,0,"",python,selection_command
641,1532565,"sample.py",0,0,"",python,tab
642,1532566,"sample.py",6676,0,"",python,selection_command
643,1533751,"TERMINAL",0,0,"2025-08-01 10:00:42.471352: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
644,1542853,"TERMINAL",0,0,"2025-08-01 10:00:51.598886: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
645,1546802,"TERMINAL",0,0,"2025-08-01 10:00:55.548067: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
646,1549289,"TERMINAL",0,0,"Autoregressive generation time: 0.1705s
",,terminal_output
647,1551270,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
648,1552768,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
649,1566795,"sample.py",6798,0,"",python,selection_mouse
650,1568121,"sample.py",6797,0,"",python,selection_command
651,1568777,"sample.py",6797,1,"2",python,content
652,1571863,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
653,1572261,"TERMINAL",0,0,"

",,terminal_output
654,1583497,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
655,1597121,"TERMINAL",0,0,"2025-08-01 10:01:45.866574: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
656,1601346,"TERMINAL",0,0,"2025-08-01 10:01:50.097160: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
657,1610282,"TERMINAL",0,0,"2025-08-01 10:01:59.024760: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
658,1614159,"TERMINAL",0,0,"2025-08-01 10:02:02.901544: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
659,1616672,"TERMINAL",0,0,"Autoregressive generation time: 0.3248s
",,terminal_output
660,1618636,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
661,1620184,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
662,1630538,"sample.py",6797,1,"1",python,content
663,1633142,"experiments/sample.sh",0,0,"",shellscript,tab
664,1634624,"experiments/sample.sh",248,1,"3",shellscript,content
665,1635656,"experiments/sample.sh",248,1,"4",shellscript,content
666,1637685,"sample.py",0,0,"",python,tab
667,1639179,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
668,1639485,"TERMINAL",0,0,"

",,terminal_output
669,1640647,"TERMINAL",0,0,"a",,terminal_output
670,1648572,"sample.py",6697,0,"",python,selection_command
671,1648613,"sample.py",6668,0,"",python,selection_command
672,1648774,"sample.py",6603,0,"",python,selection_command
673,1648890,"sample.py",6516,0,"",python,selection_command
674,1649510,"sample.py",6501,86,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
675,1649980,"sample.py",6501,151,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.",python,selection_command
676,1650048,"sample.py",6501,180,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(",python,selection_command
677,1650194,"sample.py",6501,280,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
678,1650330,"sample.py",6501,298,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
679,1650506,"sample.py",6501,304,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
680,1650727,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
681,1651131,"sample.py",6501,365,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
682,1651572,"sample.py",6821,0,"",python,selection_command
683,1659121,"TERMINAL",0,0,"2025-08-01 10:02:47.864570: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
684,1663561,"TERMINAL",0,0,"2025-08-01 10:02:52.309889: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
685,1672691,"TERMINAL",0,0,"2025-08-01 10:03:01.437954: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
686,1686162,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 210, in <module>
    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 396, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
687,1687403,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ a",,terminal_output
688,1705129,"sample.py",6882,0,"",python,selection_command
689,1705378,"sample.py",6821,0,"",python,selection_command
690,1712798,"experiments/sample.sh",0,0,"",shellscript,tab
691,1714008,"experiments/sample.sh",248,1,"3",shellscript,content
692,1715522,"sample.py",0,0,"",python,tab
693,1719207,"sample.py",6804,0,"",python,selection_command
694,1719341,"sample.py",6797,0,"",python,selection_command
695,1719405,"sample.py",6697,0,"",python,selection_command
696,1719554,"sample.py",6668,0,"",python,selection_command
697,1719776,"sample.py",6603,0,"",python,selection_command
698,1720579,"sample.py",6516,0,"",python,selection_command
699,1721097,"sample.py",6603,0,"",python,selection_command
700,1721534,"sample.py",6516,0,"",python,selection_command
701,1721980,"sample.py",6501,86,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
702,1722485,"sample.py",6501,151,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.",python,selection_command
703,1722505,"sample.py",6501,180,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(",python,selection_command
704,1722664,"sample.py",6501,280,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
705,1722808,"sample.py",6501,298,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
706,1722958,"sample.py",6501,304,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
707,1723684,"sample.py",6501,365,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
708,1724105,"sample.py",6810,0,"# ",python,content
709,1724105,"sample.py",6804,0,"# ",python,content
710,1724106,"sample.py",6786,0,"# ",python,content
711,1724106,"sample.py",6686,0,"# ",python,content
712,1724106,"sample.py",6657,0,"# ",python,content
713,1724106,"sample.py",6592,0,"# ",python,content
714,1724106,"sample.py",6505,0,"# ",python,content
715,1724630,"sample.py",6833,0,"",python,selection_command
716,1737061,"genie.py",0,0,"",python,tab
717,1738077,"genie.py",14460,0,"",python,selection_command
718,1738238,"genie.py",14387,0,"",python,selection_command
719,1738415,"genie.py",14289,0,"",python,selection_command
720,1738592,"genie.py",14232,0,"",python,selection_command
721,1784940,"genie.py",14206,56,"        for block in dynamics_causal.transformer.blocks:",python,selection_command
722,1785275,"genie.py",14206,154,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
723,1785447,"genie.py",14206,253,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
724,1786884,"genie.py",14206,254,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)
",python,selection_command
725,1788749,"genie.py",14206,255,"",python,content
726,1788773,"genie.py",14214,0,"",python,selection_command
727,1792252,"sample.py",0,0,"",python,tab
728,1793206,"sample.py",6816,0,"",python,selection_command
729,1793453,"sample.py",6805,0,"",python,selection_command
730,1793481,"sample.py",6703,0,"",python,selection_command
731,1793517,"sample.py",6672,0,"",python,selection_command
732,1793548,"sample.py",6605,0,"",python,selection_command
733,1793580,"sample.py",6516,0,"",python,selection_command
734,1793613,"sample.py",6478,0,"",python,selection_command
735,1793659,"sample.py",6462,0,"",python,selection_command
736,1793681,"sample.py",6417,0,"",python,selection_command
737,1793715,"sample.py",6374,0,"",python,selection_command
738,1793745,"sample.py",6321,0,"",python,selection_command
739,1793781,"sample.py",6269,0,"",python,selection_command
740,1793881,"sample.py",6321,0,"",python,selection_command
741,1794140,"sample.py",6374,0,"",python,selection_command
742,1794169,"sample.py",6417,0,"",python,selection_command
743,1794199,"sample.py",6462,0,"",python,selection_command
744,1794232,"sample.py",6478,0,"",python,selection_command
745,1794266,"sample.py",6516,0,"",python,selection_command
746,1794297,"sample.py",6605,0,"",python,selection_command
747,1794333,"sample.py",6672,0,"",python,selection_command
748,1794364,"sample.py",6703,0,"",python,selection_command
749,1794455,"sample.py",6672,0,"",python,selection_command
750,1794713,"sample.py",6605,0,"",python,selection_command
751,1794843,"sample.py",6516,0,"",python,selection_command
752,1795006,"sample.py",6478,0,"",python,selection_command
753,1795178,"sample.py",6462,0,"",python,selection_command
754,1796310,"sample.py",6417,0,"",python,selection_command
755,1796472,"sample.py",6374,0,"",python,selection_command
756,1796729,"sample.py",6321,0,"",python,selection_command
757,1796762,"sample.py",6269,0,"",python,selection_command
758,1796790,"sample.py",6196,0,"",python,selection_command
759,1796821,"sample.py",6155,0,"",python,selection_command
760,1796853,"sample.py",6121,0,"",python,selection_command
761,1796892,"sample.py",6104,0,"",python,selection_command
762,1796921,"sample.py",6091,0,"",python,selection_command
763,1796954,"sample.py",6059,0,"",python,selection_command
764,1796988,"sample.py",6036,0,"",python,selection_command
765,1797021,"sample.py",5920,0,"",python,selection_command
766,1797092,"sample.py",6036,0,"",python,selection_command
767,1797362,"sample.py",6059,0,"",python,selection_command
768,1797381,"sample.py",6091,0,"",python,selection_command
769,1797411,"sample.py",6104,0,"",python,selection_command
770,1797444,"sample.py",6121,0,"",python,selection_command
771,1797478,"sample.py",6155,0,"",python,selection_command
772,1797514,"sample.py",6196,0,"",python,selection_command
773,1797550,"sample.py",6269,0,"",python,selection_command
774,1797584,"sample.py",6321,0,"",python,selection_command
775,1797618,"sample.py",6374,0,"",python,selection_command
776,1797654,"sample.py",6417,0,"",python,selection_command
777,1797685,"sample.py",6462,0,"",python,selection_command
778,1797719,"sample.py",6478,0,"",python,selection_command
779,1797751,"sample.py",6516,0,"",python,selection_command
780,1797780,"sample.py",6605,0,"",python,selection_command
781,1797813,"sample.py",6672,0,"",python,selection_command
782,1797846,"sample.py",6703,0,"",python,selection_command
783,1797878,"sample.py",6805,0,"",python,selection_command
784,1797914,"sample.py",6816,0,"",python,selection_command
785,1797951,"sample.py",6833,0,"",python,selection_command
786,1797982,"sample.py",6896,0,"",python,selection_command
787,1798014,"sample.py",6983,0,"",python,selection_command
788,1798053,"sample.py",7045,0,"",python,selection_command
789,1798086,"sample.py",7140,0,"",python,selection_command
790,1798119,"sample.py",7045,0,"",python,selection_command
791,1798376,"sample.py",6983,0,"",python,selection_command
792,1798411,"sample.py",6896,0,"",python,selection_command
793,1798434,"sample.py",6833,0,"",python,selection_command
794,1798464,"sample.py",6816,0,"",python,selection_command
795,1799240,"sample.py",6833,0,"",python,selection_command
796,1799547,"sample.py",6816,0,"",python,selection_command
797,1799702,"sample.py",6805,0,"",python,selection_command
798,1799854,"sample.py",6703,0,"",python,selection_command
799,1799986,"sample.py",6672,0,"",python,selection_command
800,1800133,"sample.py",6605,0,"",python,selection_command
801,1800283,"sample.py",6516,0,"",python,selection_command
802,1800417,"sample.py",6478,0,"",python,selection_command
803,1802217,"sample.py",6500,0,"
    ",python,content
804,1802411,"sample.py",6501,4,"",python,content
805,1802461,"sample.py",6501,0,"
        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)
",python,content
806,1802466,"sample.py",6510,0,"",python,selection_command
807,1803334,"sample.py",6510,1,"f",python,selection_command
808,1803551,"sample.py",6510,58,"for block in dynamics_causal.transformer.blocks:
         ",python,selection_command
809,1803732,"sample.py",6510,156,"for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
         ",python,selection_command
810,1803965,"sample.py",6510,247,"for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

",python,selection_command
811,1804282,"sample.py",6510,156,"for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
         ",python,selection_command
812,1804437,"sample.py",6665,0,"",python,selection_command
813,1805799,"sample.py",6665,1," ",python,selection_command
814,1806051,"sample.py",6665,0,"",python,selection_command
815,1806822,"sample.py",6657,98,"            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
816,1807228,"sample.py",6559,196,"            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
817,1807264,"sample.py",6502,253,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
818,1808255,"sample.py",6657,12,"        ",python,content
819,1808255,"sample.py",6559,12,"        ",python,content
820,1808255,"sample.py",6502,8,"    ",python,content
821,1808268,"sample.py",6506,0,"",python,selection_command
822,1809431,"sample.py",6501,0,"",python,selection_command
823,1809801,"sample.py",6506,0,"",python,selection_command
824,1810158,"sample.py",6501,0,"",python,selection_command
825,1810254,"sample.py",6506,0,"",python,selection_command
826,1810602,"sample.py",6559,0,"",python,selection_command
827,1810735,"sample.py",6563,0,"",python,selection_command
828,1810984,"sample.py",6510,0,"",python,selection_command
829,1811070,"sample.py",6516,0,"",python,selection_command
830,1811165,"sample.py",6501,0,"",python,selection_command
831,1811474,"sample.py",6506,0,"",python,selection_command
832,1811777,"sample.py",6559,0,"",python,selection_command
833,1811851,"sample.py",6563,0,"",python,selection_command
834,1812155,"sample.py",6510,0,"",python,selection_command
835,1812238,"sample.py",6516,0,"",python,selection_command
836,1812427,"sample.py",6519,0,"",python,selection_command
837,1813559,"sample.py",6733,3,"",python,content
838,1813560,"sample.py",6732,0,"arg",python,content
839,1813560,"sample.py",6711,3,"",python,content
840,1813560,"sample.py",6710,0,"arg",python,content
841,1813560,"sample.py",6638,3,"",python,content
842,1813560,"sample.py",6637,0,"arg",python,content
843,1813560,"sample.py",6616,3,"",python,content
844,1813560,"sample.py",6615,0,"arg",python,content
845,1813560,"sample.py",6603,0,"args.",python,content
846,1813560,"sample.py",6527,7,"",python,content
847,1813560,"sample.py",6519,0,"genie.",python,content
848,1816240,"sample.py",6577,0,"",python,selection_command
849,1816494,"sample.py",6525,0,"",python,selection_command
850,1816659,"sample.py",6501,0,"",python,selection_command
851,1818871,"sample.py",6501,0,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)",python,content
852,1820098,"sample.py",6615,0,"",python,selection_command
853,1820599,"sample.py",0,0,"",python,selection_command
854,1822130,"sample.py",34,0,"",python,selection_command
855,1822308,"sample.py",46,0,"",python,selection_command
856,1822460,"sample.py",56,0,"",python,selection_command
857,1822630,"sample.py",69,0,"",python,selection_command
858,1823409,"sample.py",82,0,"
import math",python,content
859,1825369,"sample.py",83,0,"",python,selection_command
860,1826351,"sample.py",90,0,"",python,selection_command
861,1826812,"sample.py",6565,0,"",python,selection_command
862,1830032,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
863,1830642,"TERMINAL",0,0,"

",,terminal_output
864,1841971,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
865,1842945,"sample.py",6729,0,"",python,selection_mouse
866,1843112,"sample.py",6729,1,"a",python,selection_command
867,1843191,"sample.py",6729,4,"args",python,selection_command
868,1843369,"sample.py",6729,5,"args.",python,selection_command
869,1843533,"sample.py",6729,12,"args.seq_len",python,selection_command
870,1843900,"sample.py",6729,12,"",python,content
871,1845399,"sample.py",6729,0,"5",python,content
872,1845399,"sample.py",6730,0,"",python,selection_keyboard
873,1846213,"sample.py",6729,0,"",python,selection_command
874,1849570,"sample.py",6729,1,"r",python,content
875,1850847,"sample.py",6729,1,"D",python,content
876,1851023,"TERMINAL",0,0,"2025-08-01 10:05:59.770129: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
877,1851873,"sample.py",6729,1,"S",python,content
878,1855553,"TERMINAL",0,0,"2025-08-01 10:06:04.298129: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
879,1864807,"TERMINAL",0,0,"2025-08-01 10:06:13.542092: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
880,1878768,"TERMINAL",0,0,"2025-08-01 10:06:27.514616: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
881,1890916,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
882,1892618,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
883,1896517,"sample.py",0,0,"",python,tab
884,1896519,"sample.py",69,0,"",python,selection_command
885,1900848,"sample.py",6878,0,"",python,selection_mouse
886,1902220,"sample.py",6865,88,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
887,1902860,"sample.py",6865,155,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.",python,selection_command
888,1902911,"sample.py",6865,186,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(",python,selection_command
889,1903058,"sample.py",6865,288,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
890,1903189,"sample.py",6865,308,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,",python,selection_command
891,1903338,"sample.py",6865,316,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )",python,selection_command
892,1903510,"sample.py",6865,379,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
893,1903988,"sample.py",6865,380,"",python,content
894,1903997,"sample.py",6869,0,"",python,selection_command
895,1907798,"sample.py",6880,0,"",python,selection_mouse
896,1911266,"sample.py",6869,0,"# _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")
    ",python,content
897,1911282,"sample.py",6878,0,"",python,selection_command
898,1920445,"sample.py",6864,0,"",python,selection_mouse
899,1922633,"genie.py",0,0,"",python,tab
900,1932605,"genie.py",14205,0,"",python,selection_mouse
901,1946407,"sample.py",0,0,"",python,tab
902,1946409,"sample.py",69,0,"",python,selection_command
903,1951665,"sample.py",6870,0,"",python,selection_mouse
904,1953935,"sample.py",6870,1," ",python,selection_command
905,1954328,"sample.py",6869,2,"# ",python,selection_command
906,1954712,"sample.py",6869,2,"# ",python,selection_command
907,1954751,"sample.py",6869,2,"# ",python,selection_command
908,1954902,"sample.py",6869,2,"# ",python,selection_command
909,1955027,"sample.py",6869,2,"# ",python,selection_command
910,1955158,"sample.py",6869,2,"# ",python,selection_command
911,1955302,"sample.py",6869,2,"# ",python,selection_command
912,1955616,"sample.py",7186,2,"",python,content
913,1955616,"sample.py",7178,2,"",python,content
914,1955616,"sample.py",7158,2,"",python,content
915,1955616,"sample.py",7056,2,"",python,content
916,1955616,"sample.py",7025,2,"",python,content
917,1955616,"sample.py",6958,2,"",python,content
918,1955616,"sample.py",6869,2,"",python,content
919,1955632,"sample.py",6869,0,"",python,selection_command
920,1967512,"TERMINAL",0,0,"gi",,terminal_output
921,1967868,"TERMINAL",0,0,"t",,terminal_output
922,1968135,"TERMINAL",0,0," ",,terminal_output
923,1968295,"TERMINAL",0,0,"di",,terminal_output
924,1968487,"TERMINAL",0,0,"f",,terminal_output
925,1968547,"TERMINAL",0,0,"f",,terminal_output
926,1968716,"TERMINAL",0,0," ",,terminal_output
927,1968855,"TERMINAL",0,0,"eg",,terminal_output
928,1968946,"TERMINAL",0,0,"f",,terminal_output
929,1969563,"TERMINAL",0,0,"",,terminal_output
930,1969728,"TERMINAL",0,0,"",,terminal_output
931,1969845,"TERMINAL",0,0,"",,terminal_output
932,1970057,"TERMINAL",0,0,"ge",,terminal_output
933,1970178,"TERMINAL",0,0,"n",,terminal_output
934,1970528,"TERMINAL",0,0,"",,terminal_output
935,1971026,"TERMINAL",0,0,"i",,terminal_output
936,1971142,"TERMINAL",0,0,"e",,terminal_output
937,1971666,"TERMINAL",0,0,"l",,terminal_output
938,1971935,"TERMINAL",0,0,".",,terminal_output
939,1972268,"TERMINAL",0,0,"",,terminal_output
940,1972420,"TERMINAL",0,0,"",,terminal_output
941,1972582,"TERMINAL",0,0,"",,terminal_output
942,1972986,"TERMINAL",0,0,"e",,terminal_output
943,1973066,"TERMINAL",0,0,".",,terminal_output
944,1973266,"TERMINAL",0,0,"p",,terminal_output
945,1973355,"TERMINAL",0,0,"y",,terminal_output
946,1973779,"TERMINAL",0,0,"

=

>[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
947,1975534,"TERMINAL",0,0,"g",,terminal_output
948,1975714,"TERMINAL",0,0,"it",,terminal_output
949,1975965,"TERMINAL",0,0," ",,terminal_output
950,1976064,"TERMINAL",0,0,"di",,terminal_output
951,1976202,"TERMINAL",0,0,"f",,terminal_output
952,1976283,"TERMINAL",0,0,"f",,terminal_output
953,1976510,"TERMINAL",0,0,"

=
diff --git a/.gitignore b/.gitignore
index 0933318..42203f0 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,8 +2,10 @@
 *.npy
 *.png
 *.gif
+*.array_record
 
 wandb_key
 checkpoints/
 wandb/
 __pycache__/
+experiments/
diff --git a/sample.py b/sample.py
index 09b3017..f567689 100644
--- a/sample.py
+++ b/sample.py
@@ -2,6 +2,7 @@ from dataclasses import dataclass
 import time
 import os
 import optax
+import timeit
 import math
 
 import dm_pix as pix
@@ -213,6 +214,13 @@ if __name__ == ""__main__"":
         block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
         block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dty pe)
 
+    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
+    # Measure autoregressive generation time once and report it.
+    elapsed = timeit.timeit(
+        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ ready(),
+        number=1,
+    )
+    print(f""Autoregressive generation time: {elapsed:.4f}s"")
     recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
     recon_video_BSHWC = recon_video_BSHWC.astype(jnp.float32)
     gt = gt_video[:, : recon_video_BSHWC.shape[1]].clip(0, 1).reshape(-1, *gt_video.shape[2 :])

>[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
954,2003069,"genie.py",0,0,"",python,tab
955,2009729,"genie.py",0,0,"",python,tab
956,2009729,"genie.py",14206,0,"",python,selection_command
957,2013512,"genie.py",0,0,"",python,tab
958,2013513,"genie.py",14206,0,"",python,selection_command
959,2015198,"sample.py",0,0,"",python,tab
960,2015198,"sample.py",6499,0,"",python,selection_command
961,2019022,"TERMINAL",0,0,"git diff",,terminal_output
962,2019404,"TERMINAL",0,0," genie.py",,terminal_output
963,2019805,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
964,2020657,"TERMINAL",0,0,"

",,terminal_output
965,2024008,"experiments/sample.sh",0,0,"",shellscript,tab
966,2032034,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
967,2045520,"TERMINAL",0,0,"2025-08-01 10:09:14.265914: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
968,2050033,"TERMINAL",0,0,"2025-08-01 10:09:18.780260: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
969,2059053,"TERMINAL",0,0,"2025-08-01 10:09:27.801359: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
970,2073059,"TERMINAL",0,0,"2025-08-01 10:09:41.803488: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
971,2083139,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 219, in <module>
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 220, in <lambda>
    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 181, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 165, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 352, in sample_causal
    carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
972,2084410,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
973,2113154,"genie.py",0,0,"",python,tab
974,2116050,"sample.py",0,0,"",python,tab
975,2118864,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
976,2119225,"TERMINAL",0,0,"

",,terminal_output
977,2129599,"experiments/sample.sh",0,0,"",shellscript,tab
978,2130742,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
979,2139469,"TERMINAL",0,0,"2025-08-01 10:10:48.213828: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
980,2144055,"TERMINAL",0,0,"2025-08-01 10:10:52.695694: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
981,2153084,"TERMINAL",0,0,"2025-08-01 10:11:01.828888: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
982,2167062,"TERMINAL",0,0,"2025-08-01 10:11:15.768143: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
983,2173009,"sample.py",0,0,"",python,tab
984,2175609,"sample.py",7028,0,"",python,selection_mouse
985,2176642,"sample.py",7017,28,"    elapsed = timeit.timeit(",python,selection_command
986,2177099,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 219, in <module>
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 220, in <lambda>
    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 181, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 165, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 392, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
987,2178399,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
988,2178675,"sample.py",7017,128,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
989,2178849,"sample.py",7017,146,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
990,2178993,"sample.py",7017,152,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
991,2180850,"sample.py",7168,0,"",python,selection_command
992,2181579,"sample.py",7164,5,"    )",python,selection_command
993,2181886,"sample.py",7146,23,"        number=1,
    )",python,selection_command
994,2181920,"sample.py",7046,123,"        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
995,2182087,"sample.py",7017,152,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
996,2196718,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
997,2197257,"sample.py",7021,0,"",python,selection_command
998,2198254,"sample.py",7017,28,"    elapsed = timeit.timeit(",python,selection_command
999,2198710,"sample.py",7017,128,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1000,2198855,"sample.py",7017,146,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
1001,2199010,"sample.py",7017,152,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1002,2201309,"sample.py",7168,0,"",python,selection_command
1003,2201727,"sample.py",7164,5,"    )",python,selection_command
1004,2201991,"sample.py",7146,23,"        number=1,
    )",python,selection_command
1005,2202068,"sample.py",7046,123,"        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1006,2202186,"sample.py",7017,152,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1007,2202337,"sample.py",6952,217,"    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1008,2202780,"sample.py",6865,304,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1009,2203192,"sample.py",7168,0,"# ",python,content
1010,2203192,"sample.py",7150,0,"# ",python,content
1011,2203192,"sample.py",7050,0,"# ",python,content
1012,2203192,"sample.py",7021,0,"# ",python,content
1013,2203192,"sample.py",6956,0,"# ",python,content
1014,2203192,"sample.py",6869,0,"# ",python,content
1015,2203727,"sample.py",6869,0,"",python,selection_command
1016,2206471,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
1017,2206698,"TERMINAL",0,0,"

",,terminal_output
1018,2218072,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
1019,2231780,"TERMINAL",0,0,"2025-08-01 10:12:20.522526: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1020,2235521,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 223, in <module>
    print(f""Autoregressive generation time: {elapsed:.4f}s"")
NameError: name 'elapsed' is not defined
",,terminal_output
1021,2236428,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
1022,2244069,"sample.py",7196,0,"",python,selection_mouse
1023,2245263,"sample.py",7186,0,"# ",python,content
1024,2249111,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
1025,2249356,"TERMINAL",0,0,"

",,terminal_output
1026,2260367,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
1027,2269315,"TERMINAL",0,0,"2025-08-01 10:12:58.059302: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1028,2273831,"TERMINAL",0,0,"2025-08-01 10:13:02.572562: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1029,2283116,"TERMINAL",0,0,"2025-08-01 10:13:11.763132: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1030,2297166,"TERMINAL",0,0,"2025-08-01 10:13:25.911159: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1031,2309110,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
1032,2310379,"sample.py",7182,62,"    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1033,2310828,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
1034,2310891,"sample.py",7174,70,"    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1035,2310921,"sample.py",7154,90,"    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1036,2311085,"sample.py",7052,192,"    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1037,2311234,"sample.py",7021,223,"    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1038,2311419,"sample.py",6954,290,"    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1039,2311601,"sample.py",6865,379,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1040,2337638,"sample.py",6881,0,"",python,selection_command
1041,2338542,"sample.py",6865,88,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
1042,2338935,"sample.py",6865,155,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.",python,selection_command
1043,2339187,"sample.py",6865,186,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(",python,selection_command
1044,2339260,"sample.py",6865,288,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1045,2339450,"sample.py",6865,308,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,",python,selection_command
1046,2339580,"sample.py",6865,316,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )",python,selection_command
1047,2339730,"sample.py",6865,379,"    # _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # # Measure autoregressive generation time once and report it.
    # elapsed = timeit.timeit(
    #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
    #     number=1,
    # )
    # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1048,2342993,"sample.py",7186,2,"",python,content
1049,2342994,"sample.py",7178,2,"",python,content
1050,2342994,"sample.py",7158,2,"",python,content
1051,2342994,"sample.py",7056,2,"",python,content
1052,2342994,"sample.py",7025,2,"",python,content
1053,2342994,"sample.py",6958,2,"",python,content
1054,2342994,"sample.py",6869,2,"",python,content
1055,2343526,"sample.py",7186,0,"",python,selection_command
1056,2344095,"sample.py",7247,0,"",python,selection_command
1057,2345481,"sample.py",7235,0,"# ",python,content
1058,2350119,"sample.py",7338,0,"",python,selection_command
1059,2351247,"sample.py",7249,0,"",python,selection_command
1060,2354453,"sample.py",7231,0,"",python,selection_command
1061,2354683,"sample.py",7235,0,"",python,selection_command
1062,2355153,"sample.py",7235,1,"",python,content
1063,2355413,"sample.py",7235,1,"",python,content
1064,2356219,"sample.py",7230,0,"
    ",python,content
1065,2357660,"sample.py",7235,0,"exit()",python,content
1066,2357854,"sample.py",7240,0,"",python,selection_command
1067,2359663,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
1068,2359900,"TERMINAL",0,0,"

",,terminal_output
1069,2371230,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
1070,2380235,"TERMINAL",0,0,"2025-08-01 10:14:48.978485: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1071,2384744,"TERMINAL",0,0,"2025-08-01 10:14:53.487828: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1072,2393855,"TERMINAL",0,0,"2025-08-01 10:15:02.604315: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1073,2407904,"TERMINAL",0,0,"2025-08-01 10:15:16.644880: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1074,2418026,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 219, in <module>
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 220, in <lambda>
    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 181, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 165, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 392, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
1075,2419249,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
1076,2432553,"sample.py",7169,0,"",python,selection_mouse
1077,2432565,"sample.py",7168,0,"",python,selection_command
1078,2435144,"sample.py",6876,0,"",python,selection_mouse
1079,2436422,"sample.py",6869,0,"# ",python,content
1080,2439283,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
1081,2439603,"TERMINAL",0,0,"

",,terminal_output
1082,2450954,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
1083,2459889,"TERMINAL",0,0,"2025-08-01 10:16:08.638307: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1084,2464428,"TERMINAL",0,0,"2025-08-01 10:16:13.176943: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1085,2473576,"TERMINAL",0,0,"2025-08-01 10:16:22.319878: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1086,2487414,"TERMINAL",0,0,"2025-08-01 10:16:36.162972: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
1087,2497456,"TERMINAL",0,0,"Autoregressive generation time: 33.7589s
",,terminal_output
1088,2498750,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
1089,2559811,"sample.py",0,0,"",python,tab
1090,2575599,"genie.py",0,0,"",python,tab
1091,2575599,"genie.py",14206,0,"",python,selection_command
1092,2613106,"sample.py",0,0,"",python,tab
1093,2616235,"sample.py",6967,0,"",python,selection_command
1094,2616398,"sample.py",7032,0,"",python,selection_command
1095,2616444,"sample.py",7061,0,"",python,selection_command
1096,2617053,"sample.py",7032,0,"",python,selection_command
1097,2619823,"sample.py",7019,28,"    elapsed = timeit.timeit(",python,selection_command
1098,2620204,"sample.py",7019,128,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1099,2620271,"sample.py",7019,146,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
1100,2620518,"sample.py",7019,152,"    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1101,2620924,"sample.py",7170,0,"",python,selection_command
1102,2621150,"sample.py",7161,0,"",python,selection_command
1103,2621300,"sample.py",7061,0,"",python,selection_command
1104,2621434,"sample.py",7032,0,"",python,selection_command
1105,2621701,"sample.py",6967,0,"",python,selection_command
1106,2621871,"sample.py",6878,0,"",python,selection_command
1107,2622253,"sample.py",6871,0,"",python,selection_command
1108,2622421,"sample.py",6869,0,"",python,selection_command
1109,2622809,"sample.py",6869,1,"",python,content
1110,2623005,"sample.py",6869,1,"",python,content
1111,2625541,"sample.py",6869,0," ",python,content
1112,2625543,"sample.py",6869,0,"",python,selection_command
1113,2625855,"sample.py",6869,0,"#",python,content
1114,2625858,"sample.py",6869,0,"",python,selection_command
1115,2626529,"sample.py",6958,0,"",python,selection_command
1116,2626675,"sample.py",7023,0,"",python,selection_command
1117,2626741,"sample.py",7052,0,"",python,selection_command
1118,2626906,"sample.py",7152,0,"",python,selection_command
1119,2627024,"sample.py",7170,0,"",python,selection_command
1120,2627161,"sample.py",7176,0,"",python,selection_command
1121,2627451,"sample.py",7237,0,"",python,selection_command
1122,2628752,"sample.py",7233,11,"",python,content
1123,2628769,"sample.py",7237,0,"",python,selection_command
1124,2629046,"sample.py",7176,0,"",python,selection_command
1125,2630198,"sample.py",7172,60,"    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1126,2631082,"sample.py",7176,0,"",python,selection_command
1127,2631275,"sample.py",7170,0,"",python,selection_command
1128,2631445,"sample.py",7152,0,"",python,selection_command
1129,2631510,"sample.py",7052,0,"",python,selection_command
1130,2631657,"sample.py",7023,0,"",python,selection_command
1131,2631815,"sample.py",6958,0,"",python,selection_command
1132,2631946,"sample.py",6869,0,"",python,selection_command
1133,2632148,"sample.py",6869,1,"",python,content
1134,2632336,"sample.py",6869,1,"",python,content
1135,2632783,"sample.py",6865,86,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
1136,2633080,"sample.py",6865,151,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.",python,selection_command
1137,2633147,"sample.py",6865,180,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(",python,selection_command
1138,2633283,"sample.py",6865,280,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1139,2633395,"sample.py",6865,298,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
1140,2633544,"sample.py",6865,304,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1141,2633801,"sample.py",6865,365,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1142,2633918,"sample.py",6865,366,"",python,content
1143,2633920,"sample.py",6869,0,"",python,selection_command
1144,2637698,"genie.py",0,0,"",python,tab
1145,2639023,"genie.py",14206,0,"",python,selection_command
1146,2639266,"genie.py",14234,0,"",python,selection_command
1147,2639300,"genie.py",14322,0,"",python,selection_command
1148,2639322,"genie.py",14399,0,"",python,selection_command
1149,2639367,"genie.py",14465,0,"",python,selection_command
1150,2639399,"genie.py",14509,0,"",python,selection_command
1151,2639430,"genie.py",14552,0,"",python,selection_command
1152,2639465,"genie.py",14553,0,"",python,selection_command
1153,2639497,"genie.py",14600,0,"",python,selection_command
1154,2639530,"genie.py",14673,0,"",python,selection_command
1155,2639566,"genie.py",14674,0,"",python,selection_command
1156,2639597,"genie.py",14715,0,"",python,selection_command
1157,2639628,"genie.py",14794,0,"",python,selection_command
1158,2639663,"genie.py",14871,0,"",python,selection_command
1159,2639697,"genie.py",14950,0,"",python,selection_command
1160,2639724,"genie.py",15042,0,"",python,selection_command
1161,2639764,"genie.py",15131,0,"",python,selection_command
1162,2639800,"genie.py",15243,0,"",python,selection_command
1163,2639831,"genie.py",15315,0,"",python,selection_command
1164,2639858,"genie.py",15316,0,"",python,selection_command
1165,2639898,"genie.py",15372,0,"",python,selection_command
1166,2639932,"genie.py",15402,0,"",python,selection_command
1167,2639958,"genie.py",15478,0,"",python,selection_command
1168,2639999,"genie.py",15496,0,"",python,selection_command
1169,2640030,"genie.py",15546,0,"",python,selection_command
1170,2640064,"genie.py",15631,0,"",python,selection_command
1171,2640095,"genie.py",15669,0,"",python,selection_command
1172,2640135,"genie.py",15761,0,"",python,selection_command
1173,2640166,"genie.py",15785,0,"",python,selection_command
1174,2640198,"genie.py",15786,0,"",python,selection_command
1175,2640226,"genie.py",15858,0,"",python,selection_command
1176,2640265,"genie.py",15893,0,"",python,selection_command
1177,2640294,"genie.py",15894,0,"",python,selection_command
1178,2640331,"genie.py",15963,0,"",python,selection_command
1179,2640364,"genie.py",15990,0,"",python,selection_command
1180,2640391,"genie.py",16038,0,"",python,selection_command
1181,2640433,"genie.py",16047,0,"",python,selection_command
1182,2640466,"genie.py",16088,0,"",python,selection_command
1183,2640498,"genie.py",16138,0,"",python,selection_command
1184,2640715,"genie.py",16139,0,"",python,selection_command
1185,2641052,"genie.py",16138,0,"",python,selection_command
1186,2641301,"genie.py",16088,0,"",python,selection_command
1187,2641334,"genie.py",16047,0,"",python,selection_command
1188,2641583,"genie.py",16088,0,"",python,selection_command
1189,2641830,"genie.py",16138,0,"",python,selection_command
1190,2641864,"genie.py",16139,0,"",python,selection_command
1191,2641968,"genie.py",16203,0,"",python,selection_command
1192,2646226,"genie.py",16264,0,"",python,selection_command
1193,2646482,"genie.py",16366,0,"",python,selection_command
1194,2646505,"genie.py",16367,0,"",python,selection_command
1195,2646534,"genie.py",16420,0,"",python,selection_command
1196,2646570,"genie.py",16454,0,"",python,selection_command
1197,2646600,"genie.py",16480,0,"",python,selection_command
1198,2646751,"genie.py",16520,0,"",python,selection_command
1199,2646918,"genie.py",16554,0,"",python,selection_command
1200,2647072,"genie.py",16606,0,"",python,selection_command
1201,2652833,"genie.py",16619,0,"
            ",python,content
1202,2653045,"genie.py",16620,12,"",python,content
1203,2653080,"genie.py",16620,0,"
    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,content
1204,2653088,"genie.py",16625,0,"",python,selection_command
1205,2655053,"genie.py",16621,86,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
1206,2655590,"genie.py",16621,151,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.",python,selection_command
1207,2655650,"genie.py",16621,180,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(",python,selection_command
1208,2655805,"genie.py",16621,280,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1209,2655946,"genie.py",16621,298,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,",python,selection_command
1210,2656106,"genie.py",16621,304,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )",python,selection_command
1211,2656280,"genie.py",16621,365,"    _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
    # Measure autoregressive generation time once and report it.
    elapsed = timeit.timeit(
        lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
        number=1,
    )
    print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1212,2657506,"genie.py",16926,4,"        ",python,content
1213,2657506,"genie.py",16920,4,"        ",python,content
1214,2657506,"genie.py",16902,8,"            ",python,content
1215,2657506,"genie.py",16802,8,"            ",python,content
1216,2657506,"genie.py",16773,4,"        ",python,content
1217,2657506,"genie.py",16708,4,"        ",python,content
1218,2657506,"genie.py",16621,4,"        ",python,content
1219,2657518,"genie.py",16950,8,"            ",python,content
1220,2657518,"genie.py",16940,8,"            ",python,content
1221,2657518,"genie.py",16918,12,"                ",python,content
1222,2657518,"genie.py",16814,12,"                ",python,content
1223,2657518,"genie.py",16781,8,"            ",python,content
1224,2657518,"genie.py",16712,8,"            ",python,content
1225,2657518,"genie.py",16621,8,"            ",python,content
1226,2657520,"genie.py",16974,12,"                ",python,content
1227,2657520,"genie.py",16960,12,"                ",python,content
1228,2657520,"genie.py",16934,16,"                    ",python,content
1229,2657520,"genie.py",16826,16,"                    ",python,content
1230,2657520,"genie.py",16789,12,"                ",python,content
1231,2657520,"genie.py",16716,12,"                ",python,content
1232,2657520,"genie.py",16621,12,"                ",python,content
1233,2657522,"genie.py",16637,0,"",python,selection_command
1234,2658653,"genie.py",16637,1,"_",python,selection_command
1235,2659797,"genie.py",16637,0,"",python,selection_command
1236,2660092,"genie.py",16621,98,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
1237,2660482,"genie.py",16621,175,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.",python,selection_command
1238,2660581,"genie.py",16621,216,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(",python,selection_command
1239,2660716,"genie.py",16621,328,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1240,2660865,"genie.py",16621,358,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,",python,selection_command
1241,2661018,"genie.py",16621,376,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,
                )",python,selection_command
1242,2661169,"genie.py",16621,449,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,
                )
                print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1243,2661638,"genie.py",16998,16,"            ",python,content
1244,2661638,"genie.py",16980,16,"            ",python,content
1245,2661638,"genie.py",16950,20,"                ",python,content
1246,2661638,"genie.py",16838,20,"                ",python,content
1247,2661638,"genie.py",16797,16,"            ",python,content
1248,2661638,"genie.py",16720,16,"            ",python,content
1249,2661638,"genie.py",16621,16,"            ",python,content
1250,2661640,"genie.py",16633,0,"",python,selection_command
1251,2676605,"genie.py",16621,95,"",python,content
1252,2676618,"genie.py",16633,0,"",python,selection_command
1253,2677616,"genie.py",16706,0,"",python,selection_command
1254,2678355,"genie.py",16714,0,"",python,selection_command
1255,2678532,"genie.py",16716,0,"",python,selection_command
1256,2680848,"sample.py",0,0,"",python,tab
1257,2681133,"sample.py",0,0,"",python,selection_command
1258,2682895,"sample.py",68,14,"",python,content
1259,2683779,"sample.py",67,0,"",python,selection_command
1260,2684312,"genie.py",0,0,"",python,tab
1261,2686539,"genie.py",0,0,"",python,selection_command
1262,2687505,"genie.py",23,0,"
import timeit",python,content
1263,2688996,"genie.py",16730,0,"",python,selection_command
1264,2690862,"genie.py",16767,0,"",python,selection_command
1265,2691514,"genie.py",16769,0,"",python,selection_command
1266,2778454,"genie.py",16732,0,"",python,selection_command
1267,2778721,"genie.py",16659,0,"",python,selection_command
1268,2778729,"genie.py",16634,0,"",python,selection_command
1269,2778763,"genie.py",16632,0,"",python,selection_command
1270,2778795,"genie.py",16592,0,"",python,selection_command
1271,2778829,"genie.py",16558,0,"",python,selection_command
1272,2778863,"genie.py",16518,0,"",python,selection_command
1273,2778895,"genie.py",16492,0,"",python,selection_command
1274,2778928,"genie.py",16458,0,"",python,selection_command
1275,2778962,"genie.py",16405,0,"",python,selection_command
1276,2778995,"genie.py",16380,0,"",python,selection_command
1277,2779029,"genie.py",16302,0,"",python,selection_command
1278,2779061,"genie.py",16241,0,"",python,selection_command
1279,2779096,"genie.py",16177,0,"",python,selection_command
1280,2779129,"genie.py",16152,0,"",python,selection_command
1281,2779162,"genie.py",16126,0,"",python,selection_command
1282,2779196,"genie.py",16085,0,"",python,selection_command
1283,2779229,"genie.py",16059,0,"",python,selection_command
1284,2779262,"genie.py",16028,0,"",python,selection_command
1285,2779386,"genie.py",16001,0,"",python,selection_command
1286,2779545,"genie.py",15932,0,"",python,selection_command
1287,2779875,"genie.py",15976,0,"
        ",python,content
1288,2780373,"genie.py",15985,0,"d",python,content
1289,2780373,"genie.py",15986,0,"",python,selection_keyboard
1290,2780467,"genie.py",15986,0,"e",python,content
1291,2780468,"genie.py",15987,0,"",python,selection_keyboard
1292,2780715,"genie.py",15987,0,"f",python,content
1293,2780716,"genie.py",15988,0,"",python,selection_keyboard
1294,2780800,"genie.py",15988,0," ",python,content
1295,2780801,"genie.py",15989,0,"",python,selection_keyboard
1296,2780999,"genie.py",15989,0,"g",python,content
1297,2780999,"genie.py",15990,0,"",python,selection_keyboard
1298,2781190,"genie.py",15990,0,"e",python,content
1299,2781191,"genie.py",15991,0,"",python,selection_keyboard
1300,2781324,"genie.py",15991,0,"n",python,content
1301,2781324,"genie.py",15992,0,"",python,selection_keyboard
1302,2781367,"genie.py",15992,0,"e",python,content
1303,2781368,"genie.py",15993,0,"",python,selection_keyboard
1304,2781408,"genie.py",15993,0,"r",python,content
1305,2781408,"genie.py",15994,0,"",python,selection_keyboard
1306,2781671,"genie.py",15994,0,"a",python,content
1307,2781671,"genie.py",15995,0,"",python,selection_keyboard
1308,2781673,"genie.py",15995,0,"t",python,content
1309,2781673,"genie.py",15996,0,"",python,selection_keyboard
1310,2781718,"genie.py",15996,0,"e",python,content
1311,2781718,"genie.py",15997,0,"",python,selection_keyboard
1312,2783588,"genie.py",15997,0,"_",python,content
1313,2783896,"genie.py",15998,0,"o",python,content
1314,2784034,"genie.py",15999,0,"n",python,content
1315,2784035,"genie.py",16000,0,"",python,selection_keyboard
1316,2784117,"genie.py",16000,0,"e",python,content
1317,2784117,"genie.py",16001,0,"",python,selection_keyboard
1318,2784848,"genie.py",16000,1,"",python,content
1319,2784986,"genie.py",15999,1,"",python,content
1320,2785141,"genie.py",15998,1,"",python,content
1321,2787185,"genie.py",15998,0,"f",python,content
1322,2787186,"genie.py",15999,0,"",python,selection_keyboard
1323,2787305,"genie.py",15999,0,"r",python,content
1324,2787305,"genie.py",16000,0,"",python,selection_keyboard
1325,2787506,"genie.py",16000,0,"a",python,content
1326,2787506,"genie.py",16001,0,"",python,selection_keyboard
1327,2787526,"genie.py",16001,0,"m",python,content
1328,2787526,"genie.py",16002,0,"",python,selection_keyboard
1329,2787604,"genie.py",16002,0,"e",python,content
1330,2787604,"genie.py",16003,0,"",python,selection_keyboard
1331,2790406,"genie.py",16003,0,"s(genie, rng, video_batch_BSHWC, action_batch_E):",python,content
1332,2791600,"genie.py",16051,0,"",python,selection_command
1333,2792580,"genie.py",16078,0,"",python,selection_command
1334,2792826,"genie.py",16126,0,"",python,selection_command
1335,2792847,"genie.py",16135,0,"",python,selection_command
1336,2792885,"genie.py",16176,0,"",python,selection_command
1337,2792917,"genie.py",16226,0,"",python,selection_command
1338,2792949,"genie.py",16228,0,"",python,selection_command
1339,2792990,"genie.py",16291,0,"",python,selection_command
1340,2793026,"genie.py",16352,0,"",python,selection_command
1341,2793057,"genie.py",16428,0,"",python,selection_command
1342,2793090,"genie.py",16456,0,"",python,selection_command
1343,2793257,"genie.py",16428,0,"",python,selection_command
1344,2793422,"genie.py",16352,0,"",python,selection_command
1345,2793569,"genie.py",16291,0,"",python,selection_command
1346,2793717,"genie.py",16228,0,"",python,selection_command
1347,2793894,"genie.py",16226,0,"",python,selection_command
1348,2796171,"genie.py",16228,0,"",python,selection_command
1349,2796332,"genie.py",16291,0,"",python,selection_command
1350,2799170,"genie.py",16229,63,"            # --- Reset spatial KV caches before each frame ---",python,selection_command
1351,2800707,"genie.py",16229,124,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:",python,selection_command
1352,2800966,"genie.py",16229,226,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
1353,2800987,"genie.py",16229,227,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
",python,selection_command
1354,2801012,"genie.py",16229,280,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---",python,selection_command
1355,2801046,"genie.py",16229,314,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (",python,selection_command
1356,2801080,"genie.py",16229,340,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,",python,selection_command
1357,2801113,"genie.py",16229,380,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,",python,selection_command
1358,2801147,"genie.py",16229,414,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,",python,selection_command
1359,2801180,"genie.py",16229,466,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),",python,selection_command
1360,2801215,"genie.py",16229,480,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )",python,selection_command
1361,2801246,"genie.py",16229,481,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )
",python,selection_command
1362,2801279,"genie.py",16229,554,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.",python,selection_command
1363,2801312,"genie.py",16229,591,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(",python,selection_command
1364,2801347,"genie.py",16229,699,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1365,2801379,"genie.py",16229,725,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,",python,selection_command
1366,2801414,"genie.py",16229,739,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )",python,selection_command
1367,2801445,"genie.py",16229,808,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1368,2801645,"genie.py",16229,858,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            final_carry_causal, _ = jax.lax.scan(",python,selection_command
1369,2801801,"genie.py",16229,923,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)",python,selection_command
1370,2802337,"genie.py",16229,937,"            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )",python,selection_command
1371,2823229,"genie.py",16229,938,"",python,content
1372,2823255,"genie.py",16241,0,"",python,selection_command
1373,2823854,"genie.py",16228,0,"",python,selection_command
1374,2823919,"genie.py",16190,0,"",python,selection_command
1375,2824088,"genie.py",16149,0,"",python,selection_command
1376,2824237,"genie.py",16135,0,"",python,selection_command
1377,2824401,"genie.py",16092,0,"",python,selection_command
1378,2824543,"genie.py",16135,0,"",python,selection_command
1379,2824911,"genie.py",16092,0,"",python,selection_command
1380,2825067,"genie.py",16065,0,"",python,selection_command
1381,2825211,"genie.py",15989,0,"",python,selection_command
1382,2825622,"genie.py",16052,0,"
            ",python,content
1383,2825879,"genie.py",16053,12,"",python,content
1384,2825975,"genie.py",16053,0,"
            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )",python,content
1385,2825977,"genie.py",16066,0,"",python,selection_command
1386,2826811,"genie.py",16053,0,"",python,selection_command
1387,2827217,"genie.py",16053,1,"",python,content
1388,2827233,"genie.py",16065,0,"",python,selection_command
1389,2831536,"genie.py",16129,0,"",python,selection_command
1390,2831705,"genie.py",16190,0,"",python,selection_command
1391,2832533,"genie.py",16194,0,"",python,selection_command
1392,2832763,"genie.py",16199,0,"",python,selection_command
1393,2832790,"genie.py",16200,0,"",python,selection_command
1394,2832817,"genie.py",16217,0,"",python,selection_command
1395,2832852,"genie.py",16218,0,"",python,selection_command
1396,2832885,"genie.py",16228,0,"",python,selection_command
1397,2832915,"genie.py",16230,0,"",python,selection_command
1398,2832949,"genie.py",16232,0,"",python,selection_command
1399,2832984,"genie.py",16234,0,"",python,selection_command
1400,2833017,"genie.py",16241,0,"",python,selection_command
1401,2833051,"genie.py",16243,0,"",python,selection_command
1402,2833083,"genie.py",16244,0,"",python,selection_command
1403,2833116,"genie.py",16246,0,"",python,selection_command
1404,2833167,"genie.py",16250,0,"",python,selection_command
1405,2833192,"genie.py",16251,0,"",python,selection_command
1406,2833223,"genie.py",16259,0,"",python,selection_command
1407,2833250,"genie.py",16262,0,"",python,selection_command
1408,2833511,"genie.py",16267,0,"",python,selection_command
1409,2833874,"genie.py",16268,0,"",python,selection_command
1410,2834053,"genie.py",16272,0,"",python,selection_command
1411,2834211,"genie.py",16273,0,"",python,selection_command
1412,2834398,"genie.py",16278,0,"",python,selection_command
1413,2834564,"genie.py",16280,0,"",python,selection_command
1414,2834696,"genie.py",16293,0,"",python,selection_command
1415,2836637,"genie.py",16346,0,"",python,selection_command
1416,2836874,"genie.py",16380,0,"",python,selection_command
1417,2836892,"genie.py",16406,0,"",python,selection_command
1418,2836926,"genie.py",16446,0,"",python,selection_command
1419,2837071,"genie.py",16480,0,"",python,selection_command
1420,2837234,"genie.py",16532,0,"",python,selection_command
1421,2837392,"genie.py",16534,0,"",python,selection_command
1422,2837529,"genie.py",16547,0,"",python,selection_command
1423,2837662,"genie.py",16620,0,"",python,selection_command
1424,2837979,"genie.py",16657,0,"",python,selection_command
1425,2838917,"genie.py",16661,0,"",python,selection_command
1426,2840617,"genie.py",16624,0,"",python,selection_command
1427,2841328,"genie.py",16608,36,"            elapsed = timeit.timeit(",python,selection_command
1428,2842054,"genie.py",16608,144,"            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
1429,2842153,"genie.py",16608,170,"            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,",python,selection_command
1430,2842310,"genie.py",16608,184,"            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )",python,selection_command
1431,2842639,"genie.py",16608,253,"            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
1432,2843237,"genie.py",16805,0,"# ",python,content
1433,2843237,"genie.py",16791,0,"# ",python,content
1434,2843237,"genie.py",16765,0,"# ",python,content
1435,2843237,"genie.py",16657,0,"# ",python,content
1436,2843237,"genie.py",16620,0,"# ",python,content
1437,2843926,"genie.py",16817,0,"",python,selection_command
1438,2846946,"genie.py",16888,0,"",python,selection_command
1439,2847755,"genie.py",16938,0,"",python,selection_command
1440,2848149,"genie.py",16999,0,"",python,selection_command
1441,2848594,"genie.py",17000,0,"
            ",python,content
1442,2848800,"genie.py",17013,0,"r",python,content
1443,2849338,"genie.py",17014,0,"eturn final_carry_causal[1]
",python,content
1444,2852287,"genie.py",17043,0,"",python,selection_command
1445,2852454,"genie.py",17070,0,"",python,selection_command
1446,2852600,"genie.py",17118,0,"",python,selection_command
1447,2852760,"genie.py",17127,0,"",python,selection_command
1448,2852893,"genie.py",17168,0,"",python,selection_command
1449,2853135,"genie.py",17218,0,"",python,selection_command
1450,2853166,"genie.py",17219,0,"",python,selection_command
1451,2853548,"genie.py",17218,0,"",python,selection_command
1452,2853964,"genie.py",17219,0,"",python,selection_command
1453,2855029,"genie.py",17231,0,"",python,selection_command
1454,2855181,"genie.py",17254,0,"",python,selection_command
1455,2855347,"genie.py",17256,0,"",python,selection_command
1456,2857881,"genie.py",17256,21,"",python,content
1457,2859228,"genie.py",17256,0,"generate_frames(self, rng, videos_BTHWC, action_tokens_EL)",python,content
1458,2859542,"genie.py",17313,0,"",python,selection_command
1459,2860516,"genie.py",17219,0,"",python,selection_command
1460,2862709,"genie.py",17218,0,"",python,selection_command
1461,2863288,"genie.py",17218,1,"",python,content
1462,2863310,"genie.py",17230,0,"",python,selection_command
1463,2872346,"genie.py",17253,0,"",python,selection_command
1464,2872659,"genie.py",17255,0,"",python,selection_command
1465,2872929,"genie.py",17270,0,"",python,selection_command
1466,2872939,"genie.py",17271,0,"",python,selection_command
1467,2872973,"genie.py",17275,0,"",python,selection_command
1468,2873049,"genie.py",17277,0,"",python,selection_command
1469,2873052,"genie.py",17280,0,"",python,selection_command
1470,2873079,"genie.py",17282,0,"",python,selection_command
1471,2873106,"genie.py",17294,0,"",python,selection_command
1472,2873139,"genie.py",17296,0,"",python,selection_command
1473,2873169,"genie.py",17312,0,"",python,selection_command
1474,2873204,"genie.py",17331,0,"",python,selection_command
1475,2873237,"genie.py",17352,0,"",python,selection_command
1476,2873270,"genie.py",17354,0,"",python,selection_command
1477,2873432,"genie.py",17377,0,"",python,selection_command
1478,2874018,"genie.py",17354,0,"",python,selection_command
1479,2874179,"genie.py",17352,0,"",python,selection_command
1480,2874353,"genie.py",17331,0,"",python,selection_command
1481,2874637,"genie.py",17312,0,"",python,selection_command
1482,2874794,"genie.py",17296,0,"",python,selection_command
1483,2875515,"genie.py",17294,0,"",python,selection_command
1484,2875679,"genie.py",17282,0,"",python,selection_command
1485,2875823,"genie.py",17280,0,"",python,selection_command
1486,2876011,"genie.py",17277,0,"",python,selection_command
1487,2876204,"genie.py",17275,0,"",python,selection_command
1488,2876346,"genie.py",17271,0,"",python,selection_command
1489,2876881,"genie.py",17218,0,"",python,selection_command
1490,2877451,"genie.py",17230,0,"",python,selection_command
1491,2890459,"genie.py",17180,0,"",python,selection_command
1492,2890705,"genie.py",17139,0,"",python,selection_command
1493,2890725,"genie.py",17125,0,"",python,selection_command
1494,2890754,"genie.py",17082,0,"",python,selection_command
1495,2890782,"genie.py",17055,0,"",python,selection_command
1496,2890824,"genie.py",17042,0,"",python,selection_command
1497,2890857,"genie.py",17013,0,"",python,selection_command
1498,2890889,"genie.py",16999,0,"",python,selection_command
1499,2890915,"genie.py",16934,0,"",python,selection_command
1500,2890949,"genie.py",16884,0,"",python,selection_command
1501,2890983,"genie.py",16813,0,"",python,selection_command
1502,2891022,"genie.py",16797,0,"",python,selection_command
1503,2891058,"genie.py",16769,0,"",python,selection_command
1504,2891092,"genie.py",16659,0,"",python,selection_command
1505,2891116,"genie.py",16620,0,"",python,selection_command
1506,2891149,"genie.py",16547,0,"",python,selection_command
1507,2891182,"genie.py",16534,0,"",python,selection_command
1508,2891225,"genie.py",16532,0,"",python,selection_command
1509,2891384,"genie.py",16480,0,"",python,selection_command
1510,2891545,"genie.py",16446,0,"",python,selection_command
1511,2891678,"genie.py",16406,0,"",python,selection_command
1512,2891949,"genie.py",16410,0,"",python,selection_command
1513,2896580,"genie.py",16384,0,"",python,selection_command
1514,2896839,"genie.py",16350,0,"",python,selection_command
1515,2896865,"genie.py",16297,0,"",python,selection_command
1516,2896894,"genie.py",16280,0,"",python,selection_command
1517,2896929,"genie.py",16194,0,"",python,selection_command
1518,2896957,"genie.py",16133,0,"",python,selection_command
1519,2897201,"genie.py",16069,0,"",python,selection_command
1520,2897367,"genie.py",15993,0,"",python,selection_command
1521,2897797,"genie.py",16003,0,"",python,selection_command
1522,2898063,"genie.py",16004,0,"",python,selection_command
1523,2898084,"genie.py",16009,0,"",python,selection_command
1524,2898109,"genie.py",16010,0,"",python,selection_command
1525,2898149,"genie.py",16014,0,"",python,selection_command
1526,2898368,"genie.py",16015,0,"",python,selection_command
1527,2898531,"genie.py",16033,0,"",python,selection_command
1528,2898687,"genie.py",16034,0,"",python,selection_command
1529,2898827,"genie.py",16049,0,"",python,selection_command
1530,2899311,"genie.py",16050,0,"",python,selection_command
1531,2899336,"genie.py",16050,0,",",python,content
1532,2899336,"genie.py",16051,0,"",python,selection_keyboard
1533,2899519,"genie.py",16051,0," ",python,content
1534,2899519,"genie.py",16052,0,"",python,selection_keyboard
1535,2900144,"genie.py",16052,0,"c",python,content
1536,2900144,"genie.py",16053,0,"",python,selection_keyboard
1537,2900470,"genie.py",16053,0,"u",python,content
1538,2900470,"genie.py",16054,0,"",python,selection_keyboard
1539,2901267,"genie.py",16054,0,"rrent_token_idxs_BSN, step_t",python,content
1540,2901569,"genie.py",16081,0,"",python,selection_command
1541,2903428,"genie.py",16081,1,"t",python,selection_command
1542,2903601,"genie.py",16076,6,"step_t",python,selection_command
1543,2903790,"genie.py",16074,8,", step_t",python,selection_command
1544,2904850,"genie.py",16074,0,"",python,selection_command
1545,2905276,"genie.py",16075,0,"",python,selection_command
1546,2905491,"genie.py",16076,0,"",python,selection_command
1547,2905777,"genie.py",16077,0,"",python,selection_command
1548,2907304,"genie.py",15977,0,"",python,selection_command
1549,2908638,"genie.py",16084,0,"",python,selection_command
1550,2909202,"genie.py",16083,0,"",python,selection_command
1551,2909387,"genie.py",16082,0,"",python,selection_command
1552,2909532,"genie.py",16076,0,"",python,selection_command
1553,2911442,"genie.py",16074,0,"",python,selection_command
1554,2911796,"genie.py",16052,0,"",python,selection_command
1555,2912226,"genie.py",16050,0,"",python,selection_command
1556,2912764,"genie.py",15977,0,"",python,selection_command
1557,2924729,"genie.py",15985,0,"",python,selection_command
1558,2924901,"genie.py",15989,0,"",python,selection_command
1559,2925077,"genie.py",16004,0,"",python,selection_command
1560,2925259,"genie.py",16005,0,"",python,selection_command
1561,2932299,"genie.py",16113,0,"",python,selection_command
1562,2932547,"genie.py",16177,0,"",python,selection_command
1563,2932568,"genie.py",16238,0,"",python,selection_command
1564,2932601,"genie.py",16312,0,"",python,selection_command
1565,2932633,"genie.py",16341,0,"",python,selection_command
1566,2932675,"genie.py",16394,0,"",python,selection_command
1567,2932698,"genie.py",16424,0,"",python,selection_command
1568,2932739,"genie.py",16454,0,"",python,selection_command
1569,2932772,"genie.py",16494,0,"",python,selection_command
1570,2932808,"genie.py",16528,0,"",python,selection_command
1571,2932842,"genie.py",16564,0,"",python,selection_command
1572,2932874,"genie.py",16566,0,"",python,selection_command
1573,2932908,"genie.py",16595,0,"",python,selection_command
1574,2932941,"genie.py",16668,0,"",python,selection_command
1575,2932975,"genie.py",16707,0,"",python,selection_command
1576,2933005,"genie.py",16815,0,"",python,selection_command
1577,2933033,"genie.py",16831,0,"",python,selection_command
1578,2933080,"genie.py",16861,0,"",python,selection_command
1579,2933112,"genie.py",16932,0,"",python,selection_command
1580,2933139,"genie.py",16982,0,"",python,selection_command
1581,2933173,"genie.py",17031,0,"",python,selection_command
1582,2933207,"genie.py",17061,0,"",python,selection_command
1583,2933235,"genie.py",17074,0,"",python,selection_command
1584,2933268,"genie.py",17100,0,"",python,selection_command
1585,2933308,"genie.py",17130,0,"",python,selection_command
1586,2933342,"genie.py",17157,0,"",python,selection_command
1587,2933375,"genie.py",17187,0,"",python,selection_command
1588,2933509,"genie.py",17228,0,"",python,selection_command
1589,2933824,"genie.py",17278,0,"",python,selection_command
1590,2934682,"genie.py",17345,0,"",python,selection_command
1591,2935076,"genie.py",17344,0,"",python,selection_command
1592,2936031,"genie.py",17328,0,"",python,selection_command
1593,2936301,"genie.py",17343,0,"",python,selection_command
1594,2936533,"genie.py",17344,0,"",python,selection_command
1595,2936633,"genie.py",17344,0,",",python,content
1596,2936633,"genie.py",17345,0,"",python,selection_keyboard
1597,2936853,"genie.py",17345,0," ",python,content
1598,2936853,"genie.py",17346,0,"",python,selection_keyboard
1599,2937815,"genie.py",17346,0,"current_token_idxs_BSN, step_t",python,content
1600,2938138,"genie.py",17375,0,"",python,selection_command
1601,2939366,"genie.py",17250,0,"",python,selection_command
1602,2940624,"genie.py",17200,0,"",python,selection_command
1603,2940874,"genie.py",17159,0,"",python,selection_command
1604,2940905,"genie.py",17150,0,"",python,selection_command
1605,2940931,"genie.py",17102,0,"",python,selection_command
1606,2940964,"genie.py",17075,0,"",python,selection_command
1607,2941003,"genie.py",17074,0,"",python,selection_command
1608,2941031,"genie.py",17033,0,"",python,selection_command
1609,2941067,"genie.py",17019,0,"",python,selection_command
1610,2941098,"genie.py",16954,0,"",python,selection_command
1611,2941129,"genie.py",16904,0,"",python,selection_command
1612,2941160,"genie.py",16833,0,"",python,selection_command
1613,2941193,"genie.py",16817,0,"",python,selection_command
1614,2941226,"genie.py",16789,0,"",python,selection_command
1615,2941261,"genie.py",16679,0,"",python,selection_command
1616,2941294,"genie.py",16640,0,"",python,selection_command
1617,2941327,"genie.py",16567,0,"",python,selection_command
1618,2941361,"genie.py",16566,0,"",python,selection_command
1619,2941393,"genie.py",16552,0,"",python,selection_command
1620,2941427,"genie.py",16500,0,"",python,selection_command
1621,2941462,"genie.py",16466,0,"",python,selection_command
1622,2941497,"genie.py",16426,0,"",python,selection_command
1623,2941529,"genie.py",16400,0,"",python,selection_command
1624,2941561,"genie.py",16366,0,"",python,selection_command
1625,2941597,"genie.py",16313,0,"",python,selection_command
1626,2941628,"genie.py",16312,0,"",python,selection_command
1627,2941668,"genie.py",16210,0,"",python,selection_command
1628,2941780,"genie.py",16149,0,"",python,selection_command
1629,2941948,"genie.py",16085,0,"",python,selection_command
1630,2942191,"genie.py",16149,0,"",python,selection_command
1631,2942536,"genie.py",16161,0,"",python,selection_command
1632,2942706,"genie.py",16165,0,"",python,selection_command
1633,2942873,"genie.py",16171,0,"",python,selection_command
1634,2943044,"genie.py",16174,0,"",python,selection_command
1635,2946128,"genie.py",16174,15,"",python,content
1636,2947259,"genie.py",16174,0,"g",python,content
1637,2947259,"genie.py",16175,0,"",python,selection_keyboard
1638,2947289,"genie.py",16175,0,"e",python,content
1639,2947289,"genie.py",16176,0,"",python,selection_keyboard
1640,2947370,"genie.py",16176,0,"n",python,content
1641,2947371,"genie.py",16177,0,"",python,selection_keyboard
1642,2947456,"genie.py",16177,0,"i",python,content
1643,2947456,"genie.py",16178,0,"",python,selection_keyboard
1644,2947526,"genie.py",16178,0,"e",python,content
1645,2947527,"genie.py",16179,0,"",python,selection_keyboard
1646,2947793,"genie.py",16179,0,".",python,content
1647,2947793,"genie.py",16180,0,"",python,selection_keyboard
1648,2948144,"genie.py",16180,0,"d",python,content
1649,2948145,"genie.py",16181,0,"",python,selection_keyboard
1650,2948162,"genie.py",16181,0,"y",python,content
1651,2948163,"genie.py",16182,0,"",python,selection_keyboard
1652,2948322,"genie.py",16182,0,"n",python,content
1653,2948322,"genie.py",16183,0,"",python,selection_keyboard
1654,2948506,"genie.py",16183,0,"a",python,content
1655,2948507,"genie.py",16184,0,"",python,selection_keyboard
1656,2948527,"genie.py",16184,0,"m",python,content
1657,2948528,"genie.py",16185,0,"",python,selection_keyboard
1658,2948593,"genie.py",16185,0,"i",python,content
1659,2948593,"genie.py",16186,0,"",python,selection_keyboard
1660,2948641,"genie.py",16186,0,"v",python,content
1661,2948641,"genie.py",16187,0,"",python,selection_keyboard
1662,2948852,"genie.py",16187,0,"s",python,content
1663,2948852,"genie.py",16188,0,"",python,selection_keyboard
1664,2949314,"genie.py",16187,1,"",python,content
1665,2949472,"genie.py",16186,1,"",python,content
1666,2949537,"genie.py",16186,0,"c",python,content
1667,2949537,"genie.py",16187,0,"",python,selection_keyboard
1668,2949725,"genie.py",16187,0,"s",python,content
1669,2949726,"genie.py",16188,0,"",python,selection_keyboard
1670,2949908,"genie.py",16187,0,"",python,selection_command
1671,2952018,"genie.py",16247,0,"",python,selection_command
1672,2952161,"genie.py",16187,0,"",python,selection_command
1673,2952482,"genie.py",16247,0,"",python,selection_command
1674,2952930,"genie.py",16248,0,"",python,selection_command
1675,2953127,"genie.py",16249,0,"",python,selection_command
1676,2953314,"genie.py",16259,0,"",python,selection_command
1677,2953493,"genie.py",16261,0,"",python,selection_command
1678,2953643,"genie.py",16263,0,"",python,selection_command
1679,2953798,"genie.py",16265,0,"",python,selection_command
1680,2953997,"genie.py",16272,0,"",python,selection_command
1681,2954136,"genie.py",16274,0,"",python,selection_command
1682,2954299,"genie.py",16275,0,"",python,selection_command
1683,2954834,"genie.py",16277,0,"",python,selection_command
1684,2956285,"genie.py",16281,0,"",python,selection_command
1685,2956430,"genie.py",16282,0,"",python,selection_command
1686,2956917,"genie.py",2021,0,"",python,selection_command
1687,2957414,"genie.py",16282,0,"",python,selection_command
1688,2957982,"genie.py",16290,0,"",python,selection_command
1689,2958161,"genie.py",16293,0,"",python,selection_command
1690,2958303,"genie.py",16298,0,"",python,selection_command
1691,2958482,"genie.py",16299,0,"",python,selection_command
1692,2959133,"genie.py",16303,0,"",python,selection_command
1693,2959385,"genie.py",16304,0,"",python,selection_command
1694,2960465,"genie.py",16309,0,"",python,selection_command
1695,2960882,"genie.py",16311,0,"",python,selection_command
1696,2963668,"genie.py",16312,0,"",python,selection_command
1697,2963862,"genie.py",16365,0,"",python,selection_command
1698,2964003,"genie.py",16399,0,"",python,selection_command
1699,2964146,"genie.py",16425,0,"",python,selection_command
1700,2964300,"genie.py",16465,0,"",python,selection_command
1701,2964451,"genie.py",16499,0,"",python,selection_command
1702,2964622,"genie.py",16551,0,"",python,selection_command
1703,2964785,"genie.py",16565,0,"",python,selection_command
1704,2965219,"genie.py",16566,0,"",python,selection_command
1705,2972492,"genie.py",16565,0,"",python,selection_command
1706,2973121,"genie.py",16551,0,"",python,selection_command
1707,2973271,"genie.py",16499,0,"",python,selection_command
1708,2973543,"genie.py",16465,0,"",python,selection_command
1709,2973560,"genie.py",16425,0,"",python,selection_command
1710,2973587,"genie.py",16399,0,"",python,selection_command
1711,2973624,"genie.py",16365,0,"",python,selection_command
1712,2973652,"genie.py",16312,0,"",python,selection_command
1713,2973693,"genie.py",16311,0,"",python,selection_command
1714,2973726,"genie.py",16209,0,"",python,selection_command
1715,2973851,"genie.py",16149,0,"",python,selection_command
1716,2974003,"genie.py",16085,0,"",python,selection_command
1717,2974181,"genie.py",15977,0,"",python,selection_command
1718,2974597,"genie.py",16003,0,"",python,selection_command
1719,2974597,"genie.py",16004,0,"",python,selection_command
1720,2975468,"genie.py",16003,0,"",python,selection_command
1721,2975724,"genie.py",16003,1,"",python,content
1722,2975877,"genie.py",16003,1,"",python,content
1723,2976347,"genie.py",16003,0,"(",python,content
1724,2976350,"genie.py",16003,0,"",python,selection_command
1725,2976630,"genie.py",16003,1,"",python,content
1726,2977066,"genie.py",16003,0,"(",python,content
1727,2977068,"genie.py",16003,0,"",python,selection_command
1728,2978346,"genie.py",16110,0,"",python,selection_command
1729,2978593,"genie.py",16174,0,"",python,selection_command
1730,2978609,"genie.py",16234,0,"",python,selection_command
1731,2978645,"genie.py",16310,0,"",python,selection_command
1732,2978678,"genie.py",16337,0,"",python,selection_command
1733,2978711,"genie.py",16390,0,"",python,selection_command
1734,2978747,"genie.py",16422,0,"",python,selection_command
1735,2978781,"genie.py",16450,0,"",python,selection_command
1736,2978812,"genie.py",16490,0,"",python,selection_command
1737,2978846,"genie.py",16524,0,"",python,selection_command
1738,2978879,"genie.py",16562,0,"",python,selection_command
1739,2978913,"genie.py",16564,0,"",python,selection_command
1740,2978946,"genie.py",16591,0,"",python,selection_command
1741,2978984,"genie.py",16664,0,"",python,selection_command
1742,2979014,"genie.py",16703,0,"",python,selection_command
1743,2979048,"genie.py",16813,0,"",python,selection_command
1744,2979080,"genie.py",16829,0,"",python,selection_command
1745,2979114,"genie.py",16857,0,"",python,selection_command
1746,2979146,"genie.py",16928,0,"",python,selection_command
1747,2979179,"genie.py",16978,0,"",python,selection_command
1748,2979220,"genie.py",17029,0,"",python,selection_command
1749,2979246,"genie.py",17057,0,"",python,selection_command
1750,2979281,"genie.py",17072,0,"",python,selection_command
1751,2979312,"genie.py",17098,0,"",python,selection_command
1752,2979346,"genie.py",17126,0,"",python,selection_command
1753,2979380,"genie.py",17155,0,"",python,selection_command
1754,2979414,"genie.py",17183,0,"",python,selection_command
1755,2979447,"genie.py",17224,0,"",python,selection_command
1756,2979564,"genie.py",17274,0,"",python,selection_command
1757,2980030,"genie.py",17277,0,"",python,selection_command
1758,2982783,"genie.py",17299,0,"",python,selection_command
1759,2983433,"genie.py",17299,1,"",python,content
1760,2998380,"genie.py",17300,0,"",python,selection_command
1761,2998534,"genie.py",17304,0,"",python,selection_command
1762,2998704,"genie.py",17306,0,"",python,selection_command
1763,2998997,"genie.py",17309,0,"",python,selection_command
1764,2999149,"genie.py",17311,0,"",python,selection_command
1765,3000745,"genie.py",13671,0,"",python,selection_command
1766,3004672,"genie.py",13802,0,"",python,selection_command
1767,3005397,"genie.py",17311,0,"",python,selection_command
1768,3006575,"genie.py",13802,0,"",python,selection_command
1769,3007256,"genie.py",13801,0,"",python,selection_command
1770,3007375,"genie.py",13792,0,"",python,selection_command
1771,3007557,"genie.py",13791,0,"",python,selection_command
1772,3007723,"genie.py",13782,0,"",python,selection_command
1773,3007856,"genie.py",13781,0,"",python,selection_command
1774,3007994,"genie.py",13777,0,"",python,selection_command
1775,3008253,"genie.py",13775,0,"",python,selection_command
1776,3008278,"genie.py",13761,0,"",python,selection_command
1777,3008304,"genie.py",13750,0,"",python,selection_command
1778,3008402,"genie.py",13736,0,"",python,selection_command
1779,3010093,"genie.py",15787,0,"",python,selection_command
1780,3010317,"genie.py",17392,0,"",python,selection_command
1781,3011271,"genie.py",17382,0,"",python,selection_command
1782,3011493,"genie.py",17256,0,"",python,selection_command
1783,3011526,"genie.py",17206,0,"",python,selection_command
1784,3011951,"genie.py",17210,0,"",python,selection_command
1785,3012629,"genie.py",17260,0,"",python,selection_command
1786,3012745,"genie.py",17283,0,"",python,selection_command
1787,3012913,"genie.py",17285,0,"",python,selection_command
1788,3013279,"genie.py",17299,0,"",python,selection_command
1789,3013448,"genie.py",17300,0,"",python,selection_command
1790,3013563,"genie.py",17304,0,"",python,selection_command
1791,3013796,"genie.py",17306,0,"",python,selection_command
1792,3014037,"genie.py",17309,0,"",python,selection_command
1793,3014586,"genie.py",17311,0,"",python,selection_command
1794,3016863,"genie.py",17323,0,"",python,selection_command
1795,3017019,"genie.py",17325,0,"",python,selection_command
1796,3022824,"genie.py",17340,0,"",python,selection_command
1797,3024235,"genie.py",17325,0,"",python,selection_command
1798,3024587,"genie.py",17323,0,"",python,selection_command
1799,3024834,"genie.py",17311,0,"",python,selection_command
1800,3024860,"genie.py",17309,0,"",python,selection_command
1801,3024882,"genie.py",17306,0,"",python,selection_command
1802,3024917,"genie.py",17304,0,"",python,selection_command
1803,3024950,"genie.py",17300,0,"",python,selection_command
1804,3024983,"genie.py",17299,0,"",python,selection_command
1805,3025017,"genie.py",17285,0,"",python,selection_command
1806,3025290,"genie.py",17283,0,"",python,selection_command
1807,3025542,"genie.py",17260,0,"",python,selection_command
1808,3025569,"genie.py",17246,0,"",python,selection_command
1809,3025592,"genie.py",17243,0,"",python,selection_command
1810,3025628,"genie.py",17242,0,"",python,selection_command
1811,3025660,"genie.py",17237,0,"",python,selection_command
1812,3025693,"genie.py",17236,0,"",python,selection_command
1813,3026006,"genie.py",17230,0,"",python,selection_command
1814,3026407,"genie.py",17229,0,"",python,selection_command
1815,3026541,"genie.py",17226,0,"",python,selection_command
1816,3026690,"genie.py",17224,0,"",python,selection_command
1817,3026868,"genie.py",17215,0,"",python,selection_command
1818,3027019,"genie.py",17213,0,"",python,selection_command
1819,3027178,"genie.py",17210,0,"",python,selection_command
1820,3027371,"genie.py",17195,0,"",python,selection_command
1821,3027592,"genie.py",17188,0,"",python,selection_command
1822,3028344,"genie.py",17155,0,"",python,selection_command
1823,3028568,"genie.py",17131,0,"",python,selection_command
1824,3028608,"genie.py",17098,0,"",python,selection_command
1825,3028644,"genie.py",17072,0,"",python,selection_command
1826,3028667,"genie.py",17062,0,"",python,selection_command
1827,3028705,"genie.py",17029,0,"",python,selection_command
1828,3028735,"genie.py",16983,0,"",python,selection_command
1829,3028771,"genie.py",16933,0,"",python,selection_command
1830,3028804,"genie.py",16862,0,"",python,selection_command
1831,3028836,"genie.py",16829,0,"",python,selection_command
1832,3028981,"genie.py",16813,0,"",python,selection_command
1833,3029240,"genie.py",16708,0,"",python,selection_command
1834,3029257,"genie.py",16669,0,"",python,selection_command
1835,3029289,"genie.py",16596,0,"",python,selection_command
1836,3029451,"genie.py",16564,0,"",python,selection_command
1837,3029611,"genie.py",16562,0,"",python,selection_command
1838,3029773,"genie.py",16529,0,"",python,selection_command
1839,3030251,"genie.py",16495,0,"",python,selection_command
1840,3030903,"genie.py",16455,0,"",python,selection_command
1841,3031105,"genie.py",16422,0,"",python,selection_command
1842,3031122,"genie.py",16395,0,"",python,selection_command
1843,3031160,"genie.py",16342,0,"",python,selection_command
1844,3031185,"genie.py",16310,0,"",python,selection_command
1845,3031221,"genie.py",16239,0,"",python,selection_command
1846,3031254,"genie.py",16179,0,"",python,selection_command
1847,3031378,"genie.py",16115,0,"",python,selection_command
1848,3031729,"genie.py",16008,0,"",python,selection_command
1849,3032293,"genie.py",16009,0,"",python,selection_command
1850,3032554,"genie.py",16011,0,"",python,selection_command
1851,3032558,"genie.py",16014,0,"",python,selection_command
1852,3032587,"genie.py",16016,0,"",python,selection_command
1853,3032621,"genie.py",16033,0,"",python,selection_command
1854,3032839,"genie.py",16035,0,"",python,selection_command
1855,3034274,"genie.py",16036,0,"",python,selection_command
1856,3034521,"genie.py",16037,0,"",python,selection_command
1857,3034539,"genie.py",16038,0,"",python,selection_command
1858,3034574,"genie.py",16039,0,"",python,selection_command
1859,3034605,"genie.py",16040,0,"",python,selection_command
1860,3034641,"genie.py",16041,0,"",python,selection_command
1861,3034850,"genie.py",16042,0,"",python,selection_command
1862,3035288,"genie.py",16042,7,"",python,content
1863,3035836,"genie.py",16042,0,"t",python,content
1864,3035836,"genie.py",16043,0,"",python,selection_keyboard
1865,3035911,"genie.py",16043,0,"o",python,content
1866,3035911,"genie.py",16044,0,"",python,selection_keyboard
1867,3036098,"genie.py",16044,0,"k",python,content
1868,3036098,"genie.py",16045,0,"",python,selection_keyboard
1869,3036116,"genie.py",16045,0,"e",python,content
1870,3036116,"genie.py",16046,0,"",python,selection_keyboard
1871,3036196,"genie.py",16046,0,"n",python,content
1872,3036196,"genie.py",16047,0,"",python,selection_keyboard
1873,3036363,"genie.py",16047,0,"s",python,content
1874,3036363,"genie.py",16048,0,"",python,selection_keyboard
1875,3036646,"genie.py",16048,0,"_",python,content
1876,3036646,"genie.py",16049,0,"",python,selection_keyboard
1877,3037216,"genie.py",16049,0,"E",python,content
1878,3037216,"genie.py",16050,0,"",python,selection_keyboard
1879,3037367,"genie.py",16050,0,"L",python,content
1880,3037367,"genie.py",16051,0,"",python,selection_keyboard
1881,3037669,"genie.py",16050,0,"",python,selection_command
1882,3038481,"genie.py",16148,0,"",python,selection_command
1883,3038713,"genie.py",16208,0,"",python,selection_command
1884,3038745,"genie.py",16283,0,"",python,selection_command
1885,3038775,"genie.py",16312,0,"",python,selection_command
1886,3038809,"genie.py",16364,0,"",python,selection_command
1887,3038842,"genie.py",16398,0,"",python,selection_command
1888,3038876,"genie.py",16424,0,"",python,selection_command
1889,3038908,"genie.py",16464,0,"",python,selection_command
1890,3039169,"genie.py",16498,0,"",python,selection_command
1891,3039423,"genie.py",16464,0,"",python,selection_command
1892,3039950,"genie.py",16463,0,"",python,selection_command
1893,3040512,"genie.py",16498,0,"",python,selection_command
1894,3040651,"genie.py",16497,0,"",python,selection_command
1895,3040812,"genie.py",16496,0,"",python,selection_command
1896,3043027,"genie.py",16456,0,"",python,selection_command
1897,3043283,"genie.py",16424,0,"",python,selection_command
1898,3043302,"genie.py",16396,0,"",python,selection_command
1899,3043323,"genie.py",16343,0,"",python,selection_command
1900,3043365,"genie.py",16312,0,"",python,selection_command
1901,3043391,"genie.py",16240,0,"",python,selection_command
1902,3043514,"genie.py",16180,0,"",python,selection_command
1903,3043674,"genie.py",16116,0,"",python,selection_command
1904,3043820,"genie.py",16007,0,"",python,selection_command
1905,3044573,"genie.py",16008,0,"",python,selection_command
1906,3044709,"genie.py",16009,0,"",python,selection_command
1907,3044874,"genie.py",16011,0,"",python,selection_command
1908,3045051,"genie.py",16014,0,"",python,selection_command
1909,3045243,"genie.py",16016,0,"",python,selection_command
1910,3045629,"genie.py",16016,1,"v",python,selection_command
1911,3045688,"genie.py",16016,17,"video_batch_BSHWC",python,selection_command
1912,3046222,"genie.py",16016,18,"video_batch_BSHWC,",python,selection_command
1913,3046387,"genie.py",16016,19,"video_batch_BSHWC, ",python,selection_command
1914,3046388,"genie.py",16016,19,"",python,content
1915,3048186,"genie.py",16106,0,"",python,selection_command
1916,3048446,"genie.py",16170,0,"",python,selection_command
1917,3048473,"genie.py",16230,0,"",python,selection_command
1918,3048500,"genie.py",16293,0,"",python,selection_command
1919,3048541,"genie.py",16333,0,"",python,selection_command
1920,3048569,"genie.py",16379,0,"",python,selection_command
1921,3048597,"genie.py",16405,0,"",python,selection_command
1922,3048635,"genie.py",16445,0,"",python,selection_command
1923,3048668,"genie.py",16479,0,"",python,selection_command
1924,3048701,"genie.py",16520,0,"",python,selection_command
1925,3048736,"genie.py",16545,0,"",python,selection_command
1926,3048769,"genie.py",16547,0,"",python,selection_command
1927,3048804,"genie.py",16587,0,"",python,selection_command
1928,3048836,"genie.py",16658,0,"",python,selection_command
1929,3048868,"genie.py",16699,0,"",python,selection_command
1930,3048902,"genie.py",16796,0,"",python,selection_command
1931,3048935,"genie.py",16812,0,"",python,selection_command
1932,3048990,"genie.py",16853,0,"",python,selection_command
1933,3049014,"genie.py",16924,0,"",python,selection_command
1934,3049036,"genie.py",16974,0,"",python,selection_command
1935,3049073,"genie.py",17012,0,"",python,selection_command
1936,3049099,"genie.py",17053,0,"",python,selection_command
1937,3049133,"genie.py",17055,0,"",python,selection_command
1938,3049166,"genie.py",17081,0,"",python,selection_command
1939,3049308,"genie.py",17122,0,"",python,selection_command
1940,3049471,"genie.py",17138,0,"",python,selection_command
1941,3049625,"genie.py",17179,0,"",python,selection_command
1942,3049761,"genie.py",17220,0,"",python,selection_command
1943,3049920,"genie.py",17270,0,"",python,selection_command
1944,3051090,"genie.py",17294,14,"",python,content
1945,3052612,"genie.py",17282,0,"",python,selection_command
1946,3052856,"genie.py",17283,0,"",python,selection_command
1947,3052889,"genie.py",17287,0,"",python,selection_command
1948,3052915,"genie.py",17289,0,"",python,selection_command
1949,3052949,"genie.py",17292,0,"",python,selection_command
1950,3053444,"genie.py",17289,0,"",python,selection_command
1951,3057099,"genie.py",17229,0,"",python,selection_command
1952,3057344,"genie.py",17179,0,"",python,selection_command
1953,3057364,"genie.py",17138,0,"",python,selection_command
1954,3057390,"genie.py",17129,0,"",python,selection_command
1955,3057428,"genie.py",17081,0,"",python,selection_command
1956,3057466,"genie.py",17055,0,"",python,selection_command
1957,3057495,"genie.py",17053,0,"",python,selection_command
1958,3057530,"genie.py",17012,0,"",python,selection_command
1959,3057557,"genie.py",16993,0,"",python,selection_command
1960,3057594,"genie.py",16933,0,"",python,selection_command
1961,3057623,"genie.py",16872,0,"",python,selection_command
1962,3057659,"genie.py",16812,0,"",python,selection_command
1963,3057691,"genie.py",16796,0,"",python,selection_command
1964,3057723,"genie.py",16718,0,"",python,selection_command
1965,3057758,"genie.py",16658,0,"",python,selection_command
1966,3057789,"genie.py",16606,0,"",python,selection_command
1967,3057825,"genie.py",16547,0,"",python,selection_command
1968,3057856,"genie.py",16545,0,"",python,selection_command
1969,3057892,"genie.py",16531,0,"",python,selection_command
1970,3057925,"genie.py",16479,0,"",python,selection_command
1971,3057958,"genie.py",16445,0,"",python,selection_command
1972,3057990,"genie.py",16405,0,"",python,selection_command
1973,3058024,"genie.py",16379,0,"",python,selection_command
1974,3058057,"genie.py",16345,0,"",python,selection_command
1975,3058091,"genie.py",16293,0,"",python,selection_command
1976,3058126,"genie.py",16249,0,"",python,selection_command
1977,3058354,"genie.py",16189,0,"",python,selection_command
1978,3058496,"genie.py",16125,0,"",python,selection_command
1979,3058804,"genie.py",16189,0,"",python,selection_command
1980,3058985,"genie.py",16249,0,"",python,selection_command
1981,3060176,"genie.py",16293,0,"",python,selection_command
1982,3060325,"genie.py",16345,0,"",python,selection_command
1983,3060615,"genie.py",16379,0,"",python,selection_command
1984,3061133,"genie.py",16405,0,"",python,selection_command
1985,3064608,"genie.py",16397,5,"",python,content
1986,3071178,"genie.py",16421,0,"",python,selection_command
1987,3071426,"genie.py",16461,0,"",python,selection_command
1988,3071444,"genie.py",16495,0,"",python,selection_command
1989,3071488,"genie.py",16540,0,"",python,selection_command
1990,3071512,"genie.py",16542,0,"",python,selection_command
1991,3071545,"genie.py",16562,0,"",python,selection_command
1992,3071579,"genie.py",16635,0,"",python,selection_command
1993,3071612,"genie.py",16674,0,"",python,selection_command
1994,3071647,"genie.py",16784,0,"",python,selection_command
1995,3071680,"genie.py",16807,0,"",python,selection_command
1996,3071714,"genie.py",16828,0,"",python,selection_command
1997,3071748,"genie.py",16899,0,"",python,selection_command
1998,3071788,"genie.py",16949,0,"",python,selection_command
1999,3071821,"genie.py",17007,0,"",python,selection_command
2000,3071850,"genie.py",17028,0,"",python,selection_command
2001,3071895,"genie.py",17050,0,"",python,selection_command
2002,3071924,"genie.py",17070,0,"",python,selection_command
2003,3071955,"genie.py",17097,0,"",python,selection_command
2004,3071987,"genie.py",17133,0,"",python,selection_command
2005,3072022,"genie.py",17154,0,"",python,selection_command
2006,3072053,"genie.py",17195,0,"",python,selection_command
2007,3072087,"genie.py",17245,0,"",python,selection_command
2008,3072528,"genie.py",17261,0,"",python,selection_command
2009,3072792,"genie.py",17263,0,"",python,selection_command
2010,3072807,"genie.py",17277,0,"",python,selection_command
2011,3072832,"genie.py",17278,0,"",python,selection_command
2012,3072864,"genie.py",17282,0,"",python,selection_command
2013,3072897,"genie.py",17284,0,"",python,selection_command
2014,3072931,"genie.py",17287,0,"",python,selection_command
2015,3073357,"genie.py",17284,0,"",python,selection_command
2016,3074091,"genie.py",17284,3,"",python,content
2017,3074652,"genie.py",17284,0,"s",python,content
2018,3074652,"genie.py",17285,0,"",python,selection_keyboard
2019,3074724,"genie.py",17285,0,"t",python,content
2020,3074724,"genie.py",17286,0,"",python,selection_keyboard
2021,3074758,"genie.py",17286,0,"e",python,content
2022,3074759,"genie.py",17287,0,"",python,selection_keyboard
2023,3074824,"genie.py",17287,0,"p",python,content
2024,3074825,"genie.py",17288,0,"",python,selection_keyboard
2025,3075214,"genie.py",17288,0,"_",python,content
2026,3075214,"genie.py",17289,0,"",python,selection_keyboard
2027,3075403,"genie.py",17289,0,"r",python,content
2028,3075403,"genie.py",17290,0,"",python,selection_keyboard
2029,3075545,"genie.py",17290,0,"n",python,content
2030,3075546,"genie.py",17291,0,"",python,selection_keyboard
2031,3075787,"genie.py",17290,0,"",python,selection_command
2032,3076399,"genie.py",17291,0,"",python,selection_command
2033,3076550,"genie.py",17291,0,"g",python,content
2034,3076550,"genie.py",17292,0,"",python,selection_keyboard
2035,3076831,"genie.py",17291,0,"",python,selection_command
2036,3093191,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2037,3093459,"TERMINAL",0,0,"

",,terminal_output
2038,3105187,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2039,3114244,"TERMINAL",0,0,"2025-08-01 10:27:02.987861: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2040,3118875,"TERMINAL",0,0,"2025-08-01 10:27:07.621697: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2041,3128214,"TERMINAL",0,0,"2025-08-01 10:27:16.962207: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2042,3129304,"experiments/sample.sh",0,0,"",shellscript,tab
2043,3130560,"genie.py",0,0,"",python,tab
2044,3142521,"TERMINAL",0,0,"2025-08-01 10:27:31.265518: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2045,3153850,"genie.py",17351,0,"",python,selection_command
2046,3154116,"genie.py",17291,0,"",python,selection_command
2047,3154328,"genie.py",17224,0,"",python,selection_command
2048,3154490,"genie.py",17174,0,"",python,selection_command
2049,3154726,"genie.py",17133,0,"",python,selection_command
2050,3154753,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
2051,3154755,"genie.py",17124,0,"",python,selection_command
2052,3154791,"genie.py",17076,0,"",python,selection_command
2053,3154824,"genie.py",17050,0,"",python,selection_command
2054,3154858,"genie.py",17048,0,"",python,selection_command
2055,3155021,"genie.py",17007,0,"",python,selection_command
2056,3155169,"genie.py",16993,0,"",python,selection_command
2057,3155309,"genie.py",16928,0,"",python,selection_command
2058,3155765,"genie.py",16993,0,"",python,selection_command
2059,3155931,"genie.py",17007,0,"",python,selection_command
2060,3156245,"genie.py",16993,0,"",python,selection_command
2061,3156362,"genie.py",16928,0,"",python,selection_command
2062,3156479,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2063,3156665,"genie.py",16993,0,"",python,selection_command
2064,3156800,"genie.py",16928,0,"",python,selection_command
2065,3157000,"genie.py",16874,0,"",python,selection_command
2066,3157699,"genie.py",16809,70,"            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2067,3158135,"genie.py",16793,86,"            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2068,3158158,"genie.py",16765,114,"            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2069,3158308,"genie.py",16655,224,"            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2070,3158455,"genie.py",16616,263,"            # elapsed = timeit.timeit(
            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2071,3158605,"genie.py",16543,336,"            # Measure autoregressive generation time once and report it.
            # elapsed = timeit.timeit(
            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2072,3159526,"genie.py",16542,337,"
            # Measure autoregressive generation time once and report it.
            # elapsed = timeit.timeit(
            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2073,3159810,"genie.py",16542,338,"",python,content
2074,3159835,"genie.py",16554,0,"",python,selection_command
2075,3160802,"genie.py",16541,0,"
            ",python,content
2076,3161030,"genie.py",16542,12,"",python,content
2077,3161397,"genie.py",16543,0,"",python,selection_command
2078,3161651,"genie.py",16593,0,"",python,selection_command
2079,3161678,"genie.py",16658,0,"",python,selection_command
2080,3161713,"genie.py",16672,0,"",python,selection_command
2081,3161746,"genie.py",16713,0,"",python,selection_command
2082,3161781,"genie.py",16714,0,"",python,selection_command
2083,3161805,"genie.py",16741,0,"",python,selection_command
2084,3161845,"genie.py",16789,0,"",python,selection_command
2085,3161880,"genie.py",16798,0,"",python,selection_command
2086,3161913,"genie.py",16839,0,"",python,selection_command
2087,3162112,"genie.py",16798,0,"",python,selection_command
2088,3162378,"genie.py",16839,0,"",python,selection_command
2089,3162550,"genie.py",16889,0,"",python,selection_command
2090,3162900,"genie.py",16839,0,"",python,selection_command
2091,3163086,"genie.py",16798,0,"",python,selection_command
2092,3163287,"genie.py",16839,0,"",python,selection_command
2093,3164105,"genie.py",16888,0,"
            ",python,content
2094,3164321,"genie.py",16889,12,"",python,content
2095,3164353,"genie.py",16889,0,"

            # Measure autoregressive generation time once and report it.
            # elapsed = timeit.timeit(
            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,content
2096,3164357,"genie.py",16890,0,"",python,selection_command
2097,3165621,"genie.py",16890,1,"",python,content
2098,3165624,"genie.py",16902,0,"",python,selection_command
2099,3176664,"genie.py",16975,0,"",python,selection_command
2100,3176916,"genie.py",17014,0,"",python,selection_command
2101,3176934,"genie.py",17124,0,"",python,selection_command
2102,3176968,"genie.py",17152,0,"",python,selection_command
2103,3177008,"genie.py",17168,0,"",python,selection_command
2104,3177042,"genie.py",17239,0,"",python,selection_command
2105,3177422,"genie.py",17168,0,"",python,selection_command
2106,3177834,"genie.py",17226,0,"
            ",python,content
2107,3178350,"genie.py",17227,12,"",python,content
2108,3178683,"genie.py",17228,0,"",python,selection_command
2109,3179446,"genie.py",17240,0,"# ",python,content
2110,3182201,"genie.py",17227,0,"",python,selection_command
2111,3182451,"genie.py",17156,0,"",python,selection_command
2112,3182668,"genie.py",17168,0,"",python,selection_command
2113,3183428,"genie.py",17168,1,"",python,content
2114,3184229,"genie.py",17168,1,"",python,content
2115,3184642,"genie.py",17152,0,"",python,selection_command
2116,3185586,"genie.py",17153,0,"",python,selection_command
2117,3186354,"genie.py",17153,1," ",python,selection_command
2118,3186621,"genie.py",17152,2,"# ",python,selection_command
2119,3186824,"genie.py",17124,2,"# ",python,selection_command
2120,3186995,"genie.py",17014,2,"# ",python,selection_command
2121,3187026,"genie.py",16975,2,"# ",python,selection_command
2122,3187288,"genie.py",16902,2,"# ",python,selection_command
2123,3187810,"genie.py",16975,2,"# ",python,selection_command
2124,3188177,"genie.py",17152,2,"",python,content
2125,3188177,"genie.py",17124,2,"",python,content
2126,3188177,"genie.py",17014,2,"",python,content
2127,3188177,"genie.py",16975,2,"",python,content
2128,3188190,"genie.py",16975,0,"",python,selection_command
2129,3189120,"genie.py",17012,0,"",python,selection_command
2130,3189424,"genie.py",17120,0,"",python,selection_command
2131,3189605,"genie.py",17012,0,"",python,selection_command
2132,3189804,"genie.py",16975,0,"",python,selection_command
2133,3190081,"genie.py",17012,0,"",python,selection_command
2134,3190191,"genie.py",17016,0,"",python,selection_command
2135,3190379,"genie.py",17022,0,"",python,selection_command
2136,3190695,"genie.py",17024,0,"",python,selection_command
2137,3192500,"genie.py",17024,1,"_",python,selection_command
2138,3192559,"genie.py",17024,15,"_autoreg_sample",python,selection_command
2139,3195007,"genie.py",17038,0,"",python,selection_command
2140,3195459,"genie.py",17132,0,"",python,selection_command
2141,3195479,"genie.py",17146,0,"",python,selection_command
2142,3195622,"genie.py",17186,0,"",python,selection_command
2143,3195772,"genie.py",17217,0,"",python,selection_command
2144,3196091,"genie.py",17256,0,"",python,selection_command
2145,3196546,"genie.py",17257,0,"",python,selection_command
2146,3196895,"genie.py",17257,1,"g",python,selection_command
2147,3198281,"genie.py",17257,81,"generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t)
",python,selection_command
2148,3199571,"genie.py",17257,80,"generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t)",python,selection_command
2149,3199965,"genie.py",17257,0,"",python,selection_command
2150,3200299,"genie.py",17217,0,"",python,selection_command
2151,3200456,"genie.py",17187,0,"",python,selection_command
2152,3200517,"genie.py",17146,0,"",python,selection_command
2153,3201037,"genie.py",17132,0,"",python,selection_command
2154,3201785,"genie.py",17039,0,"",python,selection_command
2155,3202408,"genie.py",17024,0,"",python,selection_command
2156,3202532,"genie.py",17022,0,"",python,selection_command
2157,3202992,"genie.py",17024,0,"",python,selection_command
2158,3203225,"genie.py",17024,1,"_",python,selection_command
2159,3203235,"genie.py",17024,15,"_autoreg_sample",python,selection_command
2160,3203974,"genie.py",17024,84,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
",python,selection_command
2161,3204637,"genie.py",17024,83,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
2162,3205388,"genie.py",17024,82,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
2163,3205840,"genie.py",17024,82,"generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t)",python,content
2164,3205846,"genie.py",17103,0,"",python,selection_command
2165,3207002,"genie.py",17000,0,"",python,selection_command
2166,3213317,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2167,3214456,"TERMINAL",0,0,"

",,terminal_output
2168,3226036,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2169,3233643,"genie.py",17105,0,"",python,selection_command
2170,3234234,"genie.py",17104,0,"",python,selection_command
2171,3235002,"genie.py",17080,23,"il_ready(",python,content
2172,3235026,"genie.py",17075,3,"",python,content
2173,3235029,"genie.py",17069,5,"",python,content
2174,3235030,"genie.py",17065,3,"",python,content
2175,3235032,"genie.py",17064,0,"c",python,content
2176,3235034,"genie.py",17063,0,"ch_E).bl",python,content
2177,3235035,"genie.py",17062,0,"ba",python,content
2178,3235040,"genie.py",17055,0,"video_batch_BSHWC, ",python,content
2179,3235044,"genie.py",17045,5,"",python,content
2180,3235045,"genie.py",17039,4,"genie",python,content
2181,3235047,"genie.py",17037,0,"pl",python,content
2182,3235048,"genie.py",17024,11,"_autoreg_s",python,content
2183,3235055,"genie.py",17024,0,"",python,selection_command
2184,3236345,"genie.py",17024,10,"generate_fr",python,content
2185,3236345,"genie.py",17037,2,"",python,content
2186,3236346,"genie.py",17039,5,"self",python,content
2187,3236346,"genie.py",17045,0,"step_",python,content
2188,3236347,"genie.py",17055,19,"",python,content
2189,3236347,"genie.py",17062,2,"",python,content
2190,3236348,"genie.py",17063,8,"",python,content
2191,3236348,"genie.py",17064,1,"",python,content
2192,3236351,"genie.py",17065,0,"ens",python,content
2193,3236361,"genie.py",17069,0,"EL, c",python,content
2194,3236363,"genie.py",17075,0,"rre",python,content
2195,3236365,"genie.py",17080,9,"_token_idxs_BSN, step_t",python,content
2196,3239020,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.20 task 0: running
",,terminal_output
2197,3239172,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.20
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T10:29:07.913] error: *** STEP 14571.20 ON hai005 CANCELLED AT 2025-08-01T10:29:07 DUE to SIGNAL Killed ***
",,terminal_output
2198,3239576,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2199,3240031,"genie.py",17037,0,"",python,selection_command
2200,3240163,"genie.py",17038,0,"",python,selection_command
2201,3240412,"genie.py",17042,0,"",python,selection_command
2202,3240442,"genie.py",17043,0,"",python,selection_command
2203,3240472,"genie.py",17052,0,"",python,selection_command
2204,3240507,"genie.py",17053,0,"",python,selection_command
2205,3240539,"genie.py",17070,0,"",python,selection_command
2206,3240573,"genie.py",17071,0,"",python,selection_command
2207,3240610,"genie.py",17094,0,"",python,selection_command
2208,3240760,"genie.py",17095,0,"",python,selection_command
2209,3240996,"genie.py",17102,0,"",python,selection_command
2210,3241193,"genie.py",17104,0,"",python,selection_command
2211,3241924,"genie.py",17104,0,".",python,content
2212,3241924,"genie.py",17105,0,"",python,selection_keyboard
2213,3243034,"genie.py",17105,0,"block_until_ready()",python,content
2214,3243441,"genie.py",17123,0,"",python,selection_command
2215,3245965,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2216,3246467,"TERMINAL",0,0,"

",,terminal_output
2217,3247190,"genie.py",17000,0,"",python,selection_command
2218,3257761,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output2228,3354138,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ 
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2229,3356221,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2230,3356300,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2231,3369963,"genie.py",17151,0,"",python,selection_mouse
2232,3369974,"genie.py",17150,0,"",python,selection_command
2233,3372741,"experiments/sample.sh",0,0,"",shellscript,tab
2234,3376478,"experiments/sample.sh",248,1,"2",shellscript,content
2235,3378810,"genie.py",0,0,"",python,tab
2236,3380147,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2237,3380448,"TERMINAL",0,0,"

",,terminal_output
2238,3391959,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2239,3398363,"genie.py",17152,0,"",python,selection_command
2240,3399368,"genie.py",17126,0,"",python,selection_command
2241,3399422,"genie.py",17000,0,"",python,selection_command
2242,3399678,"genie.py",16963,0,"",python,selection_command
2243,3399747,"genie.py",16890,0,"",python,selection_command
2244,3399748,"genie.py",16889,0,"",python,selection_command
2245,3399786,"genie.py",16839,0,"",python,selection_command
2246,3399807,"genie.py",16798,0,"",python,selection_command
2247,3399843,"genie.py",16789,0,"",python,selection_command
2248,3399867,"genie.py",16741,0,"",python,selection_command
2249,3399900,"genie.py",16714,0,"",python,selection_command
2250,3399936,"genie.py",16713,0,"",python,selection_command
2251,3399968,"genie.py",16672,0,"",python,selection_command
2252,3400000,"genie.py",16658,0,"",python,selection_command
2253,3400034,"genie.py",16593,0,"",python,selection_command
2254,3400070,"genie.py",16543,0,"",python,selection_command
2255,3400101,"genie.py",16542,0,"",python,selection_command
2256,3400134,"genie.py",16528,0,"",python,selection_command
2257,3400167,"genie.py",16476,0,"",python,selection_command
2258,3400201,"genie.py",16442,0,"",python,selection_command
2259,3400234,"genie.py",16402,0,"",python,selection_command
2260,3400267,"genie.py",16381,0,"",python,selection_command
2261,3400302,"genie.py",16347,0,"",python,selection_command
2262,3400333,"genie.py",16294,0,"",python,selection_command
2263,3400368,"genie.py",16293,0,"",python,selection_command
2264,3400407,"genie.py",16191,0,"",python,selection_command
2265,3400434,"genie.py",16131,0,"",python,selection_command
2266,3400467,"genie.py",16067,0,"",python,selection_command
2267,3400501,"genie.py",15977,0,"",python,selection_command
2268,3400534,"genie.py",15908,0,"",python,selection_command
2269,3400576,"genie.py",15907,0,"",python,selection_command
2270,3400601,"genie.py",15872,0,"",python,selection_command
2271,3400635,"genie.py",15800,0,"",python,selection_command
2272,3400668,"genie.py",15799,0,"",python,selection_command
2273,3400701,"genie.py",15775,0,"",python,selection_command
2274,3400969,"genie.py",15799,0,"",python,selection_command
2275,3401119,"genie.py",15800,0,"",python,selection_command
2276,3401181,"genie.py",15872,0,"",python,selection_command
2277,3401429,"genie.py",15907,0,"",python,selection_command
2278,3401461,"genie.py",15908,0,"",python,selection_command
2279,3401626,"genie.py",15977,0,"",python,selection_command
2280,3401698,"genie.py",16067,0,"",python,selection_command
2281,3401846,"genie.py",16131,0,"",python,selection_command
2282,3402204,"genie.py",16067,0,"",python,selection_command
2283,3402591,"genie.py",16067,63,"            # --- Reset spatial KV caches before each frame ---",python,selection_command
2284,3402887,"genie.py",16067,123,"            # --- Reset spatial KV caches before each frame ---
            for block in genie.dynamics.transformer.blocks:",python,selection_command
2285,3403033,"genie.py",16067,225,"            # --- Reset spatial KV caches before each frame ---
            for block in genie.dynamics.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
2286,3403111,"genie.py",16067,226,"            # --- Reset spatial KV caches before each frame ---
            for block in genie.dynamics.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
",python,selection_command
2287,3403767,"genie.py",16067,227,"",python,content
2288,3403786,"genie.py",16079,0,"",python,selection_command
2289,3404317,"genie.py",16132,0,"",python,selection_command
2290,3404445,"genie.py",16166,0,"",python,selection_command
2291,3404515,"genie.py",16187,0,"",python,selection_command
2292,3404661,"genie.py",16227,0,"",python,selection_command
2293,3405092,"genie.py",16261,0,"",python,selection_command
2294,3405337,"genie.py",16313,0,"",python,selection_command
2295,3405371,"genie.py",16315,0,"",python,selection_command
2296,3405396,"genie.py",16328,0,"",python,selection_command
2297,3405429,"genie.py",16378,0,"",python,selection_command
2298,3405463,"genie.py",16443,0,"",python,selection_command
2299,3405493,"genie.py",16457,0,"",python,selection_command
2300,3405527,"genie.py",16486,0,"",python,selection_command
2301,3405559,"genie.py",16499,0,"",python,selection_command
2302,3405593,"genie.py",16526,0,"",python,selection_command
2303,3405626,"genie.py",16569,0,"",python,selection_command
2304,3405660,"genie.py",16583,0,"",python,selection_command
2305,3405692,"genie.py",16624,0,"",python,selection_command
2306,3405725,"genie.py",16662,0,"",python,selection_command
2307,3405760,"genie.py",16675,0,"",python,selection_command
2308,3408811,"genie.py",16748,0,"",python,selection_command
2309,3409059,"genie.py",16785,0,"",python,selection_command
2310,3409082,"genie.py",16911,0,"",python,selection_command
2311,3409117,"genie.py",16937,0,"",python,selection_command
2312,3409145,"genie.py",16951,0,"",python,selection_command
2313,3409179,"genie.py",17008,0,"",python,selection_command
2314,3409212,"genie.py",17021,0,"",python,selection_command
2315,3409246,"genie.py",17136,0,"",python,selection_command
2316,3409279,"genie.py",17150,0,"",python,selection_command
2317,3409678,"genie.py",17136,0,"",python,selection_command
2318,3409921,"genie.py",17021,0,"",python,selection_command
2319,3409950,"genie.py",17008,0,"",python,selection_command
2320,3409971,"genie.py",16951,0,"",python,selection_command
2321,3410009,"genie.py",16937,0,"",python,selection_command
2322,3410040,"genie.py",16911,0,"",python,selection_command
2323,3410070,"genie.py",16785,0,"",python,selection_command
2324,3410109,"genie.py",16748,0,"",python,selection_command
2325,3410145,"genie.py",16675,0,"",python,selection_command
2326,3410176,"genie.py",16662,0,"",python,selection_command
2327,3410373,"genie.py",16624,0,"",python,selection_command
2328,3410519,"genie.py",16583,0,"",python,selection_command
2329,3411313,"genie.py",16624,0,"",python,selection_command
2330,3411463,"genie.py",16662,0,"",python,selection_command
2331,3411926,"genie.py",16675,0,"",python,selection_command
2332,3412081,"genie.py",16748,0,"",python,selection_command
2333,3412233,"genie.py",16785,0,"",python,selection_command
2334,3412392,"genie.py",16911,0,"",python,selection_command
2335,3412548,"genie.py",16937,0,"",python,selection_command
2336,3412728,"genie.py",16951,0,"",python,selection_command
2337,3412853,"genie.py",17008,0,"",python,selection_command
2338,3413008,"genie.py",17021,0,"",python,selection_command
2339,3413154,"genie.py",17136,0,"",python,selection_command
2340,3413321,"genie.py",17150,0,"",python,selection_command
2341,3413461,"genie.py",17192,0,"",python,selection_command
2342,3413805,"genie.py",17150,0,"",python,selection_command
2343,3414053,"genie.py",17136,0,"",python,selection_command
2344,3414071,"genie.py",17021,0,"",python,selection_command
2345,3414106,"genie.py",17008,0,"",python,selection_command
2346,3414138,"genie.py",16951,0,"",python,selection_command
2347,3414173,"genie.py",16937,0,"",python,selection_command
2348,3414204,"genie.py",16911,0,"",python,selection_command
2349,3414246,"genie.py",16785,0,"",python,selection_command
2350,3414273,"genie.py",16748,0,"",python,selection_command
2351,3414508,"genie.py",16675,0,"",python,selection_command
2352,3414676,"genie.py",16662,0,"",python,selection_command
2353,3414967,"genie.py",16624,0,"",python,selection_command
2354,3415169,"genie.py",16583,0,"",python,selection_command
2355,3420778,"genie.py",16624,0,"",python,selection_command
2356,3421540,"genie.py",16661,0,"
            # --- Reset spatial KV caches before each frame ---
            for block in genie.dynamics.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
",python,content
2357,3421553,"genie.py",16674,0,"",python,selection_command
2358,3423320,"genie.py",16675,0,"",python,selection_command
2359,3424310,"genie.py",16739,0,"",python,selection_command
2360,3425070,"genie.py",16742,0,"",python,selection_command
2361,3425224,"genie.py",16748,0,"",python,selection_command
2362,3425454,"genie.py",16751,0,"",python,selection_command
2363,3435772,"TERMINAL",0,0,"2025-08-01 10:32:24.511673: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2364,3437455,"genie.py",16751,1,"g",python,selection_command
2365,3437511,"genie.py",16751,5,"genie",python,selection_command
2366,3438035,"genie.py",16751,6,"genie.",python,selection_command
2367,3438836,"genie.py",16751,6,"",python,content
2368,3440082,"TERMINAL",0,0,"2025-08-01 10:32:28.826997: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2369,3448847,"TERMINAL",0,0,"2025-08-01 10:32:37.595778: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2370,3452695,"TERMINAL",0,0,"2025-08-01 10:32:41.437317: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2371,3452923,"genie.py",0,0,"",python,selection_command
2372,3454309,"genie.py",889,0,"",python,selection_command
2373,3454867,"genie.py",2248,0,"",python,selection_command
2374,3456787,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
2375,3458251,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2376,3461870,"genie.py",16751,0,"",python,selection_command
2377,3466653,"genie.py",16687,0,"",python,selection_command
2378,3466891,"genie.py",16637,0,"",python,selection_command
2379,3466913,"genie.py",16596,0,"",python,selection_command
2380,3466945,"genie.py",16569,0,"",python,selection_command
2381,3466979,"genie.py",16539,0,"",python,selection_command
2382,3467013,"genie.py",16512,0,"",python,selection_command
2383,3467045,"genie.py",16486,0,"",python,selection_command
2384,3467079,"genie.py",16470,0,"",python,selection_command
2385,3467114,"genie.py",16443,0,"",python,selection_command
2386,3467144,"genie.py",16391,0,"",python,selection_command
2387,3467178,"genie.py",16341,0,"",python,selection_command
2388,3467212,"genie.py",16315,0,"",python,selection_command
2389,3467246,"genie.py",16313,0,"",python,selection_command
2390,3467283,"genie.py",16274,0,"",python,selection_command
2391,3467312,"genie.py",16240,0,"",python,selection_command
2392,3467346,"genie.py",16200,0,"",python,selection_command
2393,3467378,"genie.py",16173,0,"",python,selection_command
2394,3467410,"genie.py",16145,0,"",python,selection_command
2395,3467444,"genie.py",16092,0,"",python,selection_command
2396,3467477,"genie.py",16002,0,"",python,selection_command
2397,3467869,"genie.py",16003,0,"",python,selection_command
2398,3468040,"genie.py",16004,0,"",python,selection_command
2399,3476331,"genie.py",16094,0,"",python,selection_command
2400,3476578,"genie.py",16147,0,"",python,selection_command
2401,3476591,"genie.py",16173,0,"",python,selection_command
2402,3476639,"genie.py",16202,0,"",python,selection_command
2403,3476669,"genie.py",16242,0,"",python,selection_command
2404,3476712,"genie.py",16276,0,"",python,selection_command
2405,3476736,"genie.py",16313,0,"",python,selection_command
2406,3476770,"genie.py",16315,0,"",python,selection_command
2407,3476805,"genie.py",16343,0,"",python,selection_command
2408,3476838,"genie.py",16393,0,"",python,selection_command
2409,3476870,"genie.py",16443,0,"",python,selection_command
2410,3476905,"genie.py",16472,0,"",python,selection_command
2411,3476937,"genie.py",16486,0,"",python,selection_command
2412,3476973,"genie.py",16512,0,"",python,selection_command
2413,3477005,"genie.py",16541,0,"",python,selection_command
2414,3477038,"genie.py",16569,0,"",python,selection_command
2415,3477073,"genie.py",16598,0,"",python,selection_command
2416,3477349,"genie.py",16639,0,"",python,selection_command
2417,3477494,"genie.py",16689,0,"",python,selection_command
2418,3477635,"genie.py",16753,0,"",python,selection_command
2419,3478015,"genie.py",16753,1,"n",python,selection_command
2420,3478072,"genie.py",16753,6,"namics",python,selection_command
2421,3480133,"genie.py",16758,0,"",python,selection_command
2422,3485882,"genie.py",0,0,"",python,selection_command
2423,3486798,"genie.py",2248,0,"",python,selection_command
2424,3487476,"genie.py",5167,0,"",python,selection_command
2425,3488671,"genie.py",2248,0,"",python,selection_command
2426,3493481,"genie.py",16758,0,"",python,selection_command
2427,3494526,"genie.py",16751,0,"",python,selection_command
2428,3495163,"genie.py",16751,0,"s",python,content
2429,3495164,"genie.py",16752,0,"",python,selection_keyboard
2430,3495166,"genie.py",16752,0,"e",python,content
2431,3495166,"genie.py",16753,0,"",python,selection_keyboard
2432,3495292,"genie.py",16753,0,"l",python,content
2433,3495293,"genie.py",16754,0,"",python,selection_keyboard
2434,3495370,"genie.py",16754,0,"f",python,content
2435,3495370,"genie.py",16755,0,"",python,selection_keyboard
2436,3495396,"genie.py",16755,0,".",python,content
2437,3495396,"genie.py",16756,0,"",python,selection_keyboard
2438,3496158,"genie.py",16755,0,"",python,selection_command
2439,3498326,"genie.py",16814,0,"",python,selection_command
2440,3498429,"genie.py",16887,0,"",python,selection_command
2441,3499036,"genie.py",16887,1,"",python,content
2442,3503379,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2443,3503659,"TERMINAL",0,0,"

",,terminal_output
2444,3515086,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2445,3603732,"TERMINAL",0,0,"2025-08-01 10:35:12.477771: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2446,3608083,"TERMINAL",0,0,"2025-08-01 10:35:16.827908: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2447,3616985,"TERMINAL",0,0,"2025-08-01 10:35:25.722952: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2448,3620976,"TERMINAL",0,0,"2025-08-01 10:35:29.713558: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2449,3625025,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
2450,3626606,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2451,3647925,"genie.py",16991,0,"",python,selection_mouse
2452,3651168,"genie.py",17028,0,"",python,selection_command
2453,3651189,"genie.py",17148,0,"",python,selection_command
2454,3651338,"genie.py",17162,0,"",python,selection_command
2455,3651482,"genie.py",17194,0,"",python,selection_command
2456,3654959,"genie.py",17164,68,"            print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2457,3673301,"genie.py",17194,0,"",python,selection_command
2458,3674495,"experiments/sample.sh",0,0,"",shellscript,tab
2459,3677827,"genie.py",0,0,"",python,tab
2460,3678422,"experiments/sample.sh",0,0,"",shellscript,tab
2461,3679024,"experiments/sample.sh",248,1,"3",shellscript,content
2462,3680639,"genie.py",0,0,"",python,tab
2463,3681681,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2464,3681876,"TERMINAL",0,0,"

",,terminal_output
2465,3683571,"experiments/sample.sh",0,0,"",shellscript,tab
2466,3684546,"genie.py",0,0,"",python,tab
2467,3693167,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2468,3702016,"TERMINAL",0,0,"2025-08-01 10:36:50.761707: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2469,3706510,"TERMINAL",0,0,"2025-08-01 10:36:55.249393: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2470,3709504,"genie.py",16361,0,"",python,selection_mouse
2471,3711483,"genie.py",16392,0,"",python,selection_mouse
2472,3715474,"TERMINAL",0,0,"2025-08-01 10:37:04.218733: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2473,3728728,"TERMINAL",0,0,"Autoregressive generation time: 9.8814s
",,terminal_output
2474,3729456,"TERMINAL",0,0,"2025-08-01 10:37:18.204894: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2475,3733566,"genie.py",16152,0,"",python,selection_mouse
2476,3737224,"genie.py",16153,0,"
                ",python,content
2477,3737406,"genie.py",16170,0,"g",python,content
2478,3737406,"genie.py",16171,0,"",python,selection_keyboard
2479,3737488,"genie.py",16171,0,"e",python,content
2480,3737488,"genie.py",16172,0,"",python,selection_keyboard
2481,3737535,"genie.py",16172,0,"n",python,content
2482,3737536,"genie.py",16173,0,"",python,selection_keyboard
2483,3737604,"genie.py",16173,0,"i",python,content
2484,3737605,"genie.py",16174,0,"",python,selection_keyboard
2485,3737705,"genie.py",16174,0,"e",python,content
2486,3737705,"genie.py",16175,0,"",python,selection_keyboard
2487,3737874,"genie.py",16175,0,",",python,content
2488,3737874,"genie.py",16176,0,"",python,selection_keyboard
2489,3738073,"genie.py",16175,0,"",python,selection_command
2490,3738609,"genie.py",16196,0,"",python,selection_command
2491,3738860,"genie.py",16219,0,"",python,selection_command
2492,3738892,"genie.py",16259,0,"",python,selection_command
2493,3738926,"genie.py",16293,0,"",python,selection_command
2494,3738958,"genie.py",16336,0,"",python,selection_command
2495,3739004,"genie.py",16338,0,"",python,selection_command
2496,3739033,"genie.py",16360,0,"",python,selection_command
2497,3739324,"genie.py",16410,0,"",python,selection_command
2498,3741306,"TERMINAL",0,0,"SSIM: 0.005949489772319794
",,terminal_output
2499,3743002,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2500,3747064,"genie.py",14273,0,"",python,selection_mouse
2501,3749297,"genie.py",14273,0," ",python,content
2502,3749298,"genie.py",14274,0,"",python,selection_keyboard
2503,3749666,"genie.py",14273,1,"",python,content
2504,3749801,"genie.py",14273,0,",",python,content
2505,3749801,"genie.py",14274,0,"",python,selection_keyboard
2506,3749999,"genie.py",14274,0," ",python,content
2507,3749999,"genie.py",14275,0,"",python,selection_keyboard
2508,3750356,"genie.py",14274,0,"",python,selection_command
2509,3750826,"genie.py",14273,0,"",python,selection_command
2510,3752290,"genie.py",14273,0,"G",python,content
2511,3752291,"genie.py",14274,0,"",python,selection_keyboard
2512,3752420,"genie.py",14274,0,"e",python,content
2513,3752420,"genie.py",14275,0,"",python,selection_keyboard
2514,3752503,"genie.py",14275,0,"n",python,content
2515,3752504,"genie.py",14276,0,"",python,selection_keyboard
2516,3752585,"genie.py",14276,0,"i",python,content
2517,3752586,"genie.py",14277,0,"",python,selection_keyboard
2518,3752618,"genie.py",14277,0,"e",python,content
2519,3752619,"genie.py",14278,0,"",python,selection_keyboard
2520,3752890,"genie.py",14277,0,"",python,selection_command
2521,3753517,"genie.py",14372,0,"",python,selection_command
2522,3753837,"genie.py",14371,0,"",python,selection_command
2523,3753984,"genie.py",14368,0,"",python,selection_command
2524,3755225,"genie.py",14368,0,"G",python,content
2525,3755225,"genie.py",14369,0,"",python,selection_keyboard
2526,3755390,"genie.py",14369,0,"e",python,content
2527,3755391,"genie.py",14370,0,"",python,selection_keyboard
2528,3755455,"genie.py",14370,0,"n",python,content
2529,3755456,"genie.py",14371,0,"",python,selection_keyboard
2530,3755507,"genie.py",14371,0,"i",python,content
2531,3755507,"genie.py",14372,0,"",python,selection_keyboard
2532,3755605,"genie.py",14372,0,"e",python,content
2533,3755606,"genie.py",14373,0,"",python,selection_keyboard
2534,3755741,"genie.py",14373,0,",",python,content
2535,3755741,"genie.py",14374,0,"",python,selection_keyboard
2536,3755875,"genie.py",14374,0," ",python,content
2537,3755875,"genie.py",14375,0,"",python,selection_keyboard
2538,3756210,"genie.py",14374,0,"",python,selection_command
2539,3757708,"genie.py",14458,0,"",python,selection_command
2540,3758108,"genie.py",14444,0,"",python,selection_command
2541,3758256,"genie.py",14442,0,"",python,selection_command
2542,3759781,"genie.py",14439,0,"",python,selection_command
2543,3760047,"genie.py",14439,0,"g",python,content
2544,3760048,"genie.py",14440,0,"",python,selection_keyboard
2545,3760094,"genie.py",14440,0,"n",python,content
2546,3760094,"genie.py",14441,0,"",python,selection_keyboard
2547,3760095,"genie.py",14441,0,"e",python,content
2548,3760095,"genie.py",14442,0,"",python,selection_keyboard
2549,3760188,"genie.py",14442,0,"i",python,content
2550,3760188,"genie.py",14443,0,"",python,selection_keyboard
2551,3760308,"genie.py",14443,0,"e",python,content
2552,3760309,"genie.py",14444,0,"",python,selection_keyboard
2553,3760424,"genie.py",14444,0,",",python,content
2554,3760424,"genie.py",14445,0,"",python,selection_keyboard
2555,3760617,"genie.py",14445,0," ",python,content
2556,3760617,"genie.py",14446,0,"",python,selection_keyboard
2557,3760878,"genie.py",14445,0,"",python,selection_command
2558,3761775,"genie.py",14444,0,"",python,selection_command
2559,3761907,"genie.py",14439,0,"",python,selection_command
2560,3762259,"genie.py",14439,5,"",python,content
2561,3762595,"genie.py",14439,0,"g",python,content
2562,3762596,"genie.py",14440,0,"",python,selection_keyboard
2563,3762677,"genie.py",14440,0,"e",python,content
2564,3762677,"genie.py",14441,0,"",python,selection_keyboard
2565,3762758,"genie.py",14441,0,"n",python,content
2566,3762759,"genie.py",14442,0,"",python,selection_keyboard
2567,3762842,"genie.py",14442,0,"i",python,content
2568,3762842,"genie.py",14443,0,"",python,selection_keyboard
2569,3762877,"genie.py",14443,0,"e",python,content
2570,3762877,"genie.py",14444,0,"",python,selection_keyboard
2571,3763232,"genie.py",14443,0,"",python,selection_command
2572,3764359,"genie.py",14516,0,"",python,selection_command
2573,3764611,"genie.py",14560,0,"",python,selection_command
2574,3764642,"genie.py",14587,0,"",python,selection_command
2575,3764674,"genie.py",14604,0,"",python,selection_command
2576,3764707,"genie.py",14651,0,"",python,selection_command
2577,3764740,"genie.py",14708,0,"",python,selection_command
2578,3764774,"genie.py",14725,0,"",python,selection_command
2579,3764809,"genie.py",14766,0,"",python,selection_command
2580,3764841,"genie.py",14845,0,"",python,selection_command
2581,3764874,"genie.py",14922,0,"",python,selection_command
2582,3764907,"genie.py",15001,0,"",python,selection_command
2583,3764941,"genie.py",15093,0,"",python,selection_command
2584,3764975,"genie.py",15182,0,"",python,selection_command
2585,3765007,"genie.py",15294,0,"",python,selection_command
2586,3765041,"genie.py",15350,0,"",python,selection_command
2587,3765074,"genie.py",15367,0,"",python,selection_command
2588,3765108,"genie.py",15423,0,"",python,selection_command
2589,3765141,"genie.py",15453,0,"",python,selection_command
2590,3765174,"genie.py",15529,0,"",python,selection_command
2591,3765208,"genie.py",15547,0,"",python,selection_command
2592,3765240,"genie.py",15597,0,"",python,selection_command
2593,3765274,"genie.py",15682,0,"",python,selection_command
2594,3765306,"genie.py",15720,0,"",python,selection_command
2595,3765341,"genie.py",15812,0,"",python,selection_command
2596,3765375,"genie.py",15820,0,"",python,selection_command
2597,3765409,"genie.py",15837,0,"",python,selection_command
2598,3765946,"genie.py",15841,0,"",python,selection_command
2599,3766127,"genie.py",15843,0,"",python,selection_command
2600,3766514,"genie.py",15845,0,"",python,selection_command
2601,3767508,"genie.py",15846,0,"genie, ",python,content
2602,3767508,"genie.py",15853,0,"",python,selection_command
2603,3768229,"genie.py",15820,0,"",python,selection_command
2604,3768472,"genie.py",15818,0,"",python,selection_command
2605,3768504,"genie.py",15736,0,"",python,selection_command
2606,3768539,"genie.py",15698,0,"",python,selection_command
2607,3768571,"genie.py",15613,0,"",python,selection_command
2608,3768609,"genie.py",15563,0,"",python,selection_command
2609,3768643,"genie.py",15529,0,"",python,selection_command
2610,3768676,"genie.py",15469,0,"",python,selection_command
2611,3768712,"genie.py",15435,0,"",python,selection_command
2612,3768738,"genie.py",15383,0,"",python,selection_command
2613,3768772,"genie.py",15350,0,"",python,selection_command
2614,3768805,"genie.py",15310,0,"",python,selection_command
2615,3768838,"genie.py",15198,0,"",python,selection_command
2616,3768877,"genie.py",15109,0,"",python,selection_command
2617,3768911,"genie.py",15017,0,"",python,selection_command
2618,3768944,"genie.py",14938,0,"",python,selection_command
2619,3768979,"genie.py",14861,0,"",python,selection_command
2620,3769012,"genie.py",14782,0,"",python,selection_command
2621,3769045,"genie.py",14741,0,"",python,selection_command
2622,3769078,"genie.py",14708,0,"",python,selection_command
2623,3769335,"genie.py",14667,0,"",python,selection_command
2624,3772354,"genie.py",14672,7,"",python,content
2625,3772354,"genie.py",14664,0,"genie.",python,content
2626,3775092,"genie.py",14866,7,"",python,content
2627,3775093,"genie.py",14858,0,"genie.",python,content
2628,3776910,"genie.py",15206,7,"",python,content
2629,3776911,"genie.py",15198,0,"genie.",python,content
2630,3779935,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2631,3780490,"TERMINAL",0,0,"

",,terminal_output
2632,3791914,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2633,3794521,"genie.py",15932,0,"",python,selection_mouse
2634,3800922,"TERMINAL",0,0,"2025-08-01 10:38:29.670892: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2635,3805434,"TERMINAL",0,0,"2025-08-01 10:38:34.178802: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2636,3814442,"TERMINAL",0,0,"2025-08-01 10:38:43.190697: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2637,3818035,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 410, in sample_causal
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in <lambda>
    lambda: generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 395, in generate_frame
    final_carry_causal, _ = jax.lax.scan(
TypeError: Argument 'Genie( # Param: 95,420,880 (381.7 MB), Cache: 5,677,068 (11.4 MB), Variable: 115,200,000 (460.8 MB), RngState: 2 (12 B), Total: 216,297,950 (853.8 MB)
  in_dim=3,
  tokenizer_dim=512,
  tokenizer_ffn_dim=2048,
  latent_patch_dim=32,
  num_patch_latents=1024,
  patch_size=4,
  tokenizer_num_blocks=4,
  tokenizer_num_heads=8,
  lam_dim=512,
  lam_ffn_dim=2048,
  latent_action_dim=32,
  num_latent_actions=6,
  lam_patch_size=16,
  lam_num_blocks=4,
  lam_num_heads=8,
  lam_co_train=False,
  dyna_type='causal',
  dyna_dim=512,
  dyna_ffn_dim=2048,
  dyna_num_blocks=6,
  dyna_num_heads=8,
  param_dtype=float32,
  dtype=bfloat16,
  use_flash_attention=True,
  dropout=0.0,
  mask_limit=0.0,
  tokenizer=TokenizerVQVAE( # Param: 33,750,256 (135.0 MB), Variable: 40,960,000 (163.8 MB), Total: 74,710,256 (298.8 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=1024,
    patch_size=4,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    encoder=STTransformer( # Param: 16,858,752 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,752 (149.4 MB)
      input_dim=48,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 96 (384 B)
        scale=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        num_features=48,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 25,088 (100.4 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(48, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=48,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36b016feb0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc4c10>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc77f0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc7640>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc76d0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc7880>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc7ac0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3654dc7be0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      )
    ),
    vq=VectorQuantizer( # Param: 32,768 (131.1 KB)
      latent_dim=32,
      num_latents=1024,
      dropout=0.0,
      codebook=Param( # 32,768 (131.1 KB)
        value=Array(shape=(1024, 32), dtype=dtype('float32'))
      ),
      drop=Dropout(
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs( # RngState: 2 (12 B)
          default=RngStream( # RngState: 2 (12 B)
            tag='default',
            key=RngKey( # 1 (8 B)
              value=Array((), dtype=key<fry>) overlaying:
              [ 928981903 3453687069],
              tag=""'default'""
            ),
            count=RngCount( # 1 (4 B)
              value=Array(622, dtype=uint32),
              tag=""'default'""
            )
          )
        )
      )
    ),
    out_dim=48,
    decoder=STTransformer( # Param: 16,858,736 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,736 (149.4 MB)
      input_dim=32,
      model_dim=512,
      ffn_dim=2048,
      out_dim=48,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 64 (256 B)
        scale=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        num_features=32,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 16,896 (67.6 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(32, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=32,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100276d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027760>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027250>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100271c0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axi",,terminal_output
2638,3818121,"TERMINAL",0,0,"s=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027b50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f3610027d00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      )],
      output_dense=Linear( # Param: 24,624 (98.5 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(512, 48), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=48,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      )
    )
  ),
  lam=LatentActionModel( # Param: 35,115,232 (140.5 MB), Variable: 40,960,000 (163.8 MB), RngState: 2 (12 B), Total: 76,075,234 (304.3 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=6,
    patch_size=16,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    patch_token_dim=768,
    encoder=STTransformer( # Param: 17,228,832 (68.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,708,832 (150.8 MB)
      input_dim=768,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,536 (6.1 KB)
        scale=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        num_features=768,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 393,728 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(768, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=768,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36cbb228c0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4a60>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4e50>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4dc0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4670>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e5120>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4f70>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e4ee0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f36d03a5510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f36d03a5510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f36d0279870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      )
    ),
    action_in=Param( # 768 (3.1 KB)
      value=Array(shape=(1, 1, 1, 768), dtype=dtype('float32'))
    ),
    vq=VectorQuantizer( # Param: 192 (768 B), RngState: 2 (12 B), Total: 194 (780 B)
      latent_dim=32,
      num_latents=6,
      dropout=0.0,
      codebook=Param( # 192 (768 B)
        value=Array(shape=(6, 32), dtype=dtype('float32'))
      ),
      drop=Dropout( # RngState: 2 (12 B)
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs(...)
      )
    ),
    patch_up=Linear( # Param: 393,728 (1.6 MB)
      kernel=Param( # 393,216 (1.6 MB)
        value=Array(shape=(768, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=768,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
      bias_init=<function zeros at 0x7f36d03a5510>,
      dot_general=<function dot_general at 0x7f36d0b9a290>,
      promote_dtype=<function promote_dtype at 0x7f36cb277520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
      bias_init=<function zeros at 0x7f36d03a5510>,
      dot_general=<function dot_general at 0x7f36d0b9a290>,
      promote_dtype=<function promote_dtype at 0x7f36cb277520>
    ),
    decoder=STTransformer( # Param: 17,474,816 (69.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,954,816 (151.8 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=768,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e76d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7a30>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
",,terminal_output
2639,3818186,"TERMINAL",0,0,"              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7ac0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7b50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7d00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f36100e7d90>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      )],
      output_dense=Linear( # Param: 393,984 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(512, 768), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=768,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      )
    )
  ),
  dynamics=DynamicsCausal( # Param: 26,555,392 (106.2 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 65,512,460 (250.7 MB)
    model_dim=512,
    ffn_dim=2048,
    num_latents=1024,
    latent_action_dim=32,
    num_blocks=6,
    num_heads=8,
    dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    transformer=Transformer( # Param: 26,014,208 (104.1 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 64,971,276 (248.5 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=1024,
      num_blocks=6,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
        d_model=512,
        max_len=5000,
        pe=Variable( # 2,560,000 (10.2 MB)
          value=Array(shape=(5000, 512), dtype=dtype('float32'))
        )
      ),
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f36d03a5510>,
        scale_init=<function ones at 0x7f36d0279870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8940>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8ca0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8c10>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8d30>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8ee0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a9090>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a9000>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a8f70>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
",,terminal_output
2640,3818253,"TERMINAL",0,0,"          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a91b0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a9480>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f36d03a5510>,
          scale_init=<function ones at 0x7f36d0279870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a9630>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f36d03a5510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f35d01a95a0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
            bias_init=<function zeros at 0x7f36d03a5510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f36cb277520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
          bias_init=<function zeros at 0x7f36d03a5510>,
          dot_general=<function dot_general at 0x7f36d0b9a290>,
          promote_dtype=<function promote_dtype at 0x7f36cb277520>
        )
      )],
      output_dense=Linear( # Param: 525,312 (2.1 MB)
        kernel=Param( # 524,288 (2.1 MB)
          value=Array(shape=(512, 1024), dtype=dtype('float32'))
        ),
        bias=Param( # 1,024 (4.1 KB)
          value=Array(shape=(1024,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=1024,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
        bias_init=<function zeros at 0x7f36d03a5510>,
        dot_general=<function dot_general at 0x7f36d0b9a290>,
        promote_dtype=<function promote_dtype at 0x7f36cb277520>
      )
    ),
    patch_embed=Embed( # Param: 524,288 (2.1 MB)
      embedding=Param( # 524,288 (2.1 MB)
        value=Array(shape=(1024, 512), dtype=dtype('float32'))
      ),
      num_embeddings=1024,
      features=512,
      dtype=dtype('float32'),
      param_dtype=float32,
      embedding_init=<function variance_scaling.<locals>.init at 0x7f36cb277c70>,
      promote_dtype=<function promote_dtype at 0x7f36cb277520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f36cb2779a0>,
      bias_init=<function zeros at 0x7f36d03a5510>,
      dot_general=<function dot_general at 0x7f36d0b9a290>,
      promote_dtype=<function promote_dtype at 0x7f36cb277520>
    )
  )
)' of type '<class 'genie.Genie'>' is not a valid JAX type
",,terminal_output
2641,3819170,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2642,3826365,"genie.py",14276,0,"",python,selection_mouse
2643,3831823,"genie.py",14273,0,"",python,selection_command
2644,3832422,"genie.py",14273,5,"",python,content
2645,3833186,"genie.py",14273,0,"A",python,content
2646,3833186,"genie.py",14274,0,"",python,selection_keyboard
2647,3833304,"genie.py",14274,0,"n",python,content
2648,3833305,"genie.py",14275,0,"",python,selection_keyboard
2649,3833486,"genie.py",14275,0,"y",python,content
2650,3833487,"genie.py",14276,0,"",python,selection_keyboard
2651,3834010,"genie.py",14275,0,"",python,selection_command
2652,3834673,"genie.py",14368,0,"",python,selection_command
2653,3835021,"genie.py",14366,0,"",python,selection_command
2654,3835380,"genie.py",14366,5,"",python,content
2655,3836452,"genie.py",14366,0,"N",python,content
2656,3836452,"genie.py",14367,0,"",python,selection_keyboard
2657,3836741,"genie.py",14367,0,"a",python,content
2658,3836742,"genie.py",14368,0,"",python,selection_keyboard
2659,3836804,"genie.py",14368,0,"y",python,content
2660,3836804,"genie.py",14369,0,"",python,selection_keyboard
2661,3837176,"genie.py",14368,1,"",python,content
2662,3837375,"genie.py",14367,1,"",python,content
2663,3837535,"genie.py",14366,1,"",python,content
2664,3838074,"genie.py",14366,0,"A",python,content
2665,3838074,"genie.py",14367,0,"",python,selection_keyboard
2666,3838224,"genie.py",14367,0,"n",python,content
2667,3838224,"genie.py",14368,0,"",python,selection_keyboard
2668,3838392,"genie.py",14368,0,"y",python,content
2669,3838392,"genie.py",14369,0,"",python,selection_keyboard
2670,3838591,"genie.py",14368,0,"",python,selection_command
2671,3840092,"genie.py",0,0,"",python,selection_command
2672,3841174,"genie.py",23,0,", Any",python,content
2673,3841175,"genie.py",28,0,"",python,selection_command
2674,3841809,"genie.py",27,0,"",python,selection_command
2675,3842226,"genie.py",14278,0,"",python,selection_command
2676,3843963,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2677,3844275,"TERMINAL",0,0,"

",,terminal_output
2678,3853819,"genie.py",14278,3,"",python,content
2679,3854401,"genie.py",14278,0,"n",python,content
2680,3854401,"genie.py",14279,0,"",python,selection_keyboard
2681,3854539,"genie.py",14279,0,"n",python,content
2682,3854540,"genie.py",14280,0,"",python,selection_keyboard
2683,3854603,"genie.py",14280,0,"x",python,content
2684,3854603,"genie.py",14281,0,"",python,selection_keyboard
2685,3855370,"genie.py",14281,0,".",python,content
2686,3855370,"genie.py",14282,0,"",python,selection_keyboard
2687,3855689,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2688,3856241,"genie.py",14282,0,"M",python,content
2689,3856241,"genie.py",14283,0,"",python,selection_keyboard
2690,3856691,"genie.py",14283,0,"o",python,content
2691,3856692,"genie.py",14284,0,"",python,selection_keyboard
2692,3857012,"genie.py",14284,0,"d",python,content
2693,3857012,"genie.py",14285,0,"",python,selection_keyboard
2694,3857046,"genie.py",14285,0,"u",python,content
2695,3857046,"genie.py",14286,0,"",python,selection_keyboard
2696,3857223,"genie.py",14286,0,"l",python,content
2697,3857224,"genie.py",14287,0,"",python,selection_keyboard
2698,3857255,"genie.py",14287,0,"e",python,content
2699,3857255,"genie.py",14288,0,"",python,selection_keyboard
2700,3857848,"genie.py",14287,0,"",python,selection_command
2701,3858577,".venv/lib/python3.10/site-packages/flax/nnx/module.py",0,0,"# Copyright 2024 The Flax Authors.
#
# Licensed under the Apache License, Version 2.0 (the ""License"");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an ""AS IS"" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from __future__ import annotations

import typing as tp

import jax
import jax.numpy as jnp

from flax.nnx import (
  filterlib,
  graph,
)
from flax.nnx import variablelib as variableslib
from flax.nnx.object import Object, ObjectMeta
from flax.nnx.graph import GraphState
from flax.typing import Key, Path, PathParts

A = tp.TypeVar('A')
B = tp.TypeVar('B')
M = tp.TypeVar('M', bound='Module')
S = tp.TypeVar('S', bound=tp.Union[GraphState, tuple[GraphState, ...]])
V = tp.TypeVar('V', bound=variableslib.Variable[tp.Any])
F = tp.TypeVar('F', bound=tp.Callable[..., tp.Any])

StateMapping = tp.Mapping[Path, tp.Any]
tuple_reduce = lambda xs, x: xs + (x,)
tuple_init = lambda: ()


class ModuleMeta(ObjectMeta):
  # we keep a trivial derived class just in case we need to
  # add more functionality in the future
  pass


class Module(Object, metaclass=ModuleMeta):
  """"""Base class for all neural network modules.

  Layers and models should subclass this class.

  ``Module``'s can contain submodules, and in this way can be nested in a tree
  structure. Submodules can be assigned as regular attributes inside the
  ``__init__`` method.

  You can define arbitrary ""forward pass"" methods on your ``Module`` subclass.
  While no methods are special-cased, ``__call__`` is a popular choice since
  you can call the ``Module`` directly::

    >>> from flax import nnx
    >>> import jax.numpy as jnp

    >>> class Model(nnx.Module):
    ...   def __init__(self, rngs):
    ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
    ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
    ...   def __call__(self, x):
    ...     x = self.linear1(x)
    ...     x = nnx.relu(x)
    ...     x = self.linear2(x)
    ...     return x

    >>> x = jnp.ones((1, 2))
    >>> model = Model(rngs=nnx.Rngs(0))
    >>> y = model(x)
  """"""

  def sow(
      self,
      variable_type: type[variableslib.Variable[B]] | str,
      name: str,
      value: A,
      reduce_fn: tp.Callable[[B, A], B] = tuple_reduce,
      init_fn: tp.Callable[[], B] = tuple_init,  # type: ignore
  ) -> bool:
    """"""``sow()`` can be used to collect intermediate values without
    the overhead of explicitly passing a container through each Module call.
    ``sow()`` stores a value in a new ``Module`` attribute, denoted by ``name``.
    The value will be wrapped by a :class:`Variable` of type ``variable_type``,
    which can be useful to filter for in :func:`split`, :func:`state` and
    :func:`pop`.

    By default the values are stored in a tuple and each stored value
    is appended at the end. This way all intermediates can be tracked when
    the same module is called multiple times.

    Example usage::

      >>> from flax import nnx
      >>> import jax.numpy as jnp

      >>> class Model(nnx.Module):
      ...   def __init__(self, rngs):
      ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
      ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
      ...   def __call__(self, x, add=0):
      ...     x = self.linear1(x)
      ...     self.sow(nnx.Intermediate, 'i', x+add)
      ...     x = self.linear2(x)
      ...     return x

      >>> x = jnp.ones((1, 2))
      >>> model = Model(rngs=nnx.Rngs(0))
      >>> assert not hasattr(model, 'i')

      >>> y = model(x)
      >>> assert hasattr(model, 'i')
      >>> assert len(model.i.value) == 1 # tuple of length 1
      >>> assert model.i.value[0].shape == (1, 3)

      >>> y = model(x, add=1)
      >>> assert len(model.i.value) == 2 # tuple of length 2
      >>> assert (model.i.value[0] + 1 == model.i.value[1]).all()

    Alternatively, a custom init/reduce function can be passed::

      >>> class Model(nnx.Module):
      ...   def __init__(self, rngs):
      ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
      ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
      ...   def __call__(self, x):
      ...     x = self.linear1(x)
      ...     self.sow(nnx.Intermediate, 'sum', x,
      ...              init_fn=lambda: 0,
      ...              reduce_fn=lambda prev, curr: prev+curr)
      ...     self.sow(nnx.Intermediate, 'product', x,
      ...              init_fn=lambda: 1,
      ...              reduce_fn=lambda prev, curr: prev*curr)
      ...     x = self.linear2(x)
      ...     return x

      >>> x = jnp.ones((1, 2))
      >>> model = Model(rngs=nnx.Rngs(0))

      >>> y = model(x)
      >>> assert (model.sum.value == model.product.value).all()
      >>> intermediate = model.sum.value

      >>> y = model(x)
      >>> assert (model.sum.value == intermediate*2).all()
      >>> assert (model.product.value == intermediate**2).all()

    Args:
      variable_type: The :class:`Variable` type for the stored value.
        Typically :class:`Intermediate` is used to indicate an
        intermediate value.
      name: A string denoting the ``Module`` attribute name, where
        the sowed value is stored.
      value: The value to be stored.
      reduce_fn: The function used to combine the existing value with the new
        value. The default is to append the value to a tuple.
      init_fn: For the first value stored, ``reduce_fn`` will be passed the result
        of ``init_fn`` together with the value to be stored. The default is an
        empty tuple.
    """"""
    if isinstance(variable_type, str):
      variable_type = variableslib.variable_type_from_name(
          variable_type, allow_register=True
      )

    if hasattr(self, name):
      variable = getattr(self, name)
      if not isinstance(variable, variableslib.Variable):
        raise ValueError(
          f""Expected '{name}' to be a Variable, got {type(variable).__name__}""
        )
      elif type(variable) != variable_type:
        raise ValueError(
          f""Expected '{name}' to be of type '{variable_type.__name__}', ""
          f""got '{type(variable).__name__}'""
        )
      variable.raw_value = reduce_fn(variable.raw_value, value)
    else:
      reduced_value = reduce_fn(init_fn(), value)
      setattr(self, name, variable_type(reduced_value))

    return True

  def perturb(
      self,
      name: str,
      value: tp.Any,
      variable_type: (
          str | type[variableslib.Variable[tp.Any]]
      ) = variableslib.Perturbation,
  ):
    """"""Add an zero-value variable (""perturbation"") to the intermediate value.

    The gradient of ``value`` would be the same as the gradient of this
    perturbation variable. Therefore, if you define your loss function with
    both params and perturbations as standalone arguments, you can get the
    intermediate gradients of ``value`` by running ``jax.grad`` on the
    perturbation variable.

    Since the shape of the perturbation value depends on the shape of the input,
    a perturbation variable is only created after you run a sample input through
    the model once.

    .. note::
      This creates extra dummy variables of the same size as ``value``, thus
      occupies more memory. Use it only to debug gradients in training.

    Example usage::

      >>> from flax import nnx
      >>> import jax.numpy as jnp

      >>> class Model(nnx.Module):
      ...   def __init__(self, rngs):
      ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
      ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
      ...   def __call__(self, x):
      ...     x = self.linear1(x)
      ...     x = self.perturb('xgrad', x)
      ...     x = self.linear2(x)
      ...     return x

      >>> x = jnp.ones((1, 2))
      >>> y = jnp.ones((1, 4))
      >>> model = Model(rngs=nnx.Rngs(0))
      >>> assert not hasattr(model, 'xgrad')  # perturbation requires a sample input run
      >>> _ = model(x)
      >>> assert model.xgrad.value.shape == (1, 3)   # same as the intermediate value

      >>> # Take gradients on the Param and Perturbation variables
      >>> @nnx.grad(argnums=nnx.DiffState(argnum=0, filter=nnx.Any(nnx.Param, nnx.Perturbation)))
      ... def grad_loss(model, inputs, targets):
      ...   preds = model(inputs)
      ...   return jnp.square(preds - targets).mean()

      >>> intm_grads = grad_loss(model, x, y)
      >>> # `intm_grads.xgrad.value` is the intermediate gradient
      >>> assert not jnp.array_equal(intm_grads.xgrad.value, jnp.zeros((1, 3)))

    Args:
      name: A string denoting the ``Module`` attribute name for the
        perturbation value.
      value: The value to take intermediate gradient.
      variable_type: The :class:`Variable` type for the stored perturbation.
        Defaulted at :class:`nnx.Perturbation`.
    """"""
    if isinstance(variable_type, str):
      variable_type = variableslib.variable_type_from_name(
          variable_type, allow_register=True
      )
    if not hasattr(self, name):
      zeros = jax.tree.map(jnp.zeros_like, value)
      setattr(self, name, variable_type(zeros))
    old_value: variableslib.Variable[tp.Any] = getattr(self, name)
    if not isinstance(old_value, variable_type):
      raise ValueError(
        f""Expected '{name}' to be of type '{variable_type.__name__}', ""
        f""got '{type(old_value).__name__}'""
      )
    return old_value.value + value

  def iter_modules(self) -> tp.Iterator[tuple[PathParts, Module]]:
    """"""Recursively iterates over all nested :class:`Module`'s of the current Module, including
    the current Module.

    ``iter_modules`` creates a generator that yields the path and the Module instance, where
    the path is a tuple of strings or integers representing the path to the Module from the
    root Module.

    Example::

      >>> from flax import nnx
      ...
      >>> class SubModule(nnx.Module):
      ...   def __init__(self, din, dout, rngs):
      ...     self.linear1 = nnx.Linear(din, dout, rngs=rngs)
      ...     self.linear2 = nnx.Linear(din, dout, rngs=rngs)
      ...
      >>> class Block(nnx.Module):
      ...   def __init__(self, din, dout, *, rngs: nnx.Rngs):
      ...     self.linear = nnx.Linear(din, dout, rngs=rngs)
      ...     self.submodule = SubModule(din, dout, rngs=rngs)
      ...     self.dropout = nnx.Dropout(0.5)
      ...     self.batch_norm = nnx.BatchNorm(10, rngs=rngs)
      ...
      >>> model = Block(2, 5, rngs=nnx.Rngs(0))
      >>> for path, module in model.iter_modules():
      ...   print(path, type(module).__name__)
      ...
      ('batch_norm',) BatchNorm
      ('dropout',) Dropout
      ('linear',) Linear
      ('submodule', 'linear1') Linear
      ('submodule', 'linear2') Linear
      ('submodule',) SubModule
      () Block
    """"""
    for path, value in graph.iter_graph(self):
      if isinstance(value, Module):
        yield path, value

  def iter_children(self) -> tp.Iterator[tuple[Key, Module]]:
    """"""Iterates over all children :class:`Module`'s of the current Module. This
    method is similar to :func:`iter_modules`, except it only iterates over the
    immediate children, and does not recurse further down.

    ``iter_children`` creates a generator that yields the key and the Module instance,
    where the key is a string representing the attribute name of the Module to access
    the corresponding child Module.

    Example::

      >>> from flax import nnx
      ...
      >>> class SubModule(nnx.Module):
      ...   def __init__(self, din, dout, rngs):
      ...     self.linear1 = nnx.Linear(din, dout, rngs=rngs)
      ...     self.linear2 = nnx.Linear(din, dout, rngs=rngs)
      ...
      >>> class Block(nnx.Module):
      ...   def __init__(self, din, dout, *, rngs: nnx.Rngs):
      ...     self.linear = nnx.Linear(din, dout, rngs=rngs)
      ...     self.submodule = SubModule(din, dout, rngs=rngs)
      ...     self.dropout = nnx.Dropout(0.5)
      ...     self.batch_norm = nnx.BatchNorm(10, rngs=rngs)
      ...
      >>> model = Block(2, 5, rngs=nnx.Rngs(0))
      >>> for path, module in model.iter_children():
      ...  print(path, type(module).__name__)
      ...
      batch_norm BatchNorm
      dropout Dropout
      linear Linear
      submodule SubModule
    """"""
    node_impl = graph.get_node_impl(self)
    assert node_impl is not None
    node_dict = node_impl.node_dict(self)
    for key, value in node_dict.items():
      if isinstance(value, Module):
        yield key, value

  def set_attributes(
    self,
    *filters: filterlib.Filter,
    raise_if_not_found: bool = True,
    **attributes: tp.Any,
  ) -> None:
    """"""Sets the attributes of nested Modules including the current Module.
    If the attribute is not found in the Module, it is ignored.

    Example::

      >>> from flax import nnx
      ...
      >>> class Block(nnx.Module):
      ...   def __init__(self, din, dout, *, rngs: nnx.Rngs):
      ...     self.linear = nnx.Linear(din, dout, rngs=rngs)
      ...     self.dropout = nnx.Dropout(0.5, deterministic=False)
      ...     self.batch_norm = nnx.BatchNorm(10, use_running_average=False, rngs=rngs)
      ...
      >>> block = Block(2, 5, rngs=nnx.Rngs(0))
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (False, False)
      >>> block.set_attributes(deterministic=True, use_running_average=True)
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (True, True)

    ``Filter``'s can be used to set the attributes of specific Modules::

      >>> block = Block(2, 5, rngs=nnx.Rngs(0))
      >>> block.set_attributes(nnx.Dropout, deterministic=True)
      >>> # Only the dropout will be modified
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (True, False)

    Args:
      *filters: Filters to select the Modules to set the attributes of.
      raise_if_not_found: If True (default), raises a ValueError if at least one attribute
        instance is not found in one of the selected Modules.
      **attributes: The attributes to set.
    """"""
    remaining_attributes = set(attributes.keys())
    if not filters:
      filters = (True,)
    predicates = tuple(map(filterlib.to_predicate, filters))
    for path, module in self.iter_modules():
      for predicate in predicates:
        if predicate(path, module):
          for name, value in attributes.items():
            if hasattr(module, name):
              if name in remaining_attributes:
                remaining_attributes.remove(name)
              setattr(module, name, value)
          break

    if remaining_attributes and raise_if_not_found:
      raise ValueError(
        f'Could not find at least one instance of the following attributes: {remaining_attributes}'
      )

  def train(self, **attributes):
    """"""Sets the Module to training mode.

    ``train`` uses ``set_attributes`` to recursively set attributes ``deterministic=False``
    and ``use_running_average=False`` of all nested Modules that have these attributes.
    Its primarily used to control the runtime behavior of the ``Dropout`` and ``BatchNorm``
    Modules.

    Example::

      >>> from flax import nnx
      ...
      >>> class Block(nnx.Module):
      ...   def __init__(self, din, dout, *, rngs: nnx.Rngs):
      ...     self.linear = nnx.Linear(din, dout, rngs=rngs)
      ...     # initialize Dropout and BatchNorm in eval mode
      ...     self.dropout = nnx.Dropout(0.5, deterministic=True)
      ...     self.batch_norm = nnx.BatchNorm(10, use_running_average=True, rngs=rngs)
      ...
      >>> block = Block(2, 5, rngs=nnx.Rngs(0))
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (True, True)
      >>> block.train()
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (False, False)

    Args:
      **attributes: additional attributes passed to ``set_attributes``.
    """"""
    return self.set_attributes(
      deterministic=False,
      use_running_average=False,
      **attributes,
      raise_if_not_found=False,
    )

  def eval(self, **attributes):
    """"""Sets the Module to evaluation mode.

    ``eval`` uses ``set_attributes`` to recursively set attributes ``deterministic=True``
    and ``use_running_average=True`` of all nested Modules that have these attributes.
    Its primarily used to control the runtime behavior of the ``Dropout`` and ``BatchNorm``
    Modules.

    Example::

      >>> from flax import nnx
      ...
      >>> class Block(nnx.Module):
      ...   def __init__(self, din, dout, *, rngs: nnx.Rngs):
      ...     self.linear = nnx.Linear(din, dout, rngs=rngs)
      ...     self.dropout = nnx.Dropout(0.5)
      ...     self.batch_norm = nnx.BatchNorm(10, rngs=rngs)
      ...
      >>> block = Block(2, 5, rngs=nnx.Rngs(0))
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (False, False)
      >>> block.eval()
      >>> block.dropout.deterministic, block.batch_norm.use_running_average
      (True, True)

    Args:
      **attributes: additional attributes passed to ``set_attributes``.
    """"""
    return self.set_attributes(
      deterministic=True,
      use_running_average=True,
      **attributes,
      raise_if_not_found=False,
    )


def first_from(*args: tp.Optional[A], error_msg: str) -> A:
  """"""Return the first non-None argument.

  If all arguments are None, raise a ValueError with the given error message.

  Args:
    *args: the arguments to check
    error_msg: the error message to raise if all arguments are None
  Returns:
    The first non-None argument.
  """"""
  for arg in args:
    if arg is not None:
      return arg
  raise ValueError(error_msg)
",python,tab
2702,3858577,".venv/lib/python3.10/site-packages/flax/nnx/module.py",1411,0,"",python,selection_command
2703,3863705,"genie.py",0,0,"",python,tab
2704,3863706,"genie.py",14287,0,"",python,selection_command
2705,3864637,"TERMINAL",0,0,"2025-08-01 10:39:33.383421: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2706,3865596,"genie.py",14288,0,"",python,selection_command
2707,3867823,"genie.py",14278,10,"Any",python,content
2708,3867826,"genie.py",14278,0,"",python,selection_command
2709,3869154,"TERMINAL",0,0,"2025-08-01 10:39:37.903448: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2710,3869433,"genie.py",14371,0,"",python,selection_command
2711,3869480,"genie.py",14453,0,"",python,selection_command
2712,3869628,"genie.py",14526,0,"",python,selection_command
2713,3869764,"genie.py",14570,0,"",python,selection_command
2714,3869896,"genie.py",14588,0,"",python,selection_command
2715,3878341,"TERMINAL",0,0,"2025-08-01 10:39:47.089075: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2716,3881988,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 410, in sample_causal
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in <lambda>
    lambda: generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 395, in generate_frame
    final_carry_causal, _ = jax.lax.scan(
TypeError: Argument 'Genie( # Param: 95,420,880 (381.7 MB), Cache: 5,677,068 (11.4 MB), Variable: 115,200,000 (460.8 MB), RngState: 2 (12 B), Total: 216,297,950 (853.8 MB)
  in_dim=3,
  tokenizer_dim=512,
  tokenizer_ffn_dim=2048,
  latent_patch_dim=32,
  num_patch_latents=1024,
  patch_size=4,
  tokenizer_num_blocks=4,
  tokenizer_num_heads=8,
  lam_dim=512,
  lam_ffn_dim=2048,
  latent_action_dim=32,
  num_latent_actions=6,
  lam_patch_size=16,
  lam_num_blocks=4,
  lam_num_heads=8,
  lam_co_train=False,
  dyna_type='causal',
  dyna_dim=512,
  dyna_ffn_dim=2048,
  dyna_num_blocks=6,
  dyna_num_heads=8,
  param_dtype=float32,
  dtype=bfloat16,
  use_flash_attention=True,
  dropout=0.0,
  mask_limit=0.0,
  tokenizer=TokenizerVQVAE( # Param: 33,750,256 (135.0 MB), Variable: 40,960,000 (163.8 MB), Total: 74,710,256 (298.8 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=1024,
    patch_size=4,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    encoder=STTransformer( # Param: 16,858,752 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,752 (149.4 MB)
      input_dim=48,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 96 (384 B)
        scale=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        num_features=48,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 25,088 (100.4 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(48, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=48,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fee7cde7eb0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80acc10>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80af7f0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80af640>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80af6d0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80af880>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80afac0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7fedf80afbe0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      )
    ),
    vq=VectorQuantizer( # Param: 32,768 (131.1 KB)
      latent_dim=32,
      num_latents=1024,
      dropout=0.0,
      codebook=Param( # 32,768 (131.1 KB)
        value=Array(shape=(1024, 32), dtype=dtype('float32'))
      ),
      drop=Dropout(
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs( # RngState: 2 (12 B)
          default=RngStream( # RngState: 2 (12 B)
            tag='default',
            key=RngKey( # 1 (8 B)
              value=Array((), dtype=key<fry>) overlaying:
              [ 928981903 3453687069],
              tag=""'default'""
            ),
            count=RngCount( # 1 (4 B)
              value=Array(622, dtype=uint32),
              tag=""'default'""
            )
          )
        )
      )
    ),
    out_dim=48,
    decoder=STTransformer( # Param: 16,858,736 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,736 (149.4 MB)
      input_dim=32,
      model_dim=512,
      ffn_dim=2048,
      out_dim=48,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 64 (256 B)
        scale=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        num_features=32,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 16,896 (67.6 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(32, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=32,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c36d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3760>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3250>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c31c0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
",,terminal_output
2717,3882059,"TERMINAL",0,0,"            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3b50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc0c3d00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      )],
      output_dense=Linear( # Param: 24,624 (98.5 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(512, 48), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=48,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      )
    )
  ),
  lam=LatentActionModel( # Param: 35,115,232 (140.5 MB), Variable: 40,960,000 (163.8 MB), RngState: 2 (12 B), Total: 76,075,234 (304.3 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=6,
    patch_size=16,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    patch_token_dim=768,
    encoder=STTransformer( # Param: 17,228,832 (68.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,708,832 (150.8 MB)
      input_dim=768,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,536 (6.1 KB)
        scale=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        num_features=768,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 393,728 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(768, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=768,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7feef409e8c0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178a60>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178e50>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178dc0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178670>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc179120>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178f70>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc178ee0>,
          bias_init=<function zeros at 0x7feef8921510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7feef8921510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7feef8921510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7feef87f5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      )
    ),
    action_in=Param( # 768 (3.1 KB)
      value=Array(shape=(1, 1, 1, 768), dtype=dtype('float32'))
    ),
    vq=VectorQuantizer( # Param: 192 (768 B), RngState: 2 (12 B), Total: 194 (780 B)
      latent_dim=32,
      num_latents=6,
      dropout=0.0,
      codebook=Param( # 192 (768 B)
        value=Array(shape=(6, 32), dtype=dtype('float32'))
      ),
      drop=Dropout( # RngState: 2 (12 B)
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs(...)
      )
    ),
    patch_up=Linear( # Param: 393,728 (1.6 MB)
      kernel=Param( # 393,216 (1.6 MB)
        value=Array(shape=(768, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=768,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
      bias_init=<function zeros at 0x7feef8921510>,
      dot_general=<function dot_general at 0x7feef911a290>,
      promote_dtype=<function promote_dtype at 0x7feef37f3520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
      bias_init=<function zeros at 0x7feef8921510>,
      dot_general=<function dot_general at 0x7feef911a290>,
      promote_dtype=<function promote_dtype at 0x7feef37f3520>
    ),
    decoder=STTransformer( # Param: 17,474,816 (69.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,954,816 (151.8 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=768,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17b6d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17ba30>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17b910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
",,terminal_output
2718,3882127,"TERMINAL",0,0,"          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17b880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17bac0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17bb50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17bd00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febdc17bd90>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      )],
      output_dense=Linear( # Param: 393,984 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(512, 768), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=768,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      )
    )
  ),
  dynamics=DynamicsCausal( # Param: 26,555,392 (106.2 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 65,512,460 (250.7 MB)
    model_dim=512,
    ffn_dim=2048,
    num_latents=1024,
    latent_action_dim=32,
    num_blocks=6,
    num_heads=8,
    dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    transformer=Transformer( # Param: 26,014,208 (104.1 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 64,971,276 (248.5 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=1024,
      num_blocks=6,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
        d_model=512,
        max_len=5000,
        pe=Variable( # 2,560,000 (10.2 MB)
          value=Array(shape=(5000, 512), dtype=dtype('float32'))
        )
      ),
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7feef8921510>,
        scale_init=<function ones at 0x7feef87f5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fec940>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fecca0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fecc10>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fecd30>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fecee0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed090>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed000>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fecf70>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
",,terminal_output
2719,3882177,"TERMINAL",0,0,"            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed1b0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed480>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7feef8921510>,
          scale_init=<function ones at 0x7feef87f5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed630>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7feef8921510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7febd5fed5a0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
            bias_init=<function zeros at 0x7feef8921510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7feef37f3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
          bias_init=<function zeros at 0x7feef8921510>,
          dot_general=<function dot_general at 0x7feef911a290>,
          promote_dtype=<function promote_dtype at 0x7feef37f3520>
        )
      )],
      output_dense=Linear( # Param: 525,312 (2.1 MB)
        kernel=Param( # 524,288 (2.1 MB)
          value=Array(shape=(512, 1024), dtype=dtype('float32'))
        ),
        bias=Param( # 1,024 (4.1 KB)
          value=Array(shape=(1024,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=1024,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
        bias_init=<function zeros at 0x7feef8921510>,
        dot_general=<function dot_general at 0x7feef911a290>,
        promote_dtype=<function promote_dtype at 0x7feef37f3520>
      )
    ),
    patch_embed=Embed( # Param: 524,288 (2.1 MB)
      embedding=Param( # 524,288 (2.1 MB)
        value=Array(shape=(1024, 512), dtype=dtype('float32'))
      ),
      num_embeddings=1024,
      features=512,
      dtype=dtype('float32'),
      param_dtype=float32,
      embedding_init=<function variance_scaling.<locals>.init at 0x7feef37f3c70>,
      promote_dtype=<function promote_dtype at 0x7feef37f3520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7feef37f39a0>,
      bias_init=<function zeros at 0x7feef8921510>,
      dot_general=<function dot_general at 0x7feef911a290>,
      promote_dtype=<function promote_dtype at 0x7feef37f3520>
    )
  )
)' of type '<class 'genie.Genie'>' is not a valid JAX type
",,terminal_output
2720,3883081,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2721,3889137,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
2722,3889461,"TERMINAL",0,0,"

",,terminal_output
2723,3891539,"genie.py",16377,0,"",python,selection_command
2724,3896057,"genie.py",16196,0,"",python,selection_command
2725,3896536,"genie.py",0,0,"",python,selection_command
2726,3896693,"genie.py",18273,0,"",python,selection_command
2727,3898571,"genie.py",0,0,"",python,selection_command
2728,3899559,"genie.py",300,0,"",python,selection_command
2729,3899740,"genie.py",326,0,"",python,selection_command
2730,3900139,"genie.py",14440,0,"",python,selection_command
2731,3900588,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
2732,3902443,"genie.py",14665,0,"",python,selection_command
2733,3902824,"genie.py",14859,0,"",python,selection_command
2734,3903189,"genie.py",15199,0,"",python,selection_command
2735,3903435,"genie.py",15844,0,"",python,selection_command
2736,3903795,"genie.py",16030,0,"",python,selection_command
2737,3905133,"genie.py",16196,0,"",python,selection_command
2738,3909518,"TERMINAL",0,0,"2025-08-01 10:40:18.250629: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2739,3914039,"TERMINAL",0,0,"2025-08-01 10:40:22.789691: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2740,3923022,"TERMINAL",0,0,"2025-08-01 10:40:31.770343: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
2741,3926607,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 410, in sample_causal
    elapsed = timeit.timeit(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 234, in timeit
    return Timer(stmt, setup, timer, globals).timeit(number)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/timeit.py"", line 178, in timeit
    timing = self.inner(it, self.timer)
  File ""<timeit-src>"", line 6, in inner
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in <lambda>
    lambda: generate_frame(self, step_rng, action_tokens_EL, current_token_idxs_BSN, step_t).block_until_ready(),
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 395, in generate_frame
    final_carry_causal, _ = jax.lax.scan(
TypeError: Argument 'Genie( # Param: 95,420,880 (381.7 MB), Cache: 5,677,068 (11.4 MB), Variable: 115,200,000 (460.8 MB), RngState: 2 (12 B), Total: 216,297,950 (853.8 MB)
  in_dim=3,
  tokenizer_dim=512,
  tokenizer_ffn_dim=2048,
  latent_patch_dim=32,
  num_patch_latents=1024,
  patch_size=4,
  tokenizer_num_blocks=4,
  tokenizer_num_heads=8,
  lam_dim=512,
  lam_ffn_dim=2048,
  latent_action_dim=32,
  num_latent_actions=6,
  lam_patch_size=16,
  lam_num_blocks=4,
  lam_num_heads=8,
  lam_co_train=False,
  dyna_type='causal',
  dyna_dim=512,
  dyna_ffn_dim=2048,
  dyna_num_blocks=6,
  dyna_num_heads=8,
  param_dtype=float32,
  dtype=bfloat16,
  use_flash_attention=True,
  dropout=0.0,
  mask_limit=0.0,
  tokenizer=TokenizerVQVAE( # Param: 33,750,256 (135.0 MB), Variable: 40,960,000 (163.8 MB), Total: 74,710,256 (298.8 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=1024,
    patch_size=4,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    encoder=STTransformer( # Param: 16,858,752 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,752 (149.4 MB)
      input_dim=48,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 96 (384 B)
        scale=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        num_features=48,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 25,088 (100.4 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(48, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=48,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8e1013feb0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbcc10>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbf7f0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbf640>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbf6d0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbf880>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbfac0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0dbfbe0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      )
    ),
    vq=VectorQuantizer( # Param: 32,768 (131.1 KB)
      latent_dim=32,
      num_latents=1024,
      dropout=0.0,
      codebook=Param( # 32,768 (131.1 KB)
        value=Array(shape=(1024, 32), dtype=dtype('float32'))
      ),
      drop=Dropout(
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs( # RngState: 2 (12 B)
          default=RngStream( # RngState: 2 (12 B)
            tag='default',
            key=RngKey( # 1 (8 B)
              value=Array((), dtype=key<fry>) overlaying:
              [ 928981903 3453687069],
              tag=""'default'""
            ),
            count=RngCount( # 1 (4 B)
              value=Array(622, dtype=uint32),
              tag=""'default'""
            )
          )
        )
      )
    ),
    out_dim=48,
    decoder=STTransformer( # Param: 16,858,736 (67.4 MB), Variable: 20,480,000 (81.9 MB), Total: 37,338,736 (149.4 MB)
      input_dim=32,
      model_dim=512,
      ffn_dim=2048,
      out_dim=48,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 64 (256 B)
        scale=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        num_features=32,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 16,896 (67.6 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(32, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=32,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e636d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63760>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63250>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e631c0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_gro",,terminal_output
2742,3926665,"TERMINAL",0,0,"ups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63b50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8dc0e63d00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      )],
      output_dense=Linear( # Param: 24,624 (98.5 KB)
        kernel=Param( # 24,576 (98.3 KB)
          value=Array(shape=(512, 48), dtype=dtype('float32'))
        ),
        bias=Param( # 48 (192 B)
          value=Array(shape=(48,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=48,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      )
    )
  ),
  lam=LatentActionModel( # Param: 35,115,232 (140.5 MB), Variable: 40,960,000 (163.8 MB), RngState: 2 (12 B), Total: 76,075,234 (304.3 MB)
    in_dim=3,
    model_dim=512,
    ffn_dim=2048,
    latent_dim=32,
    num_latents=6,
    patch_size=16,
    num_blocks=4,
    num_heads=8,
    dropout=0.0,
    codebook_dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    patch_token_dim=768,
    encoder=STTransformer( # Param: 17,228,832 (68.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,708,832 (150.8 MB)
      input_dim=768,
      model_dim=512,
      ffn_dim=2048,
      out_dim=32,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,536 (6.1 KB)
        scale=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        num_features=768,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 393,728 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(768, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=768,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8e3ef5e8c0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817ca60>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817ce50>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817cdc0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817c670>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817d120>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        dropout=0.0,
        dtype=bfloat16,
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=512,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=2048,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          dtype=bfloat16,
          in_features=2048,
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_features=512,
          param_dtype=float32,
          precision=None,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
          use_bias=True
        ),
        ffn_dim=2048,
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        num_heads=8,
        param_dtype=float32,
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817cf70>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817cee0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          broadcast_dropout=True,
          cache_index=None,
          cached_key=None,
          cached_value=None,
          decode=False,
          deterministic=None,
          dropout_rate=0.0,
          dtype=bfloat16,
          head_dim=64,
          in_features=512,
          in_kv_features=512,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          key_ln=None,
          normalize_qk=False,
          num_heads=8,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(8, 64),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(512,),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          out_bias_init=None,
          out_dot_general=None,
          out_dot_general_cls=None,
          out_features=512,
          out_kernel_init=None,
          param_dtype=float32,
          precision=None,
          qkv_dot_general=None,
          qkv_dot_general_cls=None,
          qkv_features=512,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          ),
          query_ln=None,
          rngs=None,
          use_bias=True,
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            axis=(-1,),
            batch_axis=FrozenDict({}),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            ),
            bias_init=<function zeros at 0x7f8e437d1510>,
            dot_general=None,
            dot_general_cls=None,
            dtype=bfloat16,
            in_features=(512,),
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            out_features=(8, 64),
            param_dtype=float32,
            precision=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            use_bias=True
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          axis_index_groups=None,
          axis_name=None,
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias_init=<function zeros at 0x7f8e437d1510>,
          dtype=bfloat16,
          epsilon=1e-06,
          feature_axes=-1,
          num_features=512,
          param_dtype=float32,
          reduction_axes=-1,
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          scale_init=<function ones at 0x7f8e436a5870>,
          use_bias=True,
          use_fast_variance=True,
          use_scale=True
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        use_flash_attention=True
      )],
      output_dense=Linear( # Param: 16,416 (65.7 KB)
        kernel=Param( # 16,384 (65.5 KB)
          value=Array(shape=(512, 32), dtype=dtype('float32'))
        ),
        bias=Param( # 32 (128 B)
          value=Array(shape=(32,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=32,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      )
    ),
    action_in=Param( # 768 (3.1 KB)
      value=Array(shape=(1, 1, 1, 768), dtype=dtype('float32'))
    ),
    vq=VectorQuantizer( # Param: 192 (768 B), RngState: 2 (12 B), Total: 194 (780 B)
      latent_dim=32,
      num_latents=6,
      dropout=0.0,
      codebook=Param( # 192 (768 B)
        value=Array(shape=(6, 32), dtype=dtype('float32'))
      ),
      drop=Dropout( # RngState: 2 (12 B)
        rate=0.0,
        broadcast_dims=(),
        deterministic=False,
        rng_collection='dropout',
        rngs=Rngs(...)
      )
    ),
    patch_up=Linear( # Param: 393,728 (1.6 MB)
      kernel=Param( # 393,216 (1.6 MB)
        value=Array(shape=(768, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=768,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
      bias_init=<function zeros at 0x7f8e437d1510>,
      dot_general=<function dot_general at 0x7f8e43fda290>,
      promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
      bias_init=<function zeros at 0x7f8e437d1510>,
      dot_general=<function dot_general at 0x7f8e43fda290>,
      promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
    ),
    decoder=STTransformer( # Param: 17,474,816 (69.9 MB), Variable: 20,480,000 (81.9 MB), Total: 37,954,816 (151.8 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=768,
      num_blocks=4,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817f6d0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817fa30>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817f910>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,",,terminal_output
2743,3926725,"TERMINAL",0,0,"
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817f880>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817fac0>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817fb50>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), STBlock( # Param: 4,204,032 (16.8 MB), Variable: 5,120,000 (20.5 MB), Total: 9,324,032 (37.3 MB)
        dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817fd00>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Param: 1,050,624 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b2817fd90>,
          decode=False,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=None,
          cached_value=None,
          cache_index=None
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      )],
      output_dense=Linear( # Param: 393,984 (1.6 MB)
        kernel=Param( # 393,216 (1.6 MB)
          value=Array(shape=(512, 768), dtype=dtype('float32'))
        ),
        bias=Param( # 768 (3.1 KB)
          value=Array(shape=(768,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=768,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      )
    )
  ),
  dynamics=DynamicsCausal( # Param: 26,555,392 (106.2 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 65,512,460 (250.7 MB)
    model_dim=512,
    ffn_dim=2048,
    num_latents=1024,
    latent_action_dim=32,
    num_blocks=6,
    num_heads=8,
    dropout=0.0,
    param_dtype=float32,
    dtype=bfloat16,
    use_flash_attention=True,
    transformer=Transformer( # Param: 26,014,208 (104.1 MB), Cache: 5,677,068 (11.4 MB), Variable: 33,280,000 (133.1 MB), Total: 64,971,276 (248.5 MB)
      input_dim=512,
      model_dim=512,
      ffn_dim=2048,
      out_dim=1024,
      num_blocks=6,
      num_heads=8,
      dropout=0.0,
      param_dtype=float32,
      dtype=bfloat16,
      use_flash_attention=True,
      pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
        d_model=512,
        max_len=5000,
        pe=Variable( # 2,560,000 (10.2 MB)
          value=Array(shape=(5000, 512), dtype=dtype('float32'))
        )
      ),
      input_norm1=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      input_dense=Linear( # Param: 262,656 (1.1 MB)
        kernel=Param( # 262,144 (1.0 MB)
          value=Array(shape=(512, 512), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=512,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      ),
      input_norm2=LayerNorm( # Param: 1,024 (4.1 KB)
        scale=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        bias=Param( # 512 (2.0 KB)
          value=Array(shape=(512,), dtype=dtype('float32'))
        ),
        num_features=512,
        epsilon=1e-06,
        dtype=bfloat16,
        param_dtype=float32,
        use_bias=True,
        use_scale=True,
        bias_init=<function zeros at 0x7f8e437d1510>,
        scale_init=<function ones at 0x7f8e436a5870>,
        reduction_axes=-1,
        feature_axes=-1,
        axis_name=None,
        axis_index_groups=None,
        use_fast_variance=True
      ),
      blocks=[TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ec940>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ecca0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ecc10>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ecd30>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ecee0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed090>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed000>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ecf70>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))",,terminal_output
2744,3926779,"TERMINAL",0,0,"
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed1b0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed480>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      ), TransformerBlock( # Param: 4,204,032 (16.8 MB), Cache: 946,178 (1.9 MB), Variable: 5,120,000 (20.5 MB), Total: 10,270,210 (39.2 MB)
        model_dim=512,
        ffn_dim=2048,
        num_heads=8,
        dropout=0.0,
        param_dtype=float32,
        dtype=bfloat16,
        use_flash_attention=True,
        decode=True,
        temporal_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        spatial_pos_enc=PositionalEncoding( # Variable: 2,560,000 (10.2 MB)
          d_model=512,
          max_len=5000,
          pe=Variable( # 2,560,000 (10.2 MB)
            value=Array(shape=(5000, 512), dtype=dtype('float32'))
          )
        ),
        temporal_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        spatial_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        ffn_norm=LayerNorm( # Param: 1,024 (4.1 KB)
          scale=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          num_features=512,
          epsilon=1e-06,
          dtype=bfloat16,
          param_dtype=float32,
          use_bias=True,
          use_scale=True,
          bias_init=<function zeros at 0x7f8e437d1510>,
          scale_init=<function ones at 0x7f8e436a5870>,
          reduction_axes=-1,
          feature_axes=-1,
          axis_name=None,
          axis_index_groups=None,
          use_fast_variance=True
        ),
        temporal_attention=MultiHeadAttention( # Cache: 943,105 (1.9 MB), Param: 1,050,624 (4.2 MB), Total: 1,993,729 (6.1 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed630>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 471,552 (943.1 KB)
            value=Array(shape=(921, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        spatial_attention=MultiHeadAttention( # Cache: 3,073 (6.1 KB), Param: 1,050,624 (4.2 MB), Total: 1,053,697 (4.2 MB)
          num_heads=8,
          in_features=512,
          qkv_features=512,
          out_features=512,
          in_kv_features=512,
          dtype=bfloat16,
          param_dtype=float32,
          broadcast_dropout=True,
          dropout_rate=0.0,
          deterministic=None,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          out_kernel_init=None,
          bias_init=<function zeros at 0x7f8e437d1510>,
          out_bias_init=None,
          use_bias=True,
          attention_fn=<function _create_flash_attention_fn.<locals>.attention_fn at 0x7f8b217ed5a0>,
          decode=True,
          normalize_qk=False,
          qkv_dot_general=None,
          out_dot_general=None,
          qkv_dot_general_cls=None,
          out_dot_general_cls=None,
          head_dim=64,
          query=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          key=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          value=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(512,),
            out_features=(8, 64),
            axis=(-1,),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(512, 8, 64), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(8, 64), dtype=dtype('float32'))
            )
          ),
          query_ln=None,
          key_ln=None,
          out=LinearGeneral( # Param: 262,656 (1.1 MB)
            in_features=(8, 64),
            out_features=(512,),
            axis=(-2, -1),
            batch_axis=FrozenDict({}),
            use_bias=True,
            dtype=bfloat16,
            param_dtype=float32,
            kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
            bias_init=<function zeros at 0x7f8e437d1510>,
            precision=None,
            dot_general=None,
            dot_general_cls=None,
            promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>,
            kernel=Param( # 262,144 (1.0 MB)
              value=Array(shape=(8, 64, 512), dtype=dtype('float32'))
            ),
            bias=Param( # 512 (2.0 KB)
              value=Array(shape=(512,), dtype=dtype('float32'))
            )
          ),
          rngs=None,
          cached_key=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cached_value=Cache( # 1,536 (3.1 KB)
            value=Array(shape=(3, 1, 8, 64), dtype=dtype(bfloat16))
          ),
          cache_index=Cache( # 1 (4 B)
            value=Array(0, dtype=int32)
          )
        ),
        ffn_dense1=Linear( # Param: 1,050,624 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(512, 2048), dtype=dtype('float32'))
          ),
          bias=Param( # 2,048 (8.2 KB)
            value=Array(shape=(2048,), dtype=dtype('float32'))
          ),
          in_features=512,
          out_features=2048,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        ),
        ffn_dense2=Linear( # Param: 1,049,088 (4.2 MB)
          kernel=Param( # 1,048,576 (4.2 MB)
            value=Array(shape=(2048, 512), dtype=dtype('float32'))
          ),
          bias=Param( # 512 (2.0 KB)
            value=Array(shape=(512,), dtype=dtype('float32'))
          ),
          in_features=2048,
          out_features=512,
          use_bias=True,
          dtype=bfloat16,
          param_dtype=float32,
          precision=None,
          kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
          bias_init=<function zeros at 0x7f8e437d1510>,
          dot_general=<function dot_general at 0x7f8e43fda290>,
          promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
        )
      )],
      output_dense=Linear( # Param: 525,312 (2.1 MB)
        kernel=Param( # 524,288 (2.1 MB)
          value=Array(shape=(512, 1024), dtype=dtype('float32'))
        ),
        bias=Param( # 1,024 (4.1 KB)
          value=Array(shape=(1024,), dtype=dtype('float32'))
        ),
        in_features=512,
        out_features=1024,
        use_bias=True,
        dtype=bfloat16,
        param_dtype=float32,
        precision=None,
        kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
        bias_init=<function zeros at 0x7f8e437d1510>,
        dot_general=<function dot_general at 0x7f8e43fda290>,
        promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
      )
    ),
    patch_embed=Embed( # Param: 524,288 (2.1 MB)
      embedding=Param( # 524,288 (2.1 MB)
        value=Array(shape=(1024, 512), dtype=dtype('float32'))
      ),
      num_embeddings=1024,
      features=512,
      dtype=dtype('float32'),
      param_dtype=float32,
      embedding_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b3c70>,
      promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
    ),
    action_up=Linear( # Param: 16,896 (67.6 KB)
      kernel=Param( # 16,384 (65.5 KB)
        value=Array(shape=(32, 512), dtype=dtype('float32'))
      ),
      bias=Param( # 512 (2.0 KB)
        value=Array(shape=(512,), dtype=dtype('float32'))
      ),
      in_features=32,
      out_features=512,
      use_bias=True,
      dtype=bfloat16,
      param_dtype=float32,
      precision=None,
      kernel_init=<function variance_scaling.<locals>.init at 0x7f8e3e6b39a0>,
      bias_init=<function zeros at 0x7f8e437d1510>,
      dot_general=<function dot_general at 0x7f8e43fda290>,
      promote_dtype=<function promote_dtype at 0x7f8e3e6b3520>
    )
  )
)' of type '<class 'genie.Genie'>' is not a valid JAX type
",,terminal_output
2745,3927770,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
2746,3943429,"genie.py",15933,0,"",python,selection_mouse
2747,3946063,"genie.py",23,5,"",python,content
2748,3946070,"genie.py",22,0,"",python,selection_command
2749,3946995,"genie.py",14366,3,"Genie",python,content
2750,3946998,"genie.py",14366,0,"",python,selection_command
2751,3947443,"genie.py",14273,3,"Genie",python,content
2752,3947445,"genie.py",14273,0,"",python,selection_command
2753,3947826,"genie.py",15212,0,"_causal",python,content
2754,3947827,"genie.py",15198,6,"",python,content
2755,3947829,"genie.py",14872,0,"_causal",python,content
2756,3947829,"genie.py",14858,6,"",python,content
2757,3947830,"genie.py",14678,0,"_causal",python,content
2758,3947830,"genie.py",14664,6,"",python,content
2759,3947832,"genie.py",15178,0,"",python,selection_command
2760,3948118,"genie.py",15846,7,"",python,content
2761,3948119,"genie.py",15853,0,"",python,selection_command
2762,3948415,"genie.py",14441,1,"",python,content
2763,3948416,"genie.py",14440,0,"n",python,content
2764,3948427,"genie.py",14439,0,"",python,selection_command
2765,3948646,"genie.py",14439,7,"",python,content
2766,3948862,"genie.py",14368,7,"",python,content
2767,3948866,"genie.py",14368,0,"",python,selection_command
2768,3949043,"genie.py",14273,5,"",python,content
2769,3949045,"genie.py",14273,0,"",python,selection_command
2770,3949278,"genie.py",14273,2,"",python,content
2771,3949964,"genie.py",16170,23,"",python,content
2772,3949967,"genie.py",16152,0,"",python,selection_command
2773,3950655,"genie.py",16888,0,"
",python,content
2774,3950656,"genie.py",16887,0,"",python,selection_command
2775,3953007,"genie.py",16888,1,"",python,content
2776,3954108,"genie.py",16888,0,"
",python,content
2777,3955156,"genie.py",16751,5,"",python,content
2778,3955157,"genie.py",16751,0,"",python,selection_command
2779,3956043,"genie.py",16751,0,"genie.",python,content
2780,3956046,"genie.py",16751,0,"",python,selection_command
2781,3960942,"genie.py",16751,6,"",python,content
2782,3961136,"genie.py",16751,0,"self.",python,content
2783,3961139,"genie.py",16751,0,"",python,selection_command
2784,3961896,"genie.py",16888,1,"",python,content
2785,3961897,"genie.py",16887,0,"",python,selection_command
2786,3962693,"genie.py",16170,0,"genie,
                ",python,content
2787,3962699,"genie.py",16152,0,"",python,selection_command
2788,3963754,"genie.py",16170,23,"",python,content
2789,3997142,"experiments/sample.sh",0,0,"",shellscript,tab
2790,3998399,"genie.py",0,0,"",python,tab
2791,4020500,"genie.py",17194,0,"",python,selection_mouse
2792,4021895,"genie.py",16888,0,"
",python,content
2793,4021911,"genie.py",16887,0,"",python,selection_command
2794,4022015,"genie.py",16751,5,"",python,content
2795,4022024,"genie.py",16751,0,"",python,selection_command
2796,4022171,"genie.py",16751,0,"genie.",python,content
2797,4022176,"genie.py",16751,0,"",python,selection_command
2798,4022309,"genie.py",16662,227,"",python,content
2799,4022312,"genie.py",16624,0,"",python,selection_command
2800,4022438,"genie.py",16085,0,"Reset spatial KV caches before each frame ---
            for block in genie.dynamics.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- ",python,content
2801,4022441,"genie.py",16067,0,"",python,selection_command
2802,4022576,"genie.py",17104,20,"",python,content
2803,4022577,"genie.py",17104,0,"",python,selection_command
2804,4022710,"genie.py",17080,23,"il_ready(",python,content
2805,4022714,"genie.py",17075,3,"",python,content
2806,4022715,"genie.py",17069,5,"",python,content
2807,4022717,"genie.py",17065,3,"",python,content
2808,4022719,"genie.py",17064,0,"c",python,content
2809,4022723,"genie.py",17063,0,"ch_E).bl",python,content
2810,4022727,"genie.py",17062,0,"ba",python,content
2811,4022729,"genie.py",17055,0,"video_batch_BSHWC, ",python,content
2812,4022730,"genie.py",17045,5,"",python,content
2813,4022732,"genie.py",17039,4,"genie",python,content
2814,4022733,"genie.py",17037,0,"pl",python,content
2815,4022735,"genie.py",17024,11,"_autoreg_s",python,content
2816,4022739,"genie.py",17024,0,"",python,selection_command
2817,4023494,"genie.py",17145,0," #",python,content
2818,4023495,"genie.py",17120,0,"# ",python,content
2819,4023525,"genie.py",17011,0," #",python,content
2820,4023529,"genie.py",16975,0,"# ",python,content
2821,4023535,"genie.py",16975,0,"",python,selection_command
2822,4023675,"genie.py",17168,0," ",python,content
2823,4023676,"genie.py",17168,0,"",python,selection_command
2824,4023858,"genie.py",17168,0,"#",python,content
2825,4023861,"genie.py",17240,2,"",python,content
2826,4023862,"genie.py",17228,0,"",python,selection_command
2827,4024058,"genie.py",17227,1,"",python,content
2828,4024061,"genie.py",17168,0,"",python,selection_command
2829,4025553,"genie.py",16890,0,"
",python,content
2830,4025562,"genie.py",16890,0,"",python,selection_command
2831,4025696,"genie.py",16890,338,"",python,content
2832,4025700,"genie.py",16889,0,"",python,selection_command
2833,4025844,"genie.py",16889,1,"",python,content
2834,4025847,"genie.py",16839,0,"",python,selection_command
2835,4025991,"genie.py",16542,1,"",python,content
2836,4025993,"genie.py",16554,0,"",python,selection_command
2837,4026128,"genie.py",16542,0,"
            # Measure autoregressive generation time once and report it.
            # elapsed = timeit.timeit(
            #     lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
            #     number=1,
            # )
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")
",python,content
2838,4026130,"genie.py",16542,0,"",python,selection_command
2839,4026289,"genie.py",17291,1,"",python,content
2840,4026290,"genie.py",17291,0,"",python,selection_command
2841,4026676,"genie.py",17291,0,"g",python,content
2842,4026678,"genie.py",17284,5,"",python,content
2843,4026680,"genie.py",17284,0,"",python,selection_command
2844,4027192,"genie.py",16397,0,"step_",python,content
2845,4027194,"genie.py",16400,0,"",python,selection_command
2846,4028109,"genie.py",17294,0,"videos_BTHWC, ",python,content
2847,4028110,"genie.py",17270,0,"",python,selection_command
2848,4028394,"genie.py",16016,0,"video_batch_BSHWC, ",python,content
2849,4028397,"genie.py",16016,0,"",python,selection_command
2850,4028617,"genie.py",16050,1,"",python,content
2851,4028617,"genie.py",16042,6,"batch",python,content
2852,4028618,"genie.py",16042,0,"",python,selection_command
2853,4028883,"genie.py",17299,0,"s",python,content
2854,4028884,"genie.py",17299,0,"",python,selection_command
2855,4029115,"genie.py",16003,0,"s",python,content
2856,4029118,"genie.py",16003,0,"",python,selection_command
2857,4029309,"genie.py",16188,0,"_causal",python,content
2858,4029311,"genie.py",16174,6,"",python,content
2859,4029311,"genie.py",16174,0,"",python,selection_command
2860,4029499,"genie.py",17344,32,"",python,content
2861,4029500,"genie.py",17344,0,"",python,selection_command
2862,4029712,"genie.py",16050,32,"",python,content
2863,4029714,"genie.py",16050,0,"",python,selection_command
2864,4029881,"genie.py",17218,0,"
",python,content
2865,4029884,"genie.py",17218,0,"",python,selection_command
2866,4030048,"genie.py",17256,58,"final_carry_causal[1]",python,content
2867,4030049,"genie.py",17256,0,"",python,selection_command
2868,4030204,"genie.py",17009,42,"",python,content
2869,4030207,"genie.py",16812,2,"",python,content
2870,4030212,"genie.py",16797,2,"",python,content
2871,4030214,"genie.py",16769,2,"",python,content
2872,4030215,"genie.py",16658,2,"",python,content
2873,4030216,"genie.py",16620,2,"",python,content
2874,4030216,"genie.py",16624,0,"",python,selection_command
2875,4030344,"genie.py",16053,0,"
",python,content
2876,4030348,"genie.py",16053,0,"",python,selection_command
2877,4030527,"genie.py",16062,938,"",python,content
2878,4033782,"genie.py",16053,1,"",python,content
2879,4033790,"genie.py",15989,0,"",python,selection_command
2880,4034110,"genie.py",16241,0,"# --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # Measure autoregressive generation time once and report it.
            elapsed = timeit.timeit(
                lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                number=1,
            )
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            ",python,content
2881,4034117,"genie.py",16291,0,"",python,selection_command
2882,4034600,"genie.py",15985,76,"",python,content
2883,4034602,"genie.py",15932,0,"",python,selection_command
2884,4035342,"genie.py",24,14,"",python,content
2885,4035362,"genie.py",0,0,"",python,selection_command
2886,4038112,"genie.py",24,0,"import timeit
",python,content
2887,4039450,"genie.py",15985,0,"def generate_frames(genie, rng, video_batch_BSHWC, action_batch_E):
        ",python,content
2888,4039451,"genie.py",15932,0,"",python,selection_command
2889,4040354,"genie.py",15985,76,"",python,content
2890,4040523,"genie.py",24,14,"",python,content
2891,4040527,"genie.py",0,0,"",python,selection_command
2892,4041040,"genie.py",16633,0,"_autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
            ",python,content
2893,4041043,"genie.py",16633,0,"",python,selection_command
2894,4041224,"genie.py",16974,0,"    ",python,content
2895,4041225,"genie.py",16972,0,"    ",python,content
2896,4041225,"genie.py",16950,0,"    ",python,content
2897,4041226,"genie.py",16842,0,"  ",python,content
2898,4041226,"genie.py",16826,0,"  ",python,content
2899,4041227,"genie.py",16801,0,"    ",python,content
2900,4041227,"genie.py",16728,0," ",python,content
2901,4041252,"genie.py",16716,0,"   ",python,content
2902,4041253,"genie.py",16633,0,"    ",python,content
2903,4042328,"genie.py",16998,12,"",python,content
2904,4042332,"genie.py",16984,12,"",python,content
2905,4042333,"genie.py",16958,12,"",python,content
2906,4042337,"genie.py",16852,6,"",python,content
2907,4042339,"genie.py",16838,6,"",python,content
2908,4042340,"genie.py",16801,12,"",python,content
2909,4042340,"genie.py",16724,12,"",python,content
2910,4042342,"genie.py",16625,12,"",python,content
2911,4044055,"genie.py",16625,0,"            ",python,content
2912,4044057,"genie.py",16724,0,"            ",python,content
2913,4044058,"genie.py",16801,0,"            ",python,content
2914,4044058,"genie.py",16838,0,"      ",python,content
2915,4044064,"genie.py",16852,0,"      ",python,content
2916,4044069,"genie.py",16958,0,"            ",python,content
2917,4044074,"genie.py",16984,0,"            ",python,content
2918,4044075,"genie.py",16998,0,"            ",python,content
2919,4044079,"genie.py",16625,0,"",python,selection_command
2920,4046704,"genie.py",16637,0,"",python,selection_command
2921,4048402,"genie.py",16621,98,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()",python,selection_command
2922,4049770,"genie.py",16621,175,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.",python,selection_command
2923,4049929,"genie.py",16621,216,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(",python,selection_command
2924,4050076,"genie.py",16621,328,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),",python,selection_command
2925,4050263,"genie.py",16621,358,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,",python,selection_command
2926,4050396,"genie.py",16621,376,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,
                )",python,selection_command
2927,4050607,"genie.py",16621,449,"                _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready()
                # Measure autoregressive generation time once and report it.
                elapsed = timeit.timeit(
                    lambda: _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E).block_until_ready(),
                    number=1,
                )
                print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
2928,4051121,"genie.py",16621,450,"",python,content
2929,4051135,"genie.py",16633,0,"",python,selection_command
2930,4056136,"genie.py",16683,0,"",python,selection_command
2931,4060930,"genie.py",16748,0,"",python,selection_command
2932,4061131,"genie.py",16749,0,"
            ",python,content
2933,4061470,"genie.py",16762,0,"f",python,content
2934,4061471,"genie.py",16763,0,"",python,selection_keyboard
2935,4061521,"genie.py",16763,0,"i",python,content
2936,4061523,"genie.py",16764,0,"",python,selection_keyboard
2937,4061725,"genie.py",16764,0,"n",python,content
2938,4061727,"genie.py",16765,0,"",python,selection_keyboard
2939,4062284,"genie.py",16765,0,"a",python,content
2940,4062286,"genie.py",16766,0,"",python,selection_keyboard
2941,4062392,"genie.py",16766,0,"l",python,content
2942,4062393,"genie.py",16767,0,"",python,selection_keyboard
2943,4063598,"genie.py",16767,0,"_carry_causal[1].block_until_ready()",python,content
2944,4064077,"genie.py",16802,0,"",python,selection_command
2945,4065581,"genie.py",16748,0,"",python,selection_command
2946,4065693,"genie.py",16723,0,"",python,selection_command
2947,4065831,"genie.py",16669,0,"",python,selection_command
2948,4067993,"genie.py",16620,0,"
",python,content
2949,4068015,"genie.py",16621,0,"            ",python,content
2950,4072437,"genie.py",16633,0,"c",python,content
2951,4072438,"genie.py",16634,0,"",python,selection_keyboard
2952,4072499,"genie.py",16634,0,"u",python,content
2953,4072500,"genie.py",16635,0,"",python,selection_keyboard
2954,4072665,"genie.py",16635,0,"r",python,content
2955,4072666,"genie.py",16636,0,"",python,selection_keyboard
2956,4072845,"genie.py",16636,0,"r",python,content
2957,4072846,"genie.py",16637,0,"",python,selection_keyboard
2958,4073755,"genie.py",16637,0,"ent_token_idxs_BSN = jax.lax.cond(",python,content
2959,4074372,"genie.py",16670,0,"",python,selection_command
2960,4074601,"genie.py",16666,0,"",python,selection_command
2961,4074770,"genie.py",16665,0,"",python,selection_command
2962,4074970,"genie.py",16662,0,"",python,selection_command
2963,4075184,"genie.py",16661,0,"",python,selection_command
2964,4075607,"genie.py",16658,0,"",python,selection_command
2965,4076118,"genie.py",16658,13,"",python,content
2966,4076456,"genie.py",16658,0,".",python,content
2967,4076457,"genie.py",16659,0,"",python,selection_keyboard
2968,4077091,"genie.py",16658,1,"",python,content
2969,4077238,"genie.py",16657,1,"",python,content
2970,4077370,"genie.py",16656,1,"",python,content
2971,4077539,"genie.py",16655,1,"",python,content
2972,4077905,"genie.py",16655,0,".",python,content
2973,4077907,"genie.py",16656,0,"",python,selection_keyboard
2974,4078428,"genie.py",16656,0,"b",python,content
2975,4078428,"genie.py",16657,0,"",python,selection_keyboard
2976,4078588,"genie.py",16657,0,"l",python,content
2977,4078589,"genie.py",16658,0,"",python,selection_keyboard
2978,4078703,"genie.py",16658,0,"o",python,content
2979,4078703,"genie.py",16659,0,"",python,selection_keyboard
2980,4078772,"genie.py",16659,0,"c",python,content
2981,4078773,"genie.py",16660,0,"",python,selection_keyboard
2982,4079206,"genie.py",16660,0,"k_until_ready()",python,content
2983,4079875,"genie.py",16674,0,"",python,selection_command
2984,4082880,"genie.py",16675,0,"
            ",python,content
2985,4084716,"genie.py",16688,0,"t",python,content
2986,4084716,"genie.py",16689,0,"",python,selection_keyboard
2987,4084756,"genie.py",16689,0,"i",python,content
2988,4084757,"genie.py",16690,0,"",python,selection_keyboard
2989,4084819,"genie.py",16690,0,"m",python,content
2990,4084819,"genie.py",16691,0,"",python,selection_keyboard
2991,4084926,"genie.py",16691,0,"e",python,content
2992,4084926,"genie.py",16692,0,"",python,selection_keyboard
2993,4085125,"genie.py",16692,0," ",python,content
2994,4085125,"genie.py",16693,0,"",python,selection_keyboard
2995,4086804,"genie.py",16693,0,"=",python,content
2996,4086805,"genie.py",16694,0,"",python,selection_keyboard
2997,4087504,"genie.py",16694,0," ",python,content
2998,4087504,"genie.py",16695,0,"",python,selection_keyboard
2999,4088366,"genie.py",16695,0,"t",python,content
3000,4088366,"genie.py",16696,0,"",python,selection_keyboard
3001,4088414,"genie.py",16696,0,"i",python,content
3002,4088415,"genie.py",16697,0,"",python,selection_keyboard
3003,4088479,"genie.py",16697,0,"m",python,content
3004,4088480,"genie.py",16698,0,"",python,selection_keyboard
3005,4088566,"genie.py",16698,0,"e",python,content
3006,4088567,"genie.py",16699,0,"",python,selection_keyboard
3007,4088718,"genie.py",16699,0,".",python,content
3008,4088718,"genie.py",16700,0,"",python,selection_keyboard
3009,4088814,"genie.py",16700,0,"t",python,content
3010,4088815,"genie.py",16701,0,"",python,selection_keyboard
3011,4088934,"genie.py",16701,0,"i",python,content
3012,4088935,"genie.py",16702,0,"",python,selection_keyboard
3013,4088984,"genie.py",16702,0,"m",python,content
3014,4088984,"genie.py",16703,0,"",python,selection_keyboard
3015,4089044,"genie.py",16703,0,"e",python,content
3016,4089044,"genie.py",16704,0,"",python,selection_keyboard
3017,4089914,"genie.py",16704,0,"()",python,content
3018,4089915,"genie.py",16705,0,"",python,selection_keyboard
3019,4089997,"genie.py",16705,1,")",python,content
3020,4089998,"genie.py",16706,0,"",python,selection_keyboard
3021,4090472,"genie.py",16705,0,"",python,selection_command
3022,4092052,"genie.py",16736,0,"",python,selection_command
3023,4092212,"genie.py",16786,0,"",python,selection_command
3024,4092363,"genie.py",16834,0,"",python,selection_command
3025,4092818,"genie.py",16865,0,"",python,selection_command
3026,4093319,"genie.py",16889,0,"
            ",python,content
3027,4093579,"genie.py",16902,0,"e",python,content
3028,4093579,"genie.py",16903,0,"",python,selection_keyboard
3029,4093888,"genie.py",16903,0,"l",python,content
3030,4093888,"genie.py",16904,0,"",python,selection_keyboard
3031,4094558,"genie.py",16904,0,"a",python,content
3032,4094559,"genie.py",16905,0,"",python,selection_keyboard
3033,4095047,"genie.py",16905,0,"psed = time.time() - time",python,content
3034,4095586,"genie.py",16929,0,"",python,selection_command
3035,4099278,"genie.py",0,0,"",python,selection_command
3036,4100565,"genie.py",23,0,"
import time",python,content
3037,4100928,"genie.py",1,0,"",python,selection_command
3038,4101274,"genie.py",0,0,"",python,selection_command
3039,4102257,"genie.py",24,0,"",python,selection_command
3040,4102631,"genie.py",25,0,"",python,selection_command
3041,4102835,"genie.py",26,0,"",python,selection_command
3042,4103001,"genie.py",27,0,"",python,selection_command
3043,4103157,"genie.py",28,0,"",python,selection_command
3044,4103314,"genie.py",29,0,"",python,selection_command
3045,4103452,"genie.py",30,0,"",python,selection_command
3046,4103623,"genie.py",31,0,"",python,selection_command
3047,4104094,"genie.py",16700,0,"",python,selection_command
3048,4105499,"genie.py",16705,0,"",python,selection_command
3049,4105686,"genie.py",16707,0,"",python,selection_command
3050,4108598,"genie.py",16738,0,"",python,selection_command
3051,4108763,"genie.py",16707,0,"",python,selection_command
3052,4109515,"genie.py",16700,0,"",python,selection_command
3053,4111684,"genie.py",16703,0,"",python,selection_command
3054,4113003,"genie.py",16731,0,"",python,selection_command
3055,4113614,"genie.py",16700,0,"",python,selection_command
3056,4114058,"genie.py",16700,4,"",python,content
3057,4114421,"genie.py",16700,0,"s",python,content
3058,4114421,"genie.py",16701,0,"",python,selection_keyboard
3059,4114440,"genie.py",16701,0,"t",python,content
3060,4114440,"genie.py",16702,0,"",python,selection_keyboard
3061,4114715,"genie.py",16702,0,"a",python,content
3062,4114716,"genie.py",16703,0,"",python,selection_keyboard
3063,4114722,"genie.py",16703,0,"r",python,content
3064,4114722,"genie.py",16704,0,"",python,selection_keyboard
3065,4114799,"genie.py",16704,0,"t",python,content
3066,4114800,"genie.py",16705,0,"",python,selection_keyboard
3067,4115162,"genie.py",16704,0,"",python,selection_command
3068,4116348,"genie.py",16736,0,"",python,selection_command
3069,4116499,"genie.py",16786,0,"",python,selection_command
3070,4116655,"genie.py",16847,0,"",python,selection_command
3071,4116780,"genie.py",16865,0,"",python,selection_command
3072,4116952,"genie.py",16919,0,"",python,selection_command
3073,4117218,"genie.py",16923,0,"",python,selection_command
3074,4117376,"genie.py",16925,0,"",python,selection_command
3075,4117553,"genie.py",16929,0,"",python,selection_command
3076,4117706,"genie.py",16930,0,"",python,selection_command
3077,4117889,"genie.py",16934,0,"",python,selection_command
3078,4118060,"genie.py",16937,0,"",python,selection_command
3079,4118285,"genie.py",16939,0,"",python,selection_command
3080,4118806,"genie.py",16939,4,"",python,content
3081,4119177,"genie.py",16939,0,"s",python,content
3082,4119177,"genie.py",16940,0,"",python,selection_keyboard
3083,4119208,"genie.py",16940,0,"t",python,content
3084,4119209,"genie.py",16941,0,"",python,selection_keyboard
3085,4119528,"genie.py",16941,0,"a",python,content
3086,4119528,"genie.py",16942,0,"",python,selection_keyboard
3087,4119529,"genie.py",16942,0,"r",python,content
3088,4119529,"genie.py",16943,0,"",python,selection_keyboard
3089,4119594,"genie.py",16943,0,"t",python,content
3090,4119594,"genie.py",16944,0,"",python,selection_keyboard
3091,4119878,"genie.py",16943,0,"",python,selection_command
3092,4123535,"genie.py",16944,0,"
            ",python,content
3093,4123713,"genie.py",16957,0,"p",python,content
3094,4123713,"genie.py",16958,0,"",python,selection_keyboard
3095,4123820,"genie.py",16958,0,"r",python,content
3096,4123821,"genie.py",16959,0,"",python,selection_keyboard
3097,4123924,"genie.py",16959,0,"i",python,content
3098,4123924,"genie.py",16960,0,"",python,selection_keyboard
3099,4124081,"genie.py",16960,0,"n",python,content
3100,4124082,"genie.py",16961,0,"",python,selection_keyboard
3101,4124146,"genie.py",16961,0,"t",python,content
3102,4124147,"genie.py",16962,0,"",python,selection_keyboard
3103,4124737,"genie.py",16962,0,"(f""Autoregressive generation time: {elapsed:.4f}s"")",python,content
3104,4125078,"genie.py",17012,0,"",python,selection_command
3105,4128240,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3106,4128517,"TERMINAL",0,0,"

",,terminal_output
3107,4139820,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3108,4153272,"TERMINAL",0,0,"2025-08-01 10:44:22.017294: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3109,4157760,"TERMINAL",0,0,"2025-08-01 10:44:26.495427: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3110,4166879,"TERMINAL",0,0,"2025-08-01 10:44:35.625040: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3111,4180242,"TERMINAL",0,0,"Autoregressive generation time: 9.9984s
",,terminal_output
3112,4180832,"TERMINAL",0,0,"2025-08-01 10:44:49.578466: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3113,4192713,"TERMINAL",0,0,"SSIM: 0.005976376589387655
",,terminal_output
3114,4194441,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3115,4217391,"genie.py",17013,0,"
            ",python,content
3116,4217499,"genie.py",17026,0,"b",python,content
3117,4217499,"genie.py",17027,0,"",python,selection_keyboard
3118,4217706,"genie.py",17027,0,"r",python,content
3119,4217706,"genie.py",17028,0,"",python,selection_keyboard
3120,4217800,"genie.py",17028,0,"e",python,content
3121,4217800,"genie.py",17029,0,"",python,selection_keyboard
3122,4218049,"genie.py",17029,0,"a",python,content
3123,4218049,"genie.py",17030,0,"",python,selection_keyboard
3124,4218134,"genie.py",17030,0,"k",python,content
3125,4218134,"genie.py",17031,0,"",python,selection_keyboard
3126,4218961,"genie.py",17031,0,"p",python,content
3127,4218962,"genie.py",17032,0,"",python,selection_keyboard
3128,4219000,"genie.py",17032,0,"o",python,content
3129,4219000,"genie.py",17033,0,"",python,selection_keyboard
3130,4219118,"genie.py",17033,0,"i",python,content
3131,4219118,"genie.py",17034,0,"",python,selection_keyboard
3132,4219322,"genie.py",17034,0,"n",python,content
3133,4219322,"genie.py",17035,0,"",python,selection_keyboard
3134,4219470,"genie.py",17035,0,"t",python,content
3135,4219470,"genie.py",17036,0,"",python,selection_keyboard
3136,4219841,"genie.py",17036,0,"()",python,content
3137,4219841,"genie.py",17037,0,"",python,selection_keyboard
3138,4219931,"genie.py",17037,1,")",python,content
3139,4219931,"genie.py",17038,0,"",python,selection_keyboard
3140,4220292,"genie.py",17037,0,"",python,selection_command
3141,4222243,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3142,4222561,"TERMINAL",0,0,"

",,terminal_output
3143,4233889,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3144,4242753,"TERMINAL",0,0,"2025-08-01 10:45:51.390952: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3145,4247258,"TERMINAL",0,0,"2025-08-01 10:45:55.903194: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3146,4256226,"TERMINAL",0,0,"2025-08-01 10:46:04.971770: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3147,4259288,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.29 task 0: running
",,terminal_output
3148,4259484,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.29
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T10:46:08.227] error: *** STEP 14571.29 ON hai005 CANCELLED AT 2025-08-01T10:46:08 DUE to SIGNAL Killed ***
",,terminal_output
3149,4259617,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.29
srun: job abort in progress
",,terminal_output
3150,4259863,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3151,4260916,"genie.py",17014,25,"",python,content
3152,4260952,"genie.py",17026,0,"",python,selection_command
3153,4265971,"experiments/sample.sh",0,0,"",shellscript,tab
3154,4268108,"genie.py",0,0,"",python,tab
3155,4269671,"genie.py",16957,0,"",python,selection_command
3156,4269742,"genie.py",16915,0,"",python,selection_command
3157,4270052,"genie.py",16957,0,"",python,selection_command
3158,4301017,"genie.py",16915,0,"",python,selection_command
3159,4301160,"genie.py",16861,0,"",python,selection_command
3160,4304428,"genie.py",16915,0,"",python,selection_command
3161,4304585,"genie.py",16957,0,"",python,selection_command
3162,4306248,"genie.py",17013,0,"
            ",python,content
3163,4306358,"genie.py",17026,0,"b",python,content
3164,4306358,"genie.py",17027,0,"",python,selection_keyboard
3165,4306605,"genie.py",17027,0,"r",python,content
3166,4306606,"genie.py",17028,0,"",python,selection_keyboard
3167,4306656,"genie.py",17028,0,"e",python,content
3168,4306656,"genie.py",17029,0,"",python,selection_keyboard
3169,4306896,"genie.py",17029,0,"a",python,content
3170,4306896,"genie.py",17030,0,"",python,selection_keyboard
3171,4306913,"genie.py",17030,0,"k",python,content
3172,4306913,"genie.py",17031,0,"",python,selection_keyboard
3173,4307904,"genie.py",17031,0,"p",python,content
3174,4307904,"genie.py",17032,0,"",python,selection_keyboard
3175,4307943,"genie.py",17032,0,"o",python,content
3176,4307944,"genie.py",17033,0,"",python,selection_keyboard
3177,4308027,"genie.py",17033,0,"i",python,content
3178,4308028,"genie.py",17034,0,"",python,selection_keyboard
3179,4308143,"genie.py",17034,0,"n",python,content
3180,4308144,"genie.py",17035,0,"",python,selection_keyboard
3181,4308243,"genie.py",17035,0,"t",python,content
3182,4308243,"genie.py",17036,0,"",python,selection_keyboard
3183,4308968,"genie.py",17036,0,"()",python,content
3184,4308968,"genie.py",17037,0,"",python,selection_keyboard
3185,4309080,"genie.py",17037,1,")",python,content
3186,4309080,"genie.py",17038,0,"",python,selection_keyboard
3187,4309529,"genie.py",17037,0,"",python,selection_command
3188,4311073,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3189,4311326,"TERMINAL",0,0,"

",,terminal_output
3190,4313834,"experiments/sample.sh",0,0,"",shellscript,tab
3191,4317359,"genie.py",0,0,"",python,tab
3192,4322652,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3193,4329198,"genie.py",16645,0,"",python,selection_command
3194,4331104,"genie.py",0,0,"",python,tab
3195,4331564,"TERMINAL",0,0,"2025-08-01 10:47:20.313508: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3196,4336099,"TERMINAL",0,0,"2025-08-01 10:47:24.849687: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3197,4337161,"genie.py",16646,0,"",python,selection_command
3198,4344069,"genie.py",16687,0,"",python,selection_keyboard
3199,4344791,"genie.py",22305,0,"",python,selection_keyboard
3200,4345225,"genie.py",0,0,"",python,selection_keyboard
3201,4345301,"TERMINAL",0,0,"2025-08-01 10:47:34.020841: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3202,4345651,"genie.py",22305,0,"",python,selection_keyboard
3203,4345919,"genie.py",0,0,"",python,selection_keyboard
3204,4346338,"genie.py",22305,0,"",python,selection_keyboard
3205,4346591,"genie.py",0,0,"",python,selection_keyboard
3206,4346824,"genie.py",22305,0,"",python,selection_keyboard
3207,4350032,"genie.py",22305,0,"
",python,content
3208,4352408,"genie.py",22305,1,"",python,content
3209,4352961,"genie.py",17026,25,"",python,content
3210,4352964,"genie.py",16957,0,"",python,selection_command
3211,4354305,"genie.py",17026,0,"breakpoint()
            ",python,content
3212,4358713,"TERMINAL",0,0,"Autoregressive generation time: 10.0497s
> /fast/home/franz.srambical/jafar/genie.py(412)sample_causal()
-> current_token_idxs_BSN = final_carry_causal[1]
",,terminal_output
3213,4365015,"TERMINAL",0,0,"c",,terminal_output
3214,4365357,"TERMINAL",0,0,"
",,terminal_output
3215,4365959,"TERMINAL",0,0,"2025-08-01 10:47:54.704778: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3216,4377956,"TERMINAL",0,0,"(Pdb) SSIM: 0.005976376589387655
",,terminal_output
3217,4378225,"genie.py",16915,0,"",python,selection_command
3218,4378484,"genie.py",16861,0,"",python,selection_command
3219,4378504,"genie.py",16847,0,"",python,selection_command
3220,4378542,"genie.py",16782,0,"",python,selection_command
3221,4378571,"genie.py",16732,0,"",python,selection_command
3222,4378602,"genie.py",16700,0,"",python,selection_command
3223,4378641,"genie.py",16645,0,"",python,selection_command
3224,4378674,"genie.py",16632,0,"",python,selection_command
3225,4378709,"genie.py",16630,0,"",python,selection_command
3226,4378744,"genie.py",16578,0,"",python,selection_command
3227,4378781,"genie.py",16544,0,"",python,selection_command
3228,4378814,"genie.py",16504,0,"",python,selection_command
3229,4378844,"genie.py",16478,0,"",python,selection_command
3230,4378876,"genie.py",16444,0,"",python,selection_command
3231,4378909,"genie.py",16391,0,"",python,selection_command
3232,4378948,"genie.py",16378,0,"",python,selection_command
3233,4378984,"genie.py",16288,0,"",python,selection_command
3234,4379015,"genie.py",16227,0,"",python,selection_command
3235,4379050,"genie.py",16163,0,"",python,selection_command
3236,4379081,"genie.py",16150,0,"",python,selection_command
3237,4379114,"genie.py",16112,0,"",python,selection_command
3238,4379309,"genie.py",16071,0,"",python,selection_command
3239,4379680,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3240,4379850,"genie.py",16076,0,"",python,selection_command
3241,4380005,"genie.py",16079,0,"",python,selection_command
3242,4380187,"genie.py",16085,0,"",python,selection_command
3243,4380356,"genie.py",16086,0,"",python,selection_command
3244,4380525,"genie.py",16087,0,"",python,selection_command
3245,4380722,"genie.py",16088,0,"",python,selection_command
3246,4380890,"genie.py",16096,0,"",python,selection_command
3247,4382107,"genie.py",16090,0,"",python,selection_command
3248,4382257,"genie.py",16088,0,"",python,selection_command
3249,4382444,"genie.py",16087,0,"",python,selection_command
3250,4387744,"genie.py",16088,0,"",python,selection_command
3251,4387895,"genie.py",16096,0,"",python,selection_command
3252,4389398,"genie.py",16097,0,"",python,selection_command
3253,4391938,"genie.py",16097,0,"+",python,content
3254,4391938,"genie.py",16098,0,"",python,selection_keyboard
3255,4392650,"genie.py",16098,0,"1",python,content
3256,4392650,"genie.py",16099,0,"",python,selection_keyboard
3257,4393265,"genie.py",16098,0,"",python,selection_command
3258,4393954,"genie.py",16098,1,"1",python,selection_command
3259,4394044,"genie.py",16097,2,"+1",python,selection_command
3260,4394264,"genie.py",16097,2,"",python,content
3261,4396502,"genie.py",16059,40,"        for step_t in range(T, seq_len):",python,selection_command
3262,4418009,"genie.py",16097,0,"",python,selection_command
3263,4419458,"genie.py",16097,0," ",python,content
3264,4419459,"genie.py",16098,0,"",python,selection_keyboard
3265,4419622,"genie.py",16098,0,"+",python,content
3266,4419623,"genie.py",16099,0,"",python,selection_keyboard
3267,4420319,"genie.py",16099,0," ",python,content
3268,4420319,"genie.py",16100,0,"",python,selection_keyboard
3269,4420550,"genie.py",16100,0,"1",python,content
3270,4420550,"genie.py",16101,0,"",python,selection_keyboard
3271,4420854,"genie.py",16100,0,"",python,selection_command
3272,4430569,"genie.py",16097,4,"",python,content
3273,4536772,"experiments/sample.sh",0,0,"",shellscript,tab
3274,4543358,"experiments/sample.sh",234,17,"    --seq_len 3 \",shellscript,selection_command
3275,4550043,"genie.py",0,0,"",python,tab
3276,4580321,"experiments/sample.sh",0,0,"",shellscript,tab
3277,4581323,"experiments/sample.sh",248,0,"",shellscript,selection_command
3278,4584939,"experiments/sample.sh",248,1,"4",shellscript,content
3279,4587185,"genie.py",0,0,"",python,tab
3280,4588557,"genie.py",16138,0,"",python,selection_command
3281,4588798,"genie.py",16150,0,"",python,selection_command
3282,4588828,"genie.py",16189,0,"",python,selection_command
3283,4588853,"genie.py",16253,0,"",python,selection_command
3284,4588890,"genie.py",16314,0,"",python,selection_command
3285,4588920,"genie.py",16378,0,"",python,selection_command
3286,4588955,"genie.py",16417,0,"",python,selection_command
3287,4588985,"genie.py",16464,0,"",python,selection_command
3288,4589021,"genie.py",16490,0,"",python,selection_command
3289,4589051,"genie.py",16530,0,"",python,selection_command
3290,4589087,"genie.py",16564,0,"",python,selection_command
3291,4589119,"genie.py",16604,0,"",python,selection_command
3292,4589155,"genie.py",16630,0,"",python,selection_command
3293,4589188,"genie.py",16632,0,"",python,selection_command
3294,4590162,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3295,4590396,"TERMINAL",0,0,"

",,terminal_output
3296,4601707,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3297,4610234,"TERMINAL",0,0,"2025-08-01 10:51:58.977673: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3298,4614719,"TERMINAL",0,0,"2025-08-01 10:52:03.464867: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3299,4623892,"TERMINAL",0,0,"2025-08-01 10:52:12.590782: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3300,4629763,"genie.py",0,0,"",python,tab
3301,4637340,"TERMINAL",0,0,"Autoregressive generation time: 10.0764s
> /fast/home/franz.srambical/jafar/genie.py(412)sample_causal()
-> current_token_idxs_BSN = final_carry_causal[1]
",,terminal_output
3302,4638755,"genie.py",0,0,"",python,tab
3303,4643281,"TERMINAL",0,0,"c",,terminal_output
3304,4643765,"TERMINAL",0,0,"
Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 393, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
3305,4644778,"TERMINAL",0,0,"(Pdb) ",,terminal_output
3306,4644926,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3307,4659215,"experiments/sample.sh",0,0,"",shellscript,tab
3308,4661851,"genie.py",0,0,"",python,tab
3309,4688056,"genie.py",16633,0,"",python,selection_command
3310,4688304,"genie.py",16688,0,"",python,selection_command
3311,4688323,"genie.py",16720,0,"",python,selection_command
3312,4688355,"genie.py",16770,0,"",python,selection_command
3313,4688390,"genie.py",16835,0,"",python,selection_command
3314,4688426,"genie.py",16849,0,"",python,selection_command
3315,4688770,"genie.py",16835,0,"",python,selection_command
3316,4688941,"genie.py",16770,0,"",python,selection_command
3317,4689110,"genie.py",16720,0,"",python,selection_command
3318,4689473,"genie.py",16688,0,"",python,selection_command
3319,4689806,"genie.py",16633,0,"",python,selection_command
3320,4689963,"genie.py",16632,0,"",python,selection_command
3321,4690328,"genie.py",16633,0,"",python,selection_command
3322,4690506,"genie.py",16688,0,"",python,selection_command
3323,4690658,"genie.py",16720,0,"",python,selection_command
3324,4690818,"genie.py",16770,0,"",python,selection_command
3325,4690929,"genie.py",16835,0,"",python,selection_command
3326,4691085,"genie.py",16849,0,"",python,selection_command
3327,4700373,"genie.py",16835,0,"",python,selection_command
3328,4700528,"genie.py",16770,0,"",python,selection_command
3329,4700685,"genie.py",16720,0,"",python,selection_command
3330,4700840,"genie.py",16688,0,"",python,selection_command
3331,4701180,"genie.py",16633,0,"",python,selection_command
3332,4702109,"genie.py",16633,54,"            current_token_idxs_BSN.block_until_ready()",python,selection_command
3333,4702505,"genie.py",16633,86,"            current_token_idxs_BSN.block_until_ready()
            start = time.time()",python,selection_command
3334,4702680,"genie.py",16633,136,"            current_token_idxs_BSN.block_until_ready()
            start = time.time()
            final_carry_causal, _ = jax.lax.scan(",python,selection_command
3335,4703042,"genie.py",16633,86,"            current_token_idxs_BSN.block_until_ready()
            start = time.time()",python,selection_command
3336,4703568,"genie.py",16700,0,"# ",python,content
3337,4703568,"genie.py",16645,0,"# ",python,content
3338,4704240,"genie.py",16690,0,"",python,selection_command
3339,4704598,"genie.py",16724,0,"",python,selection_command
3340,4704703,"genie.py",16774,0,"",python,selection_command
3341,4704857,"genie.py",16839,0,"",python,selection_command
3342,4704989,"genie.py",16853,0,"",python,selection_command
3343,4705543,"genie.py",16853,53,"            final_carry_causal[1].block_until_ready()",python,selection_command
3344,4706127,"genie.py",16853,95,"            final_carry_causal[1].block_until_ready()
            elapsed = time.time() - start",python,selection_command
3345,4706156,"genie.py",16853,164,"            final_carry_causal[1].block_until_ready()
            elapsed = time.time() - start
            print(f""Autoregressive generation time: {elapsed:.4f}s"")",python,selection_command
3346,4706293,"genie.py",16853,189,"            final_carry_causal[1].block_until_ready()
            elapsed = time.time() - start
            print(f""Autoregressive generation time: {elapsed:.4f}s"")
            breakpoint()",python,selection_command
3347,4706831,"genie.py",17030,0,"# ",python,content
3348,4706831,"genie.py",16961,0,"# ",python,content
3349,4706831,"genie.py",16919,0,"# ",python,content
3350,4706831,"genie.py",16865,0,"# ",python,content
3351,4707148,"genie.py",17024,0,"",python,selection_command
3352,4708725,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3353,4709087,"TERMINAL",0,0,"

",,terminal_output
3354,4720361,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3355,4727589,"/fast/home/franz.srambical/jafar/genie.py",0,0,"from typing import Dict
import time

import optax
import jax
import jax.numpy as jnp
import flax.nnx as nnx
import orbax.checkpoint as ocp

from models.dynamics import DynamicsMaskGIT, DynamicsCausal
from models.lam import LatentActionModel
from models.tokenizer import TokenizerVQVAE


class Genie(nnx.Module):
    """"""Genie model""""""

    def __init__(
        self,
        in_dim: int,
        tokenizer_dim: int,
        tokenizer_ffn_dim: int,
        latent_patch_dim: int,
        num_patch_latents: int,
        patch_size: int,
        tokenizer_num_blocks: int,
        tokenizer_num_heads: int,
        lam_dim: int,
        lam_ffn_dim: int,
        latent_action_dim: int,
        num_latent_actions: int,
        lam_patch_size: int,
        lam_num_blocks: int,
        lam_num_heads: int,
        lam_co_train: bool,
        dyna_type: str,
        dyna_dim: int,
        dyna_ffn_dim: int,
        dyna_num_blocks: int,
        dyna_num_heads: int,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
        dropout: float = 0.0,
        mask_limit: float = 0.0,
    ):
        # --- Tokenizer ---
        self.in_dim = in_dim
        self.tokenizer_dim = tokenizer_dim
        self.tokenizer_ffn_dim = tokenizer_ffn_dim
        self.latent_patch_dim = latent_patch_dim
        self.num_patch_latents = num_patch_latents
        self.patch_size = patch_size
        self.tokenizer_num_blocks = tokenizer_num_blocks
        self.tokenizer_num_heads = tokenizer_num_heads
        # --- LAM ---
        self.lam_dim = lam_dim
        self.lam_ffn_dim = lam_ffn_dim
        self.latent_action_dim = latent_action_dim
        self.num_latent_actions = num_latent_actions
        self.lam_patch_size = lam_patch_size
        self.lam_num_blocks = lam_num_blocks
        self.lam_num_heads = lam_num_heads
        self.lam_co_train = lam_co_train
        # --- Dynamics ---
        self.dyna_type = dyna_type
        self.dyna_dim = dyna_dim
        self.dyna_ffn_dim = dyna_ffn_dim
        self.dyna_num_blocks = dyna_num_blocks
        self.dyna_num_heads = dyna_num_heads
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention
        self.dropout = dropout
        self.mask_limit = mask_limit

        self.tokenizer = TokenizerVQVAE(
            in_dim=self.in_dim,
            model_dim=self.tokenizer_dim,
            ffn_dim=self.tokenizer_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_patch_latents,
            patch_size=self.patch_size,
            num_blocks=self.tokenizer_num_blocks,
            num_heads=self.tokenizer_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        self.lam = LatentActionModel(
            in_dim=self.in_dim,
            model_dim=self.lam_dim,
            ffn_dim=self.lam_ffn_dim,
            latent_dim=self.latent_patch_dim,
            num_latents=self.num_latent_actions,
            patch_size=self.lam_patch_size,
            num_blocks=self.lam_num_blocks,
            num_heads=self.lam_num_heads,
            dropout=0.0,
            codebook_dropout=0.0,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            use_flash_attention=self.use_flash_attention,
            rngs=rngs,
        )
        if self.dyna_type == ""maskgit"":
            self.dynamics = DynamicsMaskGIT(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                mask_limit=self.mask_limit,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                rngs=rngs,
            )
        elif self.dyna_type == ""causal"":
            self.dynamics = DynamicsCausal(
                model_dim=self.dyna_dim,
                ffn_dim=self.dyna_ffn_dim,
                num_latents=self.num_patch_latents,
                latent_action_dim=self.latent_action_dim,
                num_blocks=self.dyna_num_blocks,
                num_heads=self.dyna_num_heads,
                dropout=self.dropout,
                param_dtype=self.param_dtype,
                dtype=self.dtype,
                use_flash_attention=self.use_flash_attention,
                decode=decode,
                rngs=rngs,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {self.dyna_type}"")

    def __call__(
        self, batch: Dict[str, jax.Array], training: bool = True
    ) -> Dict[str, jax.Array]:
        videos_BTHWC = batch[""videos""]
        tokenizer_outputs = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_indices_BTN = tokenizer_outputs[""indices""]
        lam_outputs = self.lam.vq_encode(videos_BTHWC, training=False)
        z_q_BTm11L = lam_outputs[""z_q""]
        action_indices_E = lam_outputs[""indices""]
        latent_actions_BTm11L = jax.lax.cond(
            self.lam_co_train,
            lambda: z_q_BTm11L,
            lambda: jax.lax.stop_gradient(z_q_BTm11L),
        )
        outputs = dict(
            video_tokens=jax.lax.stop_gradient(token_indices_BTN),
            latent_actions=latent_actions_BTm11L,
        )
        outputs[""mask_rng""] = batch[""mask_rng""]
        dyna_logits_BTNV, dyna_mask = self.dynamics(outputs, training)
        outputs[""token_logits""] = dyna_logits_BTNV
        if dyna_mask is not None:
            outputs[""mask""] = dyna_mask
        mle_indices_BTN = jnp.argmax(outputs[""token_logits""], axis=-1)
        H, W = batch[""videos""].shape[2:4]
        outputs[""recon""] = self.tokenizer.decode(mle_indices_BTN, (H, W))
        outputs[""lam_indices""] = action_indices_E
        return outputs

    # FIXME (f.srambical): sampling should be moved to the dynamics classes
    def sample(
        self,
        batch: Dict[str, jax.Array],
        seq_len: int,
        steps: int = 25,
        temperature: float = 1,
        sample_argmax: bool = False,
    ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)

        def maskgit_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, mask_BSN, action_tokens_EL = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = self.dynamics.patch_embed(token_idxs_BSN)
            mask_token_111M = self.dynamics.mask_token.value
            mask_expanded_BSN1 = mask_BSN[..., None]
            vid_embed_BSNM = jnp.where(mask_expanded_BSN1, mask_token_111M, vid_embed_BSNM)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = self.dynamics.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNM += act_embed_BS1M
            unmasked_ratio = jnp.cos(jnp.pi * (step + 1) / (steps * 2))
            step_temp = temperature * (1.0 - unmasked_ratio)
            final_logits_BSNV = self.dynamics.transformer(vid_embed_BSNM) / step_temp

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_BSN = jnp.argmax(final_logits_BSNV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_BSN = jax.random.categorical(_rng, final_logits_BSNV)
            gather_fn = jax.vmap(jax.vmap(jax.vmap(lambda x, y: x[y])))
            final_token_probs_BSN = gather_fn(
                jax.nn.softmax(final_logits_BSNV), sampled_token_idxs_BSN
            )
            final_token_probs_BSN += ~mask_BSN
            # Update masked tokens only
            token_idxs_BSN = jnp.where(mask_BSN, sampled_token_idxs_BSN, token_idxs_BSN)

            # --- Update mask ---
            num_unmasked_tokens = jnp.round(N * (1.0 - unmasked_ratio)).astype(int)
            idx_mask_N = jnp.arange(final_token_probs_BSN.shape[-1]) > num_unmasked_tokens
            sorted_idxs_BSN = jnp.argsort(final_token_probs_BSN, axis=-1, descending=True)
            mask_update_fn = jax.vmap(lambda msk, ids: msk.at[ids].set(idx_mask_N))
            new_mask_BSN = mask_update_fn(mask_BSN, sorted_idxs_BSN)

            new_carry = (rng, token_idxs_BSN, new_mask_BSN, action_tokens_EL)
            return new_carry, None

        def generation_step_fn(
            carry: tuple[jax.Array, jax.Array], step_t: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array], None]:
            rng, current_token_idxs_BSN = carry
            rng, step_rng = jax.random.split(rng)

            # Mask current and future frames (i.e., t >= step_t)
            mask_S = jnp.arange(seq_len) >= step_t
            mask_BSN = jnp.broadcast_to(mask_S[None, :, None], (B, seq_len, N)).astype(
                bool
            )
            masked_token_idxs_BSN = current_token_idxs_BSN * ~mask_BSN

            # --- Initialize and run MaskGIT loop ---
            init_carry_maskgit = (
                step_rng,
                masked_token_idxs_BSN,
                mask_BSN,
                action_tokens_EL,
            )
            final_carry_maskgit, _ = jax.lax.scan(
                maskgit_step_fn, init_carry_maskgit, jnp.arange(steps)
            )
            updated_token_idxs_BSN = final_carry_maskgit[1]
            new_carry = (rng, updated_token_idxs_BSN)
            return new_carry, None

        # --- Run the autoregressive generation using jax.lax.scan ---
        initial_carry = (batch[""rng""], token_idxs_BSN)
        timesteps_to_scan = jnp.arange(T, seq_len)
        final_carry, _ = jax.lax.scan(
            generation_step_fn, initial_carry, timesteps_to_scan
        )
        final_token_idxs_BSN = final_carry[1]

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def sample_causal(
            self,
            batch: Dict[str, jax.Array],
            seq_len: int,
            temperature: float = 1,
            sample_argmax: bool = False,
        ) -> jax.Array:
        """"""
        Autoregressively samples up to `seq_len` future frames, following Figure 8 of the paper.

        - Input frames are tokenized once.
        - Future frames are generated autoregressively in token space.
        - All frames are detokenized in a single pass.

        Note:
        - For interactive or step-wise sampling, detokenization should occur after each action.
        - To maintain consistent tensor shapes across timesteps, all current and future frames are decoded at every step.
        - Temporal causal structure is preserved by
            a) reapplying the mask before each decoding step.
            b) a temporal causal mask is applied within each ST-transformer block.

        Dimension keys:
            B: batch size
            T: number of input (conditioning) frames
            N: number of patches per frame
            M: model dimension
            S: sequence length
            H: height
            W: width
            E: B * (S - 1)
        """"""
        # FIXME (f.srambical): reset spatial kv cache after each frame
        assert isinstance(self.dynamics, DynamicsCausal)
        # --- Encode videos and actions ---
        videos_BTHWC = batch[""videos""]
        latent_actions_E = batch[""latent_actions""]
        tokenizer_out = self.tokenizer.vq_encode(videos_BTHWC, training=False)
        token_idxs_BTN = tokenizer_out[""indices""]
        B, T, N = token_idxs_BTN.shape
        pad_shape = (B, seq_len - T, N)
        pad = jnp.zeros(pad_shape, dtype=token_idxs_BTN.dtype)
        token_idxs_BSN = jnp.concatenate([token_idxs_BTN, pad], axis=1)
        action_tokens_EL = self.lam.vq.get_codes(latent_actions_E)
        dynamics_causal: DynamicsCausal = self.dynamics

        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # current_token_idxs_BSN.block_until_ready()
            # start = time.time()
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            # final_carry_causal[1].block_until_ready()
            # elapsed = time.time() - start
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")
            # breakpoint()
            current_token_idxs_BSN = final_carry_causal[1]
        
        final_token_idxs_BSN = current_token_idxs_BSN

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC

    def vq_encode(self, batch: Dict[str, jax.Array], training: bool) -> jax.Array:
        # --- Preprocess videos ---
        video_BTHWC = batch[""videos""]
        lam_output = self.lam.vq_encode(video_BTHWC, training=training)
        lam_indices_E = lam_output[""indices""]
        return lam_indices_E

# FIXME (f.srambical): add conversion script for old checkpoints
def restore_genie_components(
    optimizer: nnx.Optimizer,
    sharding: jax.sharding.NamedSharding,
    rng: jax.Array,
    args,
) -> nnx.Optimizer:
    """"""Restore pre-trained Genie components""""""
    rngs = nnx.Rngs(rng)

    tx = optimizer.tx
    model = optimizer.model
    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )

    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    tokenizer_checkpoint_manager = ocp.CheckpointManager(
        directory=args.tokenizer_checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )
    dummy_tokenizer = TokenizerVQVAE(
        in_dim=args.image_channels,
        model_dim=args.tokenizer_dim,
        ffn_dim=args.tokenizer_ffn_dim,
        latent_dim=args.latent_patch_dim,
        num_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        num_blocks=args.tokenizer_num_blocks,
        num_heads=args.tokenizer_num_heads,
        dropout=args.dropout,
        codebook_dropout=args.dropout,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        rngs=rngs,
    )
    dummy_tokenizer_optimizer = nnx.Optimizer(dummy_tokenizer, tx)
    dummy_tokenizer_optimizer_state = nnx.state(dummy_tokenizer_optimizer)
    abstract_sharded_tokenizer_optimizer_state = _create_abstract_sharded_pytree(
        dummy_tokenizer_optimizer_state, sharding
    )
    restored_tokenizer = tokenizer_checkpoint_manager.restore(
        step=tokenizer_checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(  # type: ignore
                abstract_sharded_tokenizer_optimizer_state  # type: ignore
            ),
        ),
    )[""model_state""]
    nnx.update(dummy_tokenizer_optimizer.model, restored_tokenizer.model)
    model.tokenizer = dummy_tokenizer_optimizer.model
    tokenizer_checkpoint_manager.close()

    if args.lam_checkpoint:
        lam_checkpoint_manager = ocp.CheckpointManager(
            directory=args.lam_checkpoint,
            options=checkpoint_options,
            handler_registry=handler_registry,
        )
        dummy_lam = LatentActionModel(
            in_dim=args.image_channels,
            model_dim=args.lam_dim,
            ffn_dim=args.lam_ffn_dim,
            latent_dim=args.latent_patch_dim,
            num_latents=args.num_latent_actions,
            patch_size=args.lam_patch_size,
            num_blocks=args.lam_num_blocks,
            num_heads=args.lam_num_heads,
            dropout=args.dropout,
            codebook_dropout=args.dropout,
            param_dtype=args.param_dtype,
            dtype=args.dtype,
            use_flash_attention=args.use_flash_attention,
            rngs=rngs,
        )
        dummy_lam_optimizer = nnx.Optimizer(dummy_lam, tx)
        dummy_lam_optimizer_state = nnx.state(dummy_lam_optimizer)
        abstract_sharded_lam_optimizer_state = _create_abstract_sharded_pytree(
            dummy_lam_optimizer_state, sharding
        )
        restored_lam_optimizer = lam_checkpoint_manager.restore(
            step=lam_checkpoint_manager.latest_step(),
            args=ocp.args.Composite(
                model_state=ocp.args.PyTreeRestore(  # type: ignore
                    abstract_sharded_lam_optimizer_state  # type: ignore
                ),
            ),
        )[""model_state""]
        nnx.update(dummy_lam_optimizer.model, restored_lam_optimizer.model)
        model.lam = dummy_lam_optimizer.model
        # Remove the LAM decoder to save memory and avoid unnecessary computation.
        del model.lam.decoder
        lam_checkpoint_manager.close()
    
    # Reinitialize the optimizer states
    optimizer = nnx.Optimizer(model, tx)
    return optimizer


def _create_abstract_sharded_pytree(
    pytree_template: nnx.GraphState, sharding_spec: jax.sharding.NamedSharding
) -> jax.Array:
    """"""Replaces arrays in a pytree with ShapeDtypeStructs having the given sharding.""""""

    def map_fn(leaf_template):
        if hasattr(leaf_template, ""shape"") and hasattr(leaf_template, ""dtype""):
            return jax.ShapeDtypeStruct(
                leaf_template.shape, leaf_template.dtype, sharding=sharding_spec
            )
        return leaf_template

    return jax.tree_util.tree_map(map_fn, pytree_template)
",python,tab
3356,4727590,"/fast/home/franz.srambical/jafar/genie.py",16276,0,"",python,selection_command
3357,4728839,"TERMINAL",0,0,"2025-08-01 10:53:57.582641: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3358,4733348,"TERMINAL",0,0,"2025-08-01 10:54:02.093432: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3359,4742390,"TERMINAL",0,0,"2025-08-01 10:54:11.134885: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3360,4755917,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 393, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
3361,4757201,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3362,4769672,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3363,4772480,"genie.py",0,0,"",python,tab
3364,4772481,"genie.py",14206,0,"",python,selection_command
3365,4774626,"genie.py",14205,0,"",python,selection_mouse
3366,4775007,"genie.py",14205,1,"
",python,selection_mouse
3367,4775538,"genie.py",14206,0,"",python,selection_mouse
3368,4780689,"/fast/home/franz.srambical/jafar/genie.py",0,0,"",python,tab
3369,4781553,"/fast/home/franz.srambical/jafar/genie.py",16215,0,"",python,selection_command
3370,4781810,"/fast/home/franz.srambical/jafar/genie.py",16151,0,"",python,selection_command
3371,4781843,"/fast/home/franz.srambical/jafar/genie.py",16150,0,"",python,selection_command
3372,4781864,"/fast/home/franz.srambical/jafar/genie.py",16100,0,"",python,selection_command
3373,4781895,"/fast/home/franz.srambical/jafar/genie.py",16059,0,"",python,selection_command
3374,4781931,"/fast/home/franz.srambical/jafar/genie.py",16050,0,"",python,selection_command
3375,4781963,"/fast/home/franz.srambical/jafar/genie.py",16002,0,"",python,selection_command
3376,4781997,"/fast/home/franz.srambical/jafar/genie.py",15975,0,"",python,selection_command
3377,4782029,"/fast/home/franz.srambical/jafar/genie.py",15906,0,"",python,selection_command
3378,4782063,"/fast/home/franz.srambical/jafar/genie.py",15905,0,"",python,selection_command
3379,4782096,"/fast/home/franz.srambical/jafar/genie.py",15870,0,"",python,selection_command
3380,4782130,"/fast/home/franz.srambical/jafar/genie.py",15798,0,"",python,selection_command
3381,4782164,"/fast/home/franz.srambical/jafar/genie.py",15797,0,"",python,selection_command
3382,4782197,"/fast/home/franz.srambical/jafar/genie.py",15773,0,"",python,selection_command
3383,4782230,"/fast/home/franz.srambical/jafar/genie.py",15681,0,"",python,selection_command
3384,4782264,"/fast/home/franz.srambical/jafar/genie.py",15643,0,"",python,selection_command
3385,4782296,"/fast/home/franz.srambical/jafar/genie.py",15558,0,"",python,selection_command
3386,4782330,"/fast/home/franz.srambical/jafar/genie.py",15508,0,"",python,selection_command
3387,4782364,"/fast/home/franz.srambical/jafar/genie.py",15490,0,"",python,selection_command
3388,4782398,"/fast/home/franz.srambical/jafar/genie.py",15414,0,"",python,selection_command
3389,4782430,"/fast/home/franz.srambical/jafar/genie.py",15384,0,"",python,selection_command
3390,4782465,"/fast/home/franz.srambical/jafar/genie.py",15328,0,"",python,selection_command
3391,4782497,"/fast/home/franz.srambical/jafar/genie.py",15327,0,"",python,selection_command
3392,4782529,"/fast/home/franz.srambical/jafar/genie.py",15255,0,"",python,selection_command
3393,4782562,"/fast/home/franz.srambical/jafar/genie.py",15143,0,"",python,selection_command
3394,4782597,"/fast/home/franz.srambical/jafar/genie.py",15054,0,"",python,selection_command
3395,4782631,"/fast/home/franz.srambical/jafar/genie.py",14962,0,"",python,selection_command
3396,4782665,"/fast/home/franz.srambical/jafar/genie.py",14883,0,"",python,selection_command
3397,4782697,"/fast/home/franz.srambical/jafar/genie.py",14806,0,"",python,selection_command
3398,4782729,"/fast/home/franz.srambical/jafar/genie.py",14727,0,"",python,selection_command
3399,4782764,"/fast/home/franz.srambical/jafar/genie.py",14686,0,"",python,selection_command
3400,4782796,"/fast/home/franz.srambical/jafar/genie.py",14685,0,"",python,selection_command
3401,4782830,"/fast/home/franz.srambical/jafar/genie.py",14612,0,"",python,selection_command
3402,4782864,"/fast/home/franz.srambical/jafar/genie.py",14565,0,"",python,selection_command
3403,4782897,"/fast/home/franz.srambical/jafar/genie.py",14564,0,"",python,selection_command
3404,4782933,"/fast/home/franz.srambical/jafar/genie.py",14521,0,"",python,selection_command
3405,4782962,"/fast/home/franz.srambical/jafar/genie.py",14477,0,"",python,selection_command
3406,4782997,"/fast/home/franz.srambical/jafar/genie.py",14411,0,"",python,selection_command
3407,4783031,"/fast/home/franz.srambical/jafar/genie.py",14334,0,"",python,selection_command
3408,4783065,"/fast/home/franz.srambical/jafar/genie.py",14246,0,"",python,selection_command
3409,4783095,"/fast/home/franz.srambical/jafar/genie.py",14218,0,"",python,selection_command
3410,4783892,"/fast/home/franz.srambical/jafar/genie.py",14218,0,"+1",python,content
3411,4783896,"/fast/home/franz.srambical/jafar/genie.py",14219,0,"",python,selection_command
3412,4784761,"/fast/home/franz.srambical/jafar/genie.py",14218,2,"",python,content
3413,4785423,"/fast/home/franz.srambical/jafar/genie.py",14217,0,"
",python,content
3414,4785427,"/fast/home/franz.srambical/jafar/genie.py",14218,0,"        ",python,content
3415,4786017,"/fast/home/franz.srambical/jafar/genie.py",14226,0,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

",python,content
3416,4787802,"/fast/home/franz.srambical/jafar/genie.py",14480,0,"",python,selection_command
3417,4787855,"/fast/home/franz.srambical/jafar/genie.py",14381,0,"",python,selection_command
3418,4788095,"/fast/home/franz.srambical/jafar/genie.py",14283,0,"",python,selection_command
3419,4788155,"/fast/home/franz.srambical/jafar/genie.py",14218,0,"",python,selection_command
3420,4788639,"/fast/home/franz.srambical/jafar/genie.py",14234,0,"",python,selection_command
3421,4788893,"/fast/home/franz.srambical/jafar/genie.py",14230,4,"",python,content
3422,4789054,"/fast/home/franz.srambical/jafar/genie.py",14226,4,"",python,content
3423,4789524,"/fast/home/franz.srambical/jafar/genie.py",14225,0,"",python,selection_command
3424,4791110,"/fast/home/franz.srambical/jafar/genie.py",14282,0,"",python,selection_command
3425,4791111,"/fast/home/franz.srambical/jafar/genie.py",14380,0,"",python,selection_command
3426,4791264,"/fast/home/franz.srambical/jafar/genie.py",14472,0,"",python,selection_command
3427,4791772,"/fast/home/franz.srambical/jafar/genie.py",14472,1,"",python,content
3428,4796536,"sample.py",0,0,"",python,tab
3429,4796536,"sample.py",6499,0,"",python,selection_command
3430,4801032,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3431,4802805,"sample.py",0,0,"",python,tab
3432,4804335,"sample.py",840,0,"",python,selection_command
3433,4804798,"sample.py",1758,0,"",python,selection_command
3434,4805381,"sample.py",3018,0,"",python,selection_command
3435,4806040,"sample.py",4288,0,"",python,selection_command
3436,4807010,"sample.py",5717,0,"",python,selection_command
3437,4807506,"sample.py",7359,0,"",python,selection_command
3438,4807740,"sample.py",5717,0,"",python,selection_command
3439,4807932,"sample.py",4288,0,"",python,selection_command
3440,4808352,"sample.py",3018,0,"",python,selection_command
3441,4809085,"sample.py",1758,0,"",python,selection_command
3442,4809237,"sample.py",840,0,"",python,selection_command
3443,4809436,"sample.py",56,0,"",python,selection_command
3444,4811240,"sample.py",840,0,"",python,selection_command
3445,4811430,"sample.py",1758,0,"",python,selection_command
3446,4811739,"sample.py",3018,0,"",python,selection_command
3447,4812238,"sample.py",4288,0,"",python,selection_command
3448,4813709,"sample.py",5717,0,"",python,selection_command
3449,4814181,"sample.py",7359,0,"",python,selection_command
3450,4815868,"sample.py",5717,0,"",python,selection_command
3451,4819564,"sample.py",7359,0,"",python,selection_command
3452,4819805,"sample.py",8462,0,"",python,selection_command
3453,4820344,"sample.py",6855,0,"",python,selection_command
3454,4820871,"sample.py",5345,0,"",python,selection_command
3455,4822154,"sample.py",3888,0,"",python,selection_command
3456,4824972,"sample.py",2700,0,"",python,selection_command
3457,4825308,"sample.py",1527,0,"",python,selection_command
3458,4825475,"sample.py",573,0,"",python,selection_command
3459,4825660,"sample.py",0,0,"",python,selection_command
3460,4826931,"sample.py",0,0,"",python,tab
3461,4830143,"/fast/home/franz.srambical/jafar/genie.py",0,0,"",python,tab
3462,4830607,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3463,4835226,"sample.py",0,0,"",python,tab
3464,4835879,"sample.py",750,0,"",python,selection_command
3465,4836301,"sample.py",2980,0,"",python,selection_command
3466,4842201,"sample.py",4174,0,"",python,selection_command
3467,4843842,"sample.py",5633,0,"",python,selection_keyboard
3468,4844413,"sample.py",5676,0,"",python,selection_command
3469,4844654,"sample.py",5711,0,"",python,selection_command
3470,4844688,"sample.py",5721,0,"",python,selection_command
3471,4844714,"sample.py",5754,0,"",python,selection_command
3472,4844746,"sample.py",5782,0,"",python,selection_command
3473,4844780,"sample.py",5804,0,"",python,selection_command
3474,4844813,"sample.py",5829,0,"",python,selection_command
3475,4844845,"sample.py",5856,0,"",python,selection_command
3476,4844880,"sample.py",5882,0,"",python,selection_command
3477,4844913,"sample.py",5911,0,"",python,selection_command
3478,4844948,"sample.py",6027,0,"",python,selection_command
3479,4844979,"sample.py",6050,0,"",python,selection_command
3480,4845016,"sample.py",6082,0,"",python,selection_command
3481,4845046,"sample.py",6102,0,"",python,selection_command
3482,4845081,"sample.py",6112,0,"",python,selection_command
3483,4845114,"sample.py",6146,0,"",python,selection_command
3484,4845147,"sample.py",6187,0,"",python,selection_command
3485,4845180,"sample.py",6260,0,"",python,selection_command
3486,4845213,"sample.py",6312,0,"",python,selection_command
3487,4845247,"sample.py",6365,0,"",python,selection_command
3488,4845281,"sample.py",6408,0,"",python,selection_command
3489,4845314,"sample.py",6460,0,"",python,selection_command
3490,4845349,"sample.py",6469,0,"",python,selection_command
3491,4845381,"sample.py",6507,0,"",python,selection_command
3492,4846026,"sample.py",6499,43,"    B, S, H, W, _ = video_batch_BSHWC.shape",python,selection_command
3493,4846341,"sample.py",6499,115,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)",python,selection_command
3494,4846416,"sample.py",6499,167,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:",python,selection_command
3495,4846638,"sample.py",6499,255,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)",python,selection_command
3496,4846848,"sample.py",6499,350,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)",python,selection_command
3497,4847116,"sample.py",6499,351,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)
",python,selection_command
3498,4847395,"sample.py",6499,350,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)",python,selection_command
3499,4847623,"sample.py",6499,351,"",python,content
3500,4851462,"sample.py",6499,0,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)
",python,content
3501,4851477,"sample.py",6507,0,"",python,selection_command
3502,4852317,"sample.py",6499,43,"    B, S, H, W, _ = video_batch_BSHWC.shape",python,selection_command
3503,4852601,"sample.py",6499,115,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)",python,selection_command
3504,4852748,"sample.py",6499,167,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:",python,selection_command
3505,4852796,"sample.py",6499,255,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)",python,selection_command
3506,4853064,"sample.py",6499,350,"    B, S, H, W, _ = video_batch_BSHWC.shape
    N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    for block in genie.dynamics.transformer.blocks:
        block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
        block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)",python,selection_command
3507,4853507,"sample.py",6759,0,"# ",python,content
3508,4853507,"sample.py",6671,0,"# ",python,content
3509,4853507,"sample.py",6619,0,"# ",python,content
3510,4853507,"sample.py",6547,0,"# ",python,content
3511,4853507,"sample.py",6503,0,"# ",python,content
3512,4854514,"sample.py",6771,0,"",python,selection_command
3513,4854896,"sample.py",6763,0,"",python,selection_command
3514,4857672,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3515,4857919,"TERMINAL",0,0,"

",,terminal_output
3516,4860545,"genie.py",0,0,"",python,tab
3517,4869254,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3518,4882742,"TERMINAL",0,0,"2025-08-01 10:56:31.483110: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3519,4887266,"TERMINAL",0,0,"2025-08-01 10:56:35.914724: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3520,4896346,"TERMINAL",0,0,"2025-08-01 10:56:45.094092: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3521,4909848,"TERMINAL",0,0,"Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 397, in sample_causal
    block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 652, in init_cache
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 280, in __setattr__
    self._setattr(name, value)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 283, in _setattr
    self._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Cannot mutate 'MultiHeadAttention' from different trace level (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
3522,4911099,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3523,4931508,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3524,4937776,"genie.py",0,0,"",python,tab
3525,4937777,"genie.py",24,0,"",python,selection_command
3526,4943868,"sample.py",0,0,"",python,tab
3527,4943869,"sample.py",6499,0,"",python,selection_command
3528,4957780,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3529,4958183,"genie.py",0,0,"",python,tab
3530,4962554,"genie.py",17029,0,"",python,selection_command
3531,4969510,"genie.py",16979,0,"",python,selection_command
3532,4969762,"genie.py",16945,0,"",python,selection_command
3533,4969787,"genie.py",16888,0,"",python,selection_command
3534,4969818,"genie.py",16887,0,"",python,selection_command
3535,4969850,"genie.py",16873,0,"",python,selection_command
3536,4969945,"genie.py",16821,0,"",python,selection_command
3537,4970205,"genie.py",16787,0,"",python,selection_command
3538,4970222,"genie.py",16747,0,"",python,selection_command
3539,4970255,"genie.py",16721,0,"",python,selection_command
3540,4970290,"genie.py",16687,0,"",python,selection_command
3541,4970357,"genie.py",16634,0,"",python,selection_command
3542,4970517,"genie.py",16633,0,"",python,selection_command
3543,4970660,"genie.py",16531,0,"",python,selection_command
3544,4970798,"genie.py",16470,0,"",python,selection_command
3545,4971696,"genie.py",16470,60,"            for block in dynamics_causal.transformer.blocks:",python,selection_command
3546,4972102,"genie.py",16470,162,"            for block in dynamics_causal.transformer.blocks:
                block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)",python,selection_command
3547,4973431,"genie.py",16543,0,"# ",python,content
3548,4973431,"genie.py",16482,0,"# ",python,content
3549,4973829,"genie.py",16533,0,"",python,selection_command
3550,4975721,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3551,4975996,"TERMINAL",0,0,"

",,terminal_output
3552,4987387,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3553,5005597,"TERMINAL",0,0,"2025-08-01 10:58:34.342846: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3554,5010048,"TERMINAL",0,0,"2025-08-01 10:58:38.791463: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3555,5019109,"TERMINAL",0,0,"2025-08-01 10:58:47.856021: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3556,5035545,"TERMINAL",0,0,"2025-08-01 10:59:04.285600: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3557,5047465,"TERMINAL",0,0,"SSIM: 0.006160452030599117
",,terminal_output
3558,5049367,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3559,5151014,"experiments/sample.sh",0,0,"",shellscript,tab
3560,5153532,"experiments/sample.sh",248,1,"2",shellscript,content
3561,5154395,"experiments/sample.sh",250,0,"",shellscript,selection_command
3562,5156570,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3563,5156865,"TERMINAL",0,0,"

",,terminal_output
3564,5168180,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3565,5177261,"TERMINAL",0,0,"2025-08-01 11:01:25.937437: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3566,5181470,"TERMINAL",0,0,"2025-08-01 11:01:30.216597: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3567,5190300,"TERMINAL",0,0,"2025-08-01 11:01:39.042175: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3568,5194213,"TERMINAL",0,0,"2025-08-01 11:01:42.955476: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3569,5198343,"TERMINAL",0,0,"SSIM: 0.00996059738099575
",,terminal_output
3570,5199883,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3571,7173929,"experiments/sample.sh",327,0,"",shellscript,selection_mouse
3572,7173931,"experiments/sample.sh",326,0,"",shellscript,selection_command
3573,7180860,"experiments/sample.sh",297,0,"",shellscript,selection_command
3574,7180925,"experiments/sample.sh",272,0,"",shellscript,selection_command
3575,7181097,"experiments/sample.sh",250,0,"",shellscript,selection_command
3576,7181478,"experiments/sample.sh",248,0,"",shellscript,selection_command
3577,7182451,"experiments/sample.sh",248,1,"4",shellscript,content
3578,7184084,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3579,7184386,"TERMINAL",0,0,"

",,terminal_output
3580,7195692,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3581,7209119,"TERMINAL",0,0,"2025-08-01 11:35:17.862857: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3582,7213575,"TERMINAL",0,0,"2025-08-01 11:35:22.320153: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3583,7222736,"TERMINAL",0,0,"2025-08-01 11:35:31.442912: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3584,7239045,"TERMINAL",0,0,"2025-08-01 11:35:47.738038: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3585,7250975,"TERMINAL",0,0,"SSIM: 0.006160452030599117
",,terminal_output
3586,7252252,"genie.py",0,0,"",python,tab
3587,7252252,"genie.py",16470,0,"",python,selection_command
3588,7252675,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3589,7277345,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3590,7278077,"experiments/sample.sh",0,0,"",shellscript,tab
3591,7279863,"genie.py",0,0,"",python,tab
3592,7296432,"genie.py",16636,0,"
            ",python,content
3593,7296925,"genie.py",16649,0,"b",python,content
3594,7296925,"genie.py",16650,0,"",python,selection_keyboard
3595,7297145,"genie.py",16650,0,"r",python,content
3596,7297145,"genie.py",16651,0,"",python,selection_keyboard
3597,7297247,"genie.py",16651,0,"e",python,content
3598,7297247,"genie.py",16652,0,"",python,selection_keyboard
3599,7297394,"genie.py",16652,0,"a",python,content
3600,7297394,"genie.py",16653,0,"",python,selection_keyboard
3601,7297395,"genie.py",16653,0,"k",python,content
3602,7297395,"genie.py",16654,0,"",python,selection_keyboard
3603,7297896,"genie.py",16654,0,"p",python,content
3604,7297896,"genie.py",16655,0,"",python,selection_keyboard
3605,7297981,"genie.py",16655,0,"o",python,content
3606,7297982,"genie.py",16656,0,"",python,selection_keyboard
3607,7298058,"genie.py",16656,0,"i",python,content
3608,7298058,"genie.py",16657,0,"",python,selection_keyboard
3609,7298210,"genie.py",16657,0,"n",python,content
3610,7298210,"genie.py",16658,0,"",python,selection_keyboard
3611,7298291,"genie.py",16658,0,"t",python,content
3612,7298292,"genie.py",16659,0,"",python,selection_keyboard
3613,7298733,"genie.py",16659,0,"()",python,content
3614,7298733,"genie.py",16660,0,"",python,selection_keyboard
3615,7298826,"genie.py",16660,1,")",python,content
3616,7298826,"genie.py",16661,0,"",python,selection_keyboard
3617,7299175,"genie.py",16660,0,"",python,selection_command
3618,7300948,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3619,7301226,"TERMINAL",0,0,"

",,terminal_output
3620,7312520,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3621,7320976,"TERMINAL",0,0,"2025-08-01 11:37:09.703023: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3622,7325403,"TERMINAL",0,0,"2025-08-01 11:37:14.148189: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3623,7334473,"TERMINAL",0,0,"2025-08-01 11:37:23.218772: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3624,7337922,"TERMINAL",0,0,"> /fast/home/franz.srambical/jafar/genie.py(402)sample_causal()
-> step_rng,
",,terminal_output
3625,7338827,"utils/nn.py",0,0,"",python,tab
3626,7340232,"genie.py",0,0,"",python,tab
3627,7344251,"TERMINAL",0,0,"t",,terminal_output
3628,7344670,"TERMINAL",0,0," ",,terminal_output
3629,7344823,"TERMINAL",0,0,"ste",,terminal_output
3630,7344878,"TERMINAL",0,0,"p",,terminal_output
3631,7345337,"TERMINAL",0,0,"_",,terminal_output
3632,7345601,"TERMINAL",0,0,"t",,terminal_output
3633,7345813,"TERMINAL",0,0,"
(Pdb) 2
",,terminal_output
3634,7356434,"TERMINAL",0,0,"T",,terminal_output
3635,7356742,"TERMINAL",0,0,"
(Pdb) 2
",,terminal_output
3636,7359136,"experiments/sample.sh",0,0,"",shellscript,tab
3637,7362822,"genie.py",0,0,"",python,tab
3638,7366587,"genie.py",16342,0,"",python,selection_mouse
3639,7373696,"genie.py",13909,0,"",python,selection_mouse
3640,7376425,"genie.py",13863,0,"",python,selection_mouse
3641,7377282,"genie.py",13805,0,"",python,selection_mouse
3642,7379266,"genie.py",13685,0,"",python,selection_mouse
3643,7380873,"genie.py",12293,0,"",python,selection_mouse
3644,7382199,"sample.py",0,0,"",python,tab
3645,7382199,"sample.py",4775,0,"",python,selection_command
3646,10214613,"sample.py",5149,0,"",python,selection_mouse
3647,10214614,"sample.py",5148,0,"",python,selection_command
3648,10215057,"sample.py",5194,0,"",python,selection_mouse
3649,10221251,"sample.py",5724,0,"",python,selection_mouse
3650,10223250,"sample.py",6151,0,"",python,selection_mouse
3651,10233813,"genie.py",0,0,"",python,tab
3652,10253962,"genie.py",13891,0,"",python,selection_mouse
3653,10255656,"genie.py",13908,0,"",python,selection_mouse
3654,10256543,"genie.py",13846,0,"",python,selection_mouse
3655,10257925,"genie.py",13862,0,"",python,selection_mouse
3656,10259041,"genie.py",13810,0,"",python,selection_mouse
3657,10261143,"genie.py",12297,0,"",python,selection_mouse
3658,10262649,"sample.py",0,0,"",python,tab
3659,10266249,"sample.py",5460,0,"",python,selection_mouse
3660,10268340,"sample.py",4781,0,"",python,selection_mouse
3661,10270118,"sample.py",4808,0,"",python,selection_mouse
3662,10271187,"sample.py",4336,0,"",python,selection_mouse
3663,10272379,"sample.py",5478,0,"",python,selection_mouse
3664,10274390,"sample.py",5374,0,"",python,selection_mouse
3665,10275936,"sample.py",5237,0,"",python,selection_mouse
3666,10282878,"sample.py",5258,0,"",python,selection_mouse
3667,10350067,"sample.py",3818,0,"",python,selection_command
3668,10350497,"sample.py",2647,0,"",python,selection_command
3669,10351045,"sample.py",1495,0,"",python,selection_command
3670,10351415,"sample.py",502,0,"",python,selection_command
3671,10351718,"sample.py",0,0,"",python,selection_command
3672,10352311,"sample.py",750,0,"",python,selection_command
3673,10352656,"sample.py",1714,0,"",python,selection_command
3674,10353310,"sample.py",2980,0,"",python,selection_command
3675,10355486,"sample.py",3001,0,"",python,selection_command
3676,10355745,"sample.py",3016,0,"",python,selection_command
3677,10355765,"sample.py",3018,0,"",python,selection_command
3678,10355799,"sample.py",3027,0,"",python,selection_command
3679,10355825,"sample.py",3098,0,"",python,selection_command
3680,10355864,"sample.py",3124,0,"",python,selection_command
3681,10355895,"sample.py",3201,0,"",python,selection_command
3682,10355930,"sample.py",3211,0,"",python,selection_command
3683,10355964,"sample.py",3237,0,"",python,selection_command
3684,10355994,"sample.py",3317,0,"",python,selection_command
3685,10356030,"sample.py",3327,0,"",python,selection_command
3686,10356064,"sample.py",3382,0,"",python,selection_command
3687,10356096,"sample.py",3414,0,"",python,selection_command
3688,10356130,"sample.py",3424,0,"",python,selection_command
3689,10356162,"sample.py",3472,0,"",python,selection_command
3690,10356194,"sample.py",3497,0,"",python,selection_command
3691,10356230,"sample.py",3533,0,"",python,selection_command
3692,10356261,"sample.py",3572,0,"",python,selection_command
3693,10356296,"sample.py",3574,0,"",python,selection_command
3694,10356330,"sample.py",3583,0,"",python,selection_command
3695,10356362,"sample.py",3611,0,"",python,selection_command
3696,10356394,"sample.py",3679,0,"",python,selection_command
3697,10356429,"sample.py",3695,0,"",python,selection_command
3698,10356463,"sample.py",3711,0,"",python,selection_command
3699,10356497,"sample.py",3738,0,"",python,selection_command
3700,10356528,"sample.py",3763,0,"",python,selection_command
3701,10356562,"sample.py",3773,0,"",python,selection_command
3702,10356605,"sample.py",3818,0,"",python,selection_command
3703,10356635,"sample.py",3827,0,"",python,selection_command
3704,10356670,"sample.py",3892,0,"",python,selection_command
3705,10356695,"sample.py",3953,0,"",python,selection_command
3706,10356730,"sample.py",3996,0,"",python,selection_command
3707,10356761,"sample.py",4038,0,"",python,selection_command
3708,10357288,"sample.py",4071,0,"",python,selection_command
3709,10357469,"sample.py",4161,0,"",python,selection_command
3710,10357624,"sample.py",4168,0,"",python,selection_command
3711,10357885,"sample.py",4178,0,"",python,selection_command
3712,10357898,"sample.py",4233,0,"",python,selection_command
3713,10357931,"sample.py",4283,0,"",python,selection_command
3714,10357964,"sample.py",4292,0,"",python,selection_command
3715,10357998,"sample.py",4331,0,"",python,selection_command
3716,10358031,"sample.py",4393,0,"",python,selection_command
3717,10358066,"sample.py",4471,0,"",python,selection_command
3718,10358094,"sample.py",4511,0,"",python,selection_command
3719,10358130,"sample.py",4544,0,"",python,selection_command
3720,10358163,"sample.py",4567,0,"",python,selection_command
3721,10358198,"sample.py",4597,0,"",python,selection_command
3722,10358231,"sample.py",4633,0,"",python,selection_command
3723,10358266,"sample.py",4667,0,"",python,selection_command
3724,10358298,"sample.py",4703,0,"",python,selection_command
3725,10358333,"sample.py",4717,0,"",python,selection_command
3726,10358366,"sample.py",4758,0,"",python,selection_command
3727,10358398,"sample.py",4798,0,"",python,selection_command
3728,10358432,"sample.py",4821,0,"",python,selection_command
3729,10358466,"sample.py",4851,0,"",python,selection_command
3730,10358499,"sample.py",4885,0,"",python,selection_command
3731,10358532,"sample.py",4921,0,"",python,selection_command
3732,10358566,"sample.py",4935,0,"",python,selection_command
3733,10358599,"sample.py",4949,0,"",python,selection_command
3734,10358631,"sample.py",5014,0,"",python,selection_command
3735,10358666,"sample.py",5023,0,"",python,selection_command
3736,10358698,"sample.py",5073,0,"",python,selection_command
3737,10358731,"sample.py",5144,0,"",python,selection_command
3738,10358765,"sample.py",5158,0,"",python,selection_command
3739,10358798,"sample.py",5230,0,"",python,selection_command
3740,10358831,"sample.py",5303,0,"",python,selection_command
3741,10359208,"sample.py",5230,0,"",python,selection_command
3742,10385817,"genie.py",0,0,"",python,tab
3743,10393305,"genie.py",16342,0,"",python,selection_mouse
3744,10406237,"experiments/sample.sh",0,0,"",shellscript,tab
3745,10848720,"sample.py",0,0,"",python,tab
3746,10849662,"sample.py",5248,0,"",python,selection_command
3747,10849907,"sample.py",5250,0,"",python,selection_command
3748,10849937,"sample.py",5267,0,"",python,selection_command
3749,10849970,"sample.py",5271,0,"",python,selection_command
3750,10850002,"sample.py",5273,0,"",python,selection_command
3751,10850035,"sample.py",5277,0,"",python,selection_command
3752,10850069,"sample.py",5278,0,"",python,selection_command
3753,10850326,"sample.py",5290,0,"",python,selection_command
3754,10850893,"sample.py",5289,0,"",python,selection_command
3755,10850999,"sample.py",5289,1," ",python,selection_command
3756,10851073,"sample.py",5289,2," +",python,selection_command
3757,10851549,"sample.py",5289,4," + 1",python,selection_command
3758,10851836,"sample.py",5289,4,"",python,content
3759,10862514,"genie.py",0,0,"",python,tab
3760,10940827,"sample.py",0,0,"",python,tab
3761,10942434,"genie.py",0,0,"",python,tab
3762,10943863,"experiments/sample.sh",0,0,"",shellscript,tab
3763,10955706,"experiments/sample.sh",248,1,"2",shellscript,content
3764,10985836,"experiments/sample.sh",249,0,"",shellscript,selection_command
3765,10986690,"experiments/sample.sh",248,0,"",shellscript,selection_command
3766,10987272,"genie.py",0,0,"",python,tab
3767,10991740,"TERMINAL",0,0,"q",,terminal_output
3768,10991849,"TERMINAL",0,0,"ui",,terminal_output
3769,10992357,"TERMINAL",0,0,"t",,terminal_output
3770,10992811,"TERMINAL",0,0,"(",,terminal_output
3771,10992913,"TERMINAL",0,0,")",,terminal_output
3772,10993329,"TERMINAL",0,0,"
Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 402, in sample_causal
    step_rng,
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 402, in sample_causal
    step_rng,
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 90, in trace_dispatch
    return self.dispatch_line(frame)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 115, in dispatch_line
    if self.quitting: raise BdbQuit
bdb.BdbQuit
",,terminal_output
3773,10994062,"TERMINAL",0,0,"(Pdb) ",,terminal_output
3774,10994294,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3775,10994571,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3776,10994927,"TERMINAL",0,0,"

",,terminal_output
3777,11007223,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3778,11026148,"TERMINAL",0,0,"2025-08-01 12:38:54.892781: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3779,11030432,"TERMINAL",0,0,"2025-08-01 12:38:59.173950: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3780,11038588,"TERMINAL",0,0,"2025-08-01 12:39:07.300365: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3781,11042135,"TERMINAL",0,0,"> /fast/home/franz.srambical/jafar/genie.py(402)sample_causal()
-> step_rng,
",,terminal_output
3782,11044240,"TERMINAL",0,0,"T",,terminal_output
3783,11044574,"TERMINAL",0,0,"
(Pdb) 1
",,terminal_output
3784,11072241,"TERMINAL",0,0,"c",,terminal_output
3785,11072445,"TERMINAL",0,0,"
",,terminal_output
3786,11081788,"TERMINAL",0,0,"2025-08-01 12:39:50.529033: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3787,11091708,"TERMINAL",0,0,"(Pdb) SSIM: 0.00755678117275238
",,terminal_output
3788,11093358,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3789,11247086,"sample.py",0,0,"",python,tab
3790,11248779,"sample.py",5217,0,"",python,selection_command
3791,11248931,"sample.py",5148,0,"",python,selection_command
3792,11250054,"sample.py",5069,0,"",python,selection_command
3793,11250910,"sample.py",5140,0,"",python,selection_command
3794,11250910,"sample.py",5141,0,"",python,selection_command
3795,11251257,"sample.py",5142,0,"",python,selection_command
3796,11251427,"sample.py",5143,0,"",python,selection_command
3797,11251819,"sample.py",5140,0,"",python,selection_command
3798,11252093,"sample.py",5140,1,"",python,content
3799,11253524,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3800,11254754,"TERMINAL",0,0,"

",,terminal_output
3801,11256237,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.39 task 0: running
^Csrun: sending Ctrl-C to StepId=14571.39
srun: forcing job termination
Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 11, in <module>
    import flax.linen as nn
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/__init__.py"", line 26, in <module>
    from flax import linen
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/linen/__init__.py"", line 127, in <module>
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
    from .recurrent import (
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/linen/recurrent.py"", line 81, in <module>
[2025-08-01T12:42:44.960] error: *** STEP 14571.39 ON hai005 CANCELLED AT 2025-08-01T12:42:44 DUE to SIGNAL Killed ***
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3802,11256338,"TERMINAL",0,0,"^C


[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3803,11257988,"genie.py",0,0,"",python,tab
3804,11259219,"genie.py",16383,0,"",python,selection_command
3805,11259264,"genie.py",16405,0,"",python,selection_command
3806,11259406,"genie.py",16434,0,"",python,selection_command
3807,11259550,"genie.py",16498,0,"",python,selection_command
3808,11259680,"genie.py",16561,0,"",python,selection_command
3809,11259833,"genie.py",16660,0,"",python,selection_command
3810,11260855,"genie.py",16649,0,"",python,selection_command
3811,11262329,"genie.py",16649,0,"#",python,content
3812,11262329,"genie.py",16650,0,"",python,selection_keyboard
3813,11263002,"genie.py",16649,0,"",python,selection_command
3814,11264734,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
3815,11265268,"TERMINAL",0,0,"

",,terminal_output
3816,11276385,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
3817,11285314,"TERMINAL",0,0,"2025-08-01 12:43:14.062337: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
3818,11289531,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 129, in __call__
    pure_args_out, pure_kwargs_out, pure_out = extract.to_tree(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/extract.py"", line 234, in to_tree
    check_consistent_aliasing(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/extract.py"", line 57, in check_consistent_aliasing
    value._check_valid_context(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/object.py"", line 302, in _check_valid_context
    raise errors.TraceContextError(error_msg())
flax.errors.TraceContextError: Trying to extract graph node from different trace level, got Linear( # Param: 1,050,624 (4.2 MB)
  bias=Param( # 2,048 (8.2 KB)
    value=Array(shape=(2048,), dtype=dtype('float32'))
  ),
  bias_init=<function zeros at 0x7f26afd31510>,
  dot_general=<function dot_general at 0x7f26b052e290>,
  dtype=bfloat16,
  in_features=512,
  kernel=Param( # 1,048,576 (4.2 MB)
    value=Array(shape=(512, 2048), dtype=dtype('float32'))
  ),
  kernel_init=<function variance_scaling.<locals>.init at 0x7f26aac0b9a0>,
  out_features=2048,
  param_dtype=float32,
  precision=None,
  promote_dtype=<function promote_dtype at 0x7f26aac0b520>,
  use_bias=True
) (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.TraceContextError)
",,terminal_output
3819,11290518,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3820,11300066,"TERMINAL",0,0,"'",,terminal_output
3821,11300212,"TERMINAL",0,0,"",,terminal_output
3822,11301439,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3823,11306105,"TERMINAL",0,0,"
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
3824,11314339,"sample.py",0,0,"",python,tab
3825,11315130,"sample.py",5140,0,"#",python,content
3826,11315136,"sample.py",5140,0,"",python,selection_command
3827,11318139,"genie.py",0,0,"",python,tab
3828,11340249,"genie.py",16545,0,"",python,selection_command
3829,11340292,"genie.py",16482,0,"",python,selection_command
3830,11340548,"genie.py",16418,0,"",python,selection_command
3831,11340678,"genie.py",16405,0,"",python,selection_command
3832,11340737,"genie.py",16367,0,"",python,selection_command
3833,11340737,"genie.py",16326,0,"",python,selection_command
3834,11340738,"genie.py",16312,0,"",python,selection_command
3835,11340738,"genie.py",16269,0,"",python,selection_command
3836,11340811,"genie.py",16242,0,"",python,selection_command
3837,11340812,"genie.py",16173,0,"",python,selection_command
3838,11340845,"genie.py",16160,0,"",python,selection_command
3839,11340876,"genie.py",16137,0,"",python,selection_command
3840,11340880,"genie.py",16065,0,"",python,selection_command
3841,11340914,"genie.py",16052,0,"",python,selection_command
3842,11340941,"genie.py",16040,0,"",python,selection_command
3843,11340973,"genie.py",15948,0,"",python,selection_command
3844,11341007,"genie.py",15910,0,"",python,selection_command
3845,11341042,"genie.py",15825,0,"",python,selection_command
3846,11341076,"genie.py",15775,0,"",python,selection_command
3847,11341107,"genie.py",15757,0,"",python,selection_command
3848,11341140,"genie.py",15681,0,"",python,selection_command
3849,11341174,"genie.py",15651,0,"",python,selection_command
3850,11341209,"genie.py",15595,0,"",python,selection_command
3851,11341242,"genie.py",15582,0,"",python,selection_command
3852,11341275,"genie.py",15522,0,"",python,selection_command
3853,11341307,"genie.py",15410,0,"",python,selection_command
3854,11341340,"genie.py",15321,0,"",python,selection_command
3855,11341374,"genie.py",15229,0,"",python,selection_command
3856,11341409,"genie.py",15150,0,"",python,selection_command
3857,11341441,"genie.py",15073,0,"",python,selection_command
3858,11341474,"genie.py",14994,0,"",python,selection_command
3859,11341506,"genie.py",14953,0,"",python,selection_command
3860,11341541,"genie.py",14940,0,"",python,selection_command
3861,11341575,"genie.py",14879,0,"",python,selection_command
3862,11341608,"genie.py",14832,0,"",python,selection_command
3863,11341643,"genie.py",14819,0,"",python,selection_command
3864,11341676,"genie.py",14788,0,"",python,selection_command
3865,11341709,"genie.py",14744,0,"",python,selection_command
3866,11341743,"genie.py",14678,0,"",python,selection_command
3867,11341775,"genie.py",14601,0,"",python,selection_command
3868,11341809,"genie.py",14513,0,"",python,selection_command
3869,11341843,"genie.py",14485,0,"",python,selection_command
3870,11341877,"genie.py",14472,0,"",python,selection_command
3871,11341913,"genie.py",14385,0,"",python,selection_command
3872,11341943,"genie.py",14287,0,"",python,selection_command
3873,11341977,"genie.py",14230,0,"",python,selection_command
3874,11342011,"genie.py",14217,0,"",python,selection_command
3875,11342043,"genie.py",14173,0,"",python,selection_command
3876,11342076,"genie.py",14106,0,"",python,selection_command
3877,11342108,"genie.py",14034,0,"",python,selection_command
3878,11342142,"genie.py",13971,0,"",python,selection_command
3879,11342174,"genie.py",13931,0,"",python,selection_command
3880,11342209,"genie.py",13892,0,"",python,selection_command
3881,11342243,"genie.py",13842,0,"",python,selection_command
3882,11342276,"genie.py",13763,0,"",python,selection_command
3883,11342309,"genie.py",13712,0,"",python,selection_command
3884,11342342,"genie.py",13673,0,"",python,selection_command
3885,11342375,"genie.py",13629,0,"",python,selection_command
3886,11342411,"genie.py",13572,0,"",python,selection_command
3887,11342443,"genie.py",13501,0,"",python,selection_command
3888,11342477,"genie.py",13487,0,"",python,selection_command
3889,11342512,"genie.py",13462,0,"",python,selection_command
3890,11342553,"genie.py",13441,0,"",python,selection_command
3891,11342587,"genie.py",13419,0,"",python,selection_command
3892,11342610,"genie.py",13388,0,"",python,selection_command
3893,11342642,"genie.py",13357,0,"",python,selection_command
3894,11342677,"genie.py",13314,0,"",python,selection_command
3895,11343368,"genie.py",13261,0,"",python,selection_command
3896,11343415,"genie.py",13235,0,"",python,selection_command
3897,11343670,"genie.py",13211,0,"",python,selection_command
3898,11343697,"genie.py",13198,0,"",python,selection_command
3899,11343728,"genie.py",13127,0,"",python,selection_command
3900,11343762,"genie.py",13065,0,"",python,selection_command
3901,11343795,"genie.py",13013,0,"",python,selection_command
3902,11343829,"genie.py",12891,0,"",python,selection_command
3903,11343863,"genie.py",12795,0,"",python,selection_command
3904,11343895,"genie.py",12781,0,"",python,selection_command
3905,11343931,"genie.py",12768,0,"",python,selection_command
3906,11343963,"genie.py",12725,0,"",python,selection_command
3907,11343995,"genie.py",12654,0,"",python,selection_command
3908,11344030,"genie.py",12611,0,"",python,selection_command
3909,11344061,"genie.py",12598,0,"",python,selection_command
3910,11344095,"genie.py",12513,0,"",python,selection_command
3911,11344129,"genie.py",12499,0,"",python,selection_command
3912,11344162,"genie.py",12477,0,"",python,selection_command
3913,11344194,"genie.py",12436,0,"",python,selection_command
3914,11344630,"genie.py",12477,0,"",python,selection_command
3915,11344715,"genie.py",12499,0,"",python,selection_command
3916,11344971,"genie.py",12513,0,"",python,selection_command
3917,11344997,"genie.py",12598,0,"",python,selection_command
3918,11345025,"genie.py",12611,0,"",python,selection_command
3919,11345058,"genie.py",12654,0,"",python,selection_command
3920,11345090,"genie.py",12725,0,"",python,selection_command
3921,11345124,"genie.py",12768,0,"",python,selection_command
3922,11345158,"genie.py",12781,0,"",python,selection_command
3923,11345193,"genie.py",12795,0,"",python,selection_command
3924,11345225,"genie.py",12891,0,"",python,selection_command
3925,11345258,"genie.py",13013,0,"",python,selection_command
3926,11345291,"genie.py",13065,0,"",python,selection_command
3927,11345323,"genie.py",13127,0,"",python,selection_command
3928,11345358,"genie.py",13198,0,"",python,selection_command
3929,11345390,"genie.py",13211,0,"",python,selection_command
3930,11345425,"genie.py",13235,0,"",python,selection_command
3931,11345457,"genie.py",13261,0,"",python,selection_command
3932,11345491,"genie.py",13314,0,"",python,selection_command
3933,11345524,"genie.py",13357,0,"",python,selection_command
3934,11345559,"genie.py",13388,0,"",python,selection_command
3935,11345592,"genie.py",13419,0,"",python,selection_command
3936,11345626,"genie.py",13441,0,"",python,selection_command
3937,11345680,"genie.py",13462,0,"",python,selection_command
3938,11345705,"genie.py",13487,0,"",python,selection_command
3939,11345733,"genie.py",13501,0,"",python,selection_command
3940,11345757,"genie.py",13572,0,"",python,selection_command
3941,11345788,"genie.py",13629,0,"",python,selection_command
3942,11345825,"genie.py",13673,0,"",python,selection_command
3943,11345857,"genie.py",13712,0,"",python,selection_command
3944,11345889,"genie.py",13763,0,"",python,selection_command
3945,11345923,"genie.py",13842,0,"",python,selection_command
3946,11345959,"genie.py",13892,0,"",python,selection_command
3947,11345990,"genie.py",13931,0,"",python,selection_command
3948,11346026,"genie.py",13971,0,"",python,selection_command
3949,11346057,"genie.py",14034,0,"",python,selection_command
3950,11346090,"genie.py",14106,0,"",python,selection_command
3951,11346126,"genie.py",14173,0,"",python,selection_command
3952,11346159,"genie.py",14217,0,"",python,selection_command
3953,11346193,"genie.py",14230,0,"",python,selection_command
3954,11346231,"genie.py",14287,0,"",python,selection_command
3955,11346264,"genie.py",14385,0,"",python,selection_command
3956,11346292,"genie.py",14472,0,"",python,selection_command
3957,11346324,"genie.py",14485,0,"",python,selection_command
3958,11346357,"genie.py",14513,0,"",python,selection_command
3959,11346389,"genie.py",14601,0,"",python,selection_command
3960,11346426,"genie.py",14678,0,"",python,selection_command
3961,11346457,"genie.py",14744,0,"",python,selection_command
3962,11346491,"genie.py",14788,0,"",python,selection_command
3963,11346526,"genie.py",14819,0,"",python,selection_command
3964,11346558,"genie.py",14832,0,"",python,selection_command
3965,11346591,"genie.py",14879,0,"",python,selection_command
3966,11346626,"genie.py",14940,0,"",python,selection_command
3967,11346658,"genie.py",14953,0,"",python,selection_command
3968,11346691,"genie.py",14994,0,"",python,selection_command
3969,11346724,"genie.py",15073,0,"",python,selection_command
3970,11346759,"genie.py",15150,0,"",python,selection_command
3971,11346792,"genie.py",15229,0,"",python,selection_command
3972,11346826,"genie.py",15321,0,"",python,selection_command
3973,11346858,"genie.py",15410,0,"",python,selection_command
3974,11346893,"genie.py",15522,0,"",python,selection_command
3975,11346926,"genie.py",15582,0,"",python,selection_command
3976,11346966,"genie.py",15595,0,"",python,selection_command
3977,11346994,"genie.py",15651,0,"",python,selection_command
3978,11347026,"genie.py",15681,0,"",python,selection_command
3979,11347079,"genie.py",15757,0,"",python,selection_command
3980,11347102,"genie.py",15775,0,"",python,selection_command
3981,11347132,"genie.py",15825,0,"",python,selection_command
3982,11347160,"genie.py",15910,0,"",python,selection_command
3983,11347193,"genie.py",15948,0,"",python,selection_command
3984,11347223,"genie.py",16040,0,"",python,selection_command
3985,11347258,"genie.py",16052,0,"",python,selection_command
3986,11347296,"genie.py",16065,0,"",python,selection_command
3987,11347324,"genie.py",16137,0,"",python,selection_command
3988,11347358,"genie.py",16160,0,"",python,selection_command
3989,11347394,"genie.py",16173,0,"",python,selection_command
3990,11347427,"genie.py",16242,0,"",python,selection_command
3991,11348420,"genie.py",16173,0,"",python,selection_command
3992,11349061,"genie.py",16160,0,"",python,selection_command
3993,11349136,"genie.py",16137,0,"",python,selection_command
3994,11349392,"genie.py",16065,0,"",python,selection_command
3995,11349418,"genie.py",16052,0,"",python,selection_command
3996,11349450,"genie.py",16040,0,"",python,selection_command
3997,11349541,"genie.py",15948,0,"",python,selection_command
3998,11349542,"genie.py",15910,0,"",python,selection_command
3999,11349574,"genie.py",15825,0,"",python,selection_command
4000,11349587,"genie.py",15775,0,"",python,selection_command
4001,11349614,"genie.py",15757,0,"",python,selection_command
4002,11349648,"genie.py",15681,0,"",python,selection_command
4003,11349682,"genie.py",15651,0,"",python,selection_command
4004,11349716,"genie.py",15595,0,"",python,selection_command
4005,11349749,"genie.py",15582,0,"",python,selection_command
4006,11349782,"genie.py",15522,0,"",python,selection_command
4007,11349816,"genie.py",15410,0,"",python,selection_command
4008,11349848,"genie.py",15321,0,"",python,selection_command
4009,11349883,"genie.py",15229,0,"",python,selection_command
4010,11349916,"genie.py",15150,0,"",python,selection_command
4011,11349949,"genie.py",15073,0,"",python,selection_command
4012,11349983,"genie.py",14994,0,"",python,selection_command
4013,11350016,"genie.py",14953,0,"",python,selection_command
4014,11350049,"genie.py",14940,0,"",python,selection_command
4015,11350083,"genie.py",14879,0,"",python,selection_command
4016,11350117,"genie.py",14832,0,"",python,selection_command
4017,11350150,"genie.py",14819,0,"",python,selection_command
4018,11350182,"genie.py",14788,0,"",python,selection_command
4019,11350217,"genie.py",14744,0,"",python,selection_command
4020,11350249,"genie.py",14678,0,"",python,selection_command
4021,11350282,"genie.py",14601,0,"",python,selection_command
4022,11350316,"genie.py",14513,0,"",python,selection_command
4023,11350349,"genie.py",14485,0,"",python,selection_command
4024,11350386,"genie.py",14472,0,"",python,selection_command
4025,11350773,"genie.py",14485,0,"",python,selection_command
4026,11352803,"genie.py",14472,0,"
",python,content
4027,11352824,"genie.py",14473,0,"        ",python,content
4028,11354408,"genie.py",14481,0,"@nnx.jit",python,content
4029,11354625,"genie.py",14488,0,"",python,selection_command
4030,11356522,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4031,11356817,"TERMINAL",0,0,"

",,terminal_output
4032,11368175,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4033,11381948,"TERMINAL",0,0,"2025-08-01 12:44:50.691467: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4034,11386225,"TERMINAL",0,0,"2025-08-01 12:44:54.974791: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4035,11394339,"TERMINAL",0,0,"2025-08-01 12:45:03.089828: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4036,11408214,"TERMINAL",0,0,"2025-08-01 12:45:16.959325: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4037,11418042,"TERMINAL",0,0,"SSIM: 0.00755678117275238
",,terminal_output
4038,11419704,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4039,11652662,"sample.py",0,0,"",python,tab
4040,11655274,"sample.py",5154,0,"",python,selection_command
4041,11655683,"sample.py",5140,0,"",python,selection_command
4042,11655922,"sample.py",5069,0,"",python,selection_command
4043,11655956,"sample.py",5019,0,"",python,selection_command
4044,11655986,"sample.py",5014,0,"",python,selection_command
4045,11656021,"sample.py",4945,0,"",python,selection_command
4046,11656051,"sample.py",4931,0,"",python,selection_command
4047,11656084,"sample.py",4917,0,"",python,selection_command
4048,11656120,"sample.py",4881,0,"",python,selection_command
4049,11656152,"sample.py",4847,0,"",python,selection_command
4050,11656185,"sample.py",4817,0,"",python,selection_command
4051,11656217,"sample.py",4794,0,"",python,selection_command
4052,11656252,"sample.py",4754,0,"",python,selection_command
4053,11656286,"sample.py",4713,0,"",python,selection_command
4054,11656317,"sample.py",4699,0,"",python,selection_command
4055,11656351,"sample.py",4663,0,"",python,selection_command
4056,11656386,"sample.py",4629,0,"",python,selection_command
4057,11656417,"sample.py",4593,0,"",python,selection_command
4058,11656451,"sample.py",4563,0,"",python,selection_command
4059,11656483,"sample.py",4540,0,"",python,selection_command
4060,11656519,"sample.py",4507,0,"",python,selection_command
4061,11656552,"sample.py",4467,0,"",python,selection_command
4062,11656586,"sample.py",4389,0,"",python,selection_command
4063,11656619,"sample.py",4327,0,"",python,selection_command
4064,11656784,"sample.py",4288,0,"",python,selection_command
4065,11657535,"sample.py",4284,38,"    # --- Define sampling function ---",python,selection_command
4066,11659800,"sample.py",4284,730,"    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")
",python,selection_command
4067,11665330,"sample.py",4284,1229,"    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC
",python,selection_command
4068,11665481,"sample.py",4284,2172,"    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)
",python,selection_command
4069,11666033,"sample.py",4284,2572,"    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)

    # --- Sample + evaluate video ---
    # B, S, H, W, _ = video_batch_BSHWC.shape
    # N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    # for block in genie.dynamics.transformer.blocks:
    #     block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
    #     block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)
",python,selection_command
4070,11666882,"sample.py",4284,3046,"    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)

    # --- Sample + evaluate video ---
    # B, S, H, W, _ = video_batch_BSHWC.shape
    # N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    # for block in genie.dynamics.transformer.blocks:
    #     block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
    #     block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)

    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
    recon_video_BSHWC = recon_video_BSHWC.astype(jnp.float32)
    gt = gt_video[:, : recon_video_BSHWC.shape[1]].clip(0, 1).reshape(-1, *gt_video.shape[2:])
    recon = recon_video_BSHWC.clip(0, 1).reshape(-1, *recon_video_BSHWC.shape[2:])
    ssim = jnp.asarray(
        pix.ssim(gt[:, args.start_frame + 1 :], recon[:, args.start_frame + 1 :])
    ).mean()
    print(f""SSIM: {ssim}"")
",python,selection_command
4071,11875574,"experiments/sample.sh",0,0,"",shellscript,tab
4072,11877001,"experiments/sample.sh",248,1,"3",shellscript,content
4073,11878381,"experiments/sample.sh",248,1,"4",shellscript,content
4074,11881267,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4075,11881607,"TERMINAL",0,0,"

",,terminal_output
4076,11892969,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4077,11901446,"TERMINAL",0,0,"2025-08-01 12:53:30.189597: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4078,11905938,"TERMINAL",0,0,"2025-08-01 12:53:34.676757: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4079,11914093,"TERMINAL",0,0,"2025-08-01 12:53:42.840693: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4080,11933584,"TERMINAL",0,0,"2025-08-01 12:54:02.332971: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4081,11945529,"TERMINAL",0,0,"SSIM: 0.004542595706880093
",,terminal_output
4082,11947191,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4083,12102809,"genie.py",0,0,"",python,tab
4084,12104160,"genie.py",14473,16,"        @nnx.jit",python,selection_command
4085,12106242,"genie.py",14488,0,"",python,selection_command
4086,12106653,"genie.py",14472,0,"",python,selection_command
4087,12106715,"genie.py",14388,0,"",python,selection_command
4088,12106876,"genie.py",14290,0,"",python,selection_command
4089,12107194,"genie.py",14233,0,"",python,selection_command
4090,12107971,"genie.py",14218,56,"        for block in dynamics_causal.transformer.blocks:",python,selection_command
4091,12110743,"genie.py",14217,57,"
        for block in dynamics_causal.transformer.blocks:",python,selection_command
4092,12111987,"genie.py",14218,56,"        for block in dynamics_causal.transformer.blocks:",python,selection_command
4093,12112426,"genie.py",14218,254,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)
",python,selection_command
4094,12112903,"genie.py",14218,618,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]
",python,selection_command
4095,12113040,"genie.py",14218,739,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)
",python,selection_command
4096,12113524,"genie.py",14218,1381,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]
",python,selection_command
4097,12113759,"genie.py",14218,1851,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1
",python,selection_command
4098,12113930,"genie.py",14218,1959,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None
",python,selection_command
4099,12114069,"genie.py",14218,2204,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)
",python,selection_command
4100,12114233,"genie.py",14218,2462,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            # for block in dynamics_causal.transformer.blocks:
            #     block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            #breakpoint()
",python,selection_command
4101,12114613,"genie.py",14218,2716,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            # for block in dynamics_causal.transformer.blocks:
            #     block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            #breakpoint()

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )
",python,selection_command
4102,12114913,"genie.py",14218,3257,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            # for block in dynamics_causal.transformer.blocks:
            #     block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            #breakpoint()

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # current_token_idxs_BSN.block_until_ready()
            # start = time.time()
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            # final_carry_causal[1].block_until_ready()
            # elapsed = time.time() - start
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")
            # breakpoint()
            current_token_idxs_BSN = final_carry_causal[1]
        
        final_token_idxs_BSN = current_token_idxs_BSN
",python,selection_command
4103,12115223,"genie.py",14218,3514,"        for block in dynamics_causal.transformer.blocks:
            block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), 1, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            # for block in dynamics_causal.transformer.blocks:
            #     block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            #breakpoint()

            # --- Initialize and run causal loop ---
            init_carry_causal = (
                step_rng,
                current_token_idxs_BSN,
                action_tokens_EL,
                jnp.array(step_t, dtype=jnp.int32),
            )

            # current_token_idxs_BSN.block_until_ready()
            # start = time.time()
            final_carry_causal, _ = jax.lax.scan(
                causal_step_fn, init_carry_causal, jnp.arange(N)
            )
            # final_carry_causal[1].block_until_ready()
            # elapsed = time.time() - start
            # print(f""Autoregressive generation time: {elapsed:.4f}s"")
            # breakpoint()
            current_token_idxs_BSN = final_carry_causal[1]
        
        final_token_idxs_BSN = current_token_idxs_BSN

        # --- Decode all tokens at once at the end ---
        H, W = batch[""videos""].shape[2:4]
        final_frames_BSHWC = self.tokenizer.decode(
            final_token_idxs_BSN,
            video_hw=(H, W),
        )
        return final_frames_BSHWC
",python,selection_command
4104,12120335,"utils/nn.py",0,0,"",python,tab
4105,12123520,"utils/nn.py",15925,0,"",python,selection_command
4106,12123946,"utils/nn.py",0,0,"",python,selection_command
4107,12124000,"utils/nn.py",1722,0,"",python,selection_command
4108,12125242,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",0,0,"# Copyright 2024 The Flax Authors.
#
# Licensed under the Apache License, Version 2.0 (the ""License"");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an ""AS IS"" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

""""""Attention core modules for Flax.""""""

from __future__ import annotations

import functools
from typing import Any
from collections.abc import Callable
import math

import jax
import jax.numpy as jnp
from jax import lax, random

from flax import nnx
from flax.nnx import rnglib
from flax.nnx.module import Module, first_from
from flax.nnx.nn import initializers
from flax.nnx.nn import dtypes
from flax.nnx.nn.linear import (
  LinearGeneral,
  default_kernel_init,
)
from flax.nnx.nn.normalization import LayerNorm
from flax.typing import (
  Dtype,
  PromoteDtypeFn,
  Shape,
  Initializer,
  PrecisionLike,
  DotGeneralT,
)

Array = jax.Array


def dot_product_attention_weights(
  query: Array,
  key: Array,
  bias: Array | None = None,
  mask: Array | None = None,
  broadcast_dropout: bool = True,
  dropout_rng: Array | None = None,
  dropout_rate: float = 0.0,
  deterministic: bool = False,
  dtype: Dtype | None = None,
  precision: PrecisionLike = None,
  module: Module | None = None,
  promote_dtype: PromoteDtypeFn = dtypes.promote_dtype,
):
  """"""Computes dot-product attention weights given query and key.

  Used by :func:`dot_product_attention`, which is what you'll most likely use.
  But if you want access to the attention weights for introspection, then
  you can directly call this function and call einsum yourself.

  Args:
    query: queries for calculating attention with shape of `[batch..., q_length,
      num_heads, qk_depth_per_head]`.
    key: keys for calculating attention with shape of `[batch..., kv_length,
      num_heads, qk_depth_per_head]`.
    bias: bias for the attention weights. This should be broadcastable to the
      shape `[batch..., num_heads, q_length, kv_length]`. This can be used for
      incorporating causal masks, padding masks, proximity bias, etc.
    mask: mask for the attention weights. This should be broadcastable to the
      shape `[batch..., num_heads, q_length, kv_length]`. This can be used for
      incorporating causal masks. Attention weights are masked out if their
      corresponding mask value is `False`.
    broadcast_dropout: bool: use a broadcasted dropout along batch dims.
    dropout_rng: JAX PRNGKey: to be used for dropout
    dropout_rate: dropout rate
    deterministic: bool, deterministic or not (to apply dropout)
    dtype: the dtype of the computation (default: infer from inputs and params)
    precision: numerical precision of the computation see `jax.lax.Precision`
      for details.
    module: the Module that will sow the attention weights into the
      ``nnx.Intermediate`` collection. If ``module`` is None, the attention
      weights will not be sowed.
    promote_dtype: function to promote the dtype of the arrays to the desired
      dtype. The function should accept a tuple of ``(query, key)`` and a ``dtype``
      keyword argument, and return a tuple of arrays with the promoted dtype.

  Returns:
    Output of shape `[batch..., num_heads, q_length, kv_length]`.
  """"""
  query, key = promote_dtype((query, key), dtype=dtype)  # type: ignore[bad-unpacking]
  dtype = query.dtype

  assert query.ndim == key.ndim, 'q, k must have same rank.'
  assert query.shape[:-3] == key.shape[:-3], 'q, k batch dims must match.'
  assert query.shape[-2] == key.shape[-2], 'q, k num_heads must match.'
  assert query.shape[-1] == key.shape[-1], 'q, k depths must match.'

  # calculate attention matrix
  depth = query.shape[-1]
  query = query / jnp.sqrt(depth).astype(dtype)
  # attn weight shape is (batch..., num_heads, q_length, kv_length)
  attn_weights = jnp.einsum(
    '...qhd,...khd->...hqk', query, key, precision=precision
  )

  # apply attention bias: masking, dropout, proximity bias, etc.
  if bias is not None:
    attn_weights = attn_weights + bias
  # apply attention mask
  if mask is not None:
    big_neg = jnp.finfo(dtype).min
    attn_weights = jnp.where(mask, attn_weights, big_neg)

  # normalize the attention weights
  attn_weights = jax.nn.softmax(attn_weights).astype(dtype)

  if module:
    module.sow(nnx.Intermediate, 'attention_weights', attn_weights)

  # apply attention dropout
  if not deterministic and dropout_rate > 0.0:
    keep_prob = 1.0 - dropout_rate
    if broadcast_dropout:
      # dropout is broadcast across the batch + head dimensions
      dropout_shape = tuple([1] * (key.ndim - 2)) + attn_weights.shape[-2:]
      keep = random.bernoulli(dropout_rng, keep_prob, dropout_shape)  # type: ignore
    else:
      keep = random.bernoulli(dropout_rng, keep_prob, attn_weights.shape)  # type: ignore
    multiplier = keep.astype(dtype) / jnp.asarray(keep_prob, dtype=dtype)
    attn_weights = attn_weights * multiplier

  return attn_weights


def dot_product_attention(
  query: Array,
  key: Array,
  value: Array,
  bias: Array | None = None,
  mask: Array | None = None,
  broadcast_dropout: bool = True,
  dropout_rng: Array | None = None,
  dropout_rate: float = 0.0,
  deterministic: bool = False,
  dtype: Dtype | None = None,
  precision: PrecisionLike = None,
  module: Module | None = None,
  promote_dtype: PromoteDtypeFn = dtypes.promote_dtype,
):
  """"""Computes dot-product attention given query, key, and value.

  This is the core function for applying attention based on
  https://arxiv.org/abs/1706.03762. It calculates the attention weights given
  query and key and combines the values using the attention weights.

  Will use the more optimized `jax.nn.dot_product_attention` if dropout is
  not activated and `module=None`.

  .. note::
    ``query``, ``key``, ``value`` needn't have any batch dimensions.

  Args:
    query: queries for calculating attention with shape of ``[batch..., q_length,
      num_heads, qk_depth_per_head]``.
    key: keys for calculating attention with shape of ``[batch..., kv_length,
      num_heads, qk_depth_per_head]``.
    value: values to be used in attention with shape of ``[batch..., kv_length,
      num_heads, v_depth_per_head]``.
    bias: bias for the attention weights. This should be broadcastable to the
      shape `[batch..., num_heads, q_length, kv_length]`. This can be used for
      incorporating causal masks, padding masks, proximity bias, etc.
    mask: mask for the attention weights. This should be broadcastable to the
      shape `[batch..., num_heads, q_length, kv_length]`. This can be used for
      incorporating causal masks. Attention weights are masked out if their
      corresponding mask value is `False`.
    broadcast_dropout: bool: use a broadcasted dropout along batch dims.
    dropout_rng: JAX PRNGKey: to be used for dropout
    dropout_rate: dropout rate
    deterministic: bool, deterministic or not (to apply dropout)
    dtype: the dtype of the computation (default: infer from inputs)
    precision: numerical precision of the computation see `jax.lax.Precision`
      for details.
    module: the Module that will sow the attention weights into the
      ``nnx.Intermediate`` collection. If ``module`` is None, the attention
      weights will not be sowed.
    promote_dtype: function to promote the dtype of the arrays to the desired
      dtype. The function should accept a tuple of ``(query, key, value)`` and a
      ``dtype`` keyword argument, and return a tuple of arrays with the promoted
      dtype.

  Returns:
    Output of shape `[batch..., q_length, num_heads, v_depth_per_head]`.
  """"""
  query, key, value = promote_dtype((query, key, value), dtype=dtype)  # type: ignore[bad-unpacking]
  dtype = query.dtype
  assert key.ndim == query.ndim == value.ndim, 'q, k, v must have same rank.'
  assert (
    query.shape[:-3] == key.shape[:-3] == value.shape[:-3]
  ), 'q, k, v batch dims must match.'
  assert (
    query.shape[-2] == key.shape[-2] == value.shape[-2]
  ), 'q, k, v num_heads must match.'
  assert key.shape[-3] == value.shape[-3], 'k, v lengths must match.'

  # Criteria that invoke the more optimized dot product attention
  if dropout_rate == 0.0 and module == None:
    # make sure qkv batch are compressed to one dim
    query_shape = query.shape
    if len(query_shape) > 4:
      def reshape_4d(x):
        return jnp.reshape(x, (math.prod(x.shape[:-3]), *x.shape[-3:]))
      query, key, value, bias, mask = jax.tree.map(
        reshape_4d, (query, key, value, bias, mask))
    if mask is not None:
      mask = mask.astype(jnp.bool)
    out = jax.nn.dot_product_attention(query, key, value, bias, mask)
    if len(query_shape) > 4:
      out = jnp.reshape(out, query_shape)
    return out

  # compute attention weights
  attn_weights = dot_product_attention_weights(
    query,
    key,
    bias,
    mask,
    broadcast_dropout,
    dropout_rng,
    dropout_rate,
    deterministic,
    dtype,
    precision,
    module,
  )

  # return weighted sum over values for each query position
  return jnp.einsum(
    '...hqk,...khd->...qhd', attn_weights, value, precision=precision
  )


class MultiHeadAttention(Module):
  """"""Multi-head attention.

  Example usage::

    >>> from flax import nnx
    >>> import jax

    >>> layer = nnx.MultiHeadAttention(num_heads=8, in_features=5, qkv_features=16,
    ...                                decode=False, rngs=nnx.Rngs(0))
    >>> key1, key2, key3 = jax.random.split(jax.random.key(0), 3)
    >>> shape = (4, 3, 2, 5)
    >>> q, k, v = (
    ...   jax.random.uniform(key1, shape),
    ...   jax.random.uniform(key2, shape),
    ...   jax.random.uniform(key3, shape),
    ... )

    >>> # different inputs for inputs_q, inputs_k and inputs_v
    >>> out = layer(q, k, v)
    >>> # equivalent output when inferring v
    >>> assert (layer(q, k) == layer(q, k, k)).all()
    >>> # equivalent output when inferring k and v
    >>> assert (layer(q) == layer(q, q)).all()
    >>> assert (layer(q) == layer(q, q, q)).all()

  Args:
    num_heads: number of attention heads. Features (i.e. inputs_q.shape[-1])
      should be divisible by the number of heads.
    in_features: int or tuple with number of input features.
    qkv_features: dimension of the key, query, and value.
    out_features: dimension of the last projection.
    in_kv_features: number of input features for computing key and value.
    dtype: the dtype of the computation (default: infer from inputs and params)
    param_dtype: the dtype passed to parameter initializers (default: float32)
    broadcast_dropout: bool: use a broadcasted dropout along batch dims.
    dropout_rate: dropout rate
    deterministic: if false, the attention weight is masked randomly using
      dropout, whereas if true, the attention weights are deterministic.
    precision: numerical precision of the computation see `jax.lax.Precision`
      for details.
    kernel_init: initializer for the kernel of the Dense layers.
    out_kernel_init: optional initializer for the kernel of the output Dense layer,
      if None, the kernel_init is used.
    bias_init: initializer for the bias of the Dense layers.
    out_bias_init: optional initializer for the bias of the output Dense layer,
      if None, the bias_init is used.
    use_bias: bool: whether pointwise QKVO dense transforms use bias.
    attention_fn: dot_product_attention or compatible function. Accepts query,
      key, value, and returns output of shape `[bs, dim1, dim2, ..., dimN,,
      num_heads, value_channels]``
    decode: whether to prepare and use an autoregressive cache.
    normalize_qk: should QK normalization be applied (arxiv.org/abs/2302.05442).
    rngs: rng key.
    keep_rngs: whether to store the input rngs as attribute (i.e. `self.rngs = rngs`)
      (default: True). If rngs is stored, we should split the module as
      `graphdef, params, nondiff = nnx.split(module, nnx.Param, ...)` where `nondiff`
      contains RNG object associated with stored `self.rngs`.
  """"""

  __data__ = (
    'query',
    'key',
    'value',
    'out',
    'query_ln',
    'key_ln',
    'cached_key',
    'cached_value',
    'cache_index',
    'rngs',
  )

  def __init__(
    self,
    num_heads: int,
    in_features: int,
    qkv_features: int | None = None,
    out_features: int | None = None,
    in_kv_features: int | None = None,
    *,
    dtype: Dtype | None = None,
    param_dtype: Dtype = jnp.float32,
    broadcast_dropout: bool = True,
    dropout_rate: float = 0.0,
    deterministic: bool | None = None,
    precision: PrecisionLike = None,
    kernel_init: Initializer = default_kernel_init,
    out_kernel_init: Initializer | None = None,
    bias_init: Initializer = initializers.zeros_init(),
    out_bias_init: Initializer | None = None,
    use_bias: bool = True,
    attention_fn: Callable[..., Array] = dot_product_attention,
    decode: bool | None = None,
    normalize_qk: bool = False,
    # Deprecated, will be removed.
    qkv_dot_general: DotGeneralT | None = None,
    out_dot_general: DotGeneralT | None = None,
    qkv_dot_general_cls: Any = None,
    out_dot_general_cls: Any = None,
    rngs: rnglib.Rngs,
    keep_rngs: bool = True,
  ):
    self.num_heads = num_heads
    self.in_features = in_features
    self.qkv_features = (
      qkv_features if qkv_features is not None else in_features
    )
    self.out_features = (
      out_features if out_features is not None else in_features
    )
    self.in_kv_features = (
      in_kv_features if in_kv_features is not None else in_features
    )
    self.dtype = dtype
    self.param_dtype = param_dtype
    self.broadcast_dropout = broadcast_dropout
    self.dropout_rate = dropout_rate
    self.deterministic = deterministic
    self.precision = precision
    self.kernel_init = kernel_init
    self.out_kernel_init = out_kernel_init
    self.bias_init = bias_init
    self.out_bias_init = out_bias_init
    self.use_bias = use_bias
    self.attention_fn = attention_fn
    self.decode = decode
    self.normalize_qk = normalize_qk
    self.qkv_dot_general = qkv_dot_general
    self.out_dot_general = out_dot_general
    self.qkv_dot_general_cls = qkv_dot_general_cls
    self.out_dot_general_cls = out_dot_general_cls

    if self.qkv_features % self.num_heads != 0:
      raise ValueError(
        f'Memory dimension ({self.qkv_features}) must be divisible by '
        f""'num_heads' heads ({self.num_heads}).""
      )

    self.head_dim = self.qkv_features // self.num_heads

    linear_general = functools.partial(
      LinearGeneral,
      out_features=(self.num_heads, self.head_dim),
      dtype=self.dtype,
      param_dtype=self.param_dtype,
      kernel_init=self.kernel_init,
      bias_init=self.bias_init,
      use_bias=self.use_bias,
      precision=self.precision,
      dot_general=self.qkv_dot_general,
      dot_general_cls=self.qkv_dot_general_cls,
    )
    # project inputs_q to multi-headed q/k/v
    # dimensions are then [batch..., length, n_heads, n_features_per_head]
    self.query = linear_general(self.in_features, rngs=rngs)
    self.key = linear_general(self.in_kv_features, rngs=rngs)
    self.value = linear_general(self.in_kv_features, rngs=rngs)

    self.query_ln: LayerNorm | None
    self.key_ln: LayerNorm | None
    if self.normalize_qk:
      # Normalizing query and key projections stabilizes training with higher
      # LR. See ViT-22B paper http://arxiv.org/abs/2302.05442 for analysis.
      self.query_ln = LayerNorm(
        self.head_dim,
        use_bias=False,
        dtype=self.dtype,
        param_dtype=self.param_dtype,
        rngs=rngs,
      )
      self.key_ln = LayerNorm(
        self.head_dim,
        use_bias=False,
        dtype=self.dtype,
        param_dtype=self.param_dtype,
        rngs=rngs,
      )
    else:
      self.query_ln = None
      self.key_ln = None

    self.out = LinearGeneral(
      in_features=(self.num_heads, self.head_dim),
      out_features=self.out_features,
      axis=(-2, -1),
      kernel_init=self.out_kernel_init or self.kernel_init,
      bias_init=self.out_bias_init or self.bias_init,
      use_bias=self.use_bias,
      dtype=self.dtype,
      param_dtype=self.param_dtype,
      precision=self.precision,
      dot_general=self.out_dot_general,
      dot_general_cls=self.out_dot_general_cls,
      rngs=rngs,
    )
    self.rngs = rngs if keep_rngs and dropout_rate > 0 else None

    self.cached_key: nnx.Cache[Array] | None = None
    self.cached_value: nnx.Cache[Array] | None = None
    self.cache_index: nnx.Cache[Array] | None = None

  def __call__(
    self,
    inputs_q: Array,
    inputs_k: Array | None = None,
    inputs_v: Array | None = None,
    *,
    mask: Array | None = None,
    deterministic: bool | None = None,
    rngs: rnglib.Rngs | None = None,
    sow_weights: bool = False,
    decode: bool | None = None,
  ):
    """"""Applies multi-head dot product attention on the input data.

    Projects the inputs into multi-headed query, key, and value vectors,
    applies dot-product attention and project the results to an output vector.

    If both inputs_k and inputs_v are None, they will both copy the value of
    inputs_q (self attention).
    If only inputs_v is None, it will copy the value of inputs_k.

    Args:
      inputs_q: input queries of shape `[batch_sizes..., length, features]`.
      inputs_k: key of shape `[batch_sizes..., length, features]`. If None,
        inputs_k will copy the value of inputs_q.
      inputs_v: values of shape `[batch_sizes..., length, features]`. If None,
        inputs_v will copy the value of inputs_k.
      mask: attention mask of shape `[batch_sizes..., num_heads, query_length,
        key/value_length]`. Attention weights are masked out if their
        corresponding mask value is `False`.
      deterministic: if false, the attention weight is masked randomly using
        dropout, whereas if true, the attention weights are deterministic. The
        ``deterministic`` flag passed into the call method will take precedence
        over the ``deterministic`` flag passed into the constructor.
      rngs: rng key. The rng key passed into the call method will take
        precedence over the rng key passed into the constructor.
      sow_weights: if ``True``, the attention weights are sowed into the
        'intermediates' collection.
      decode: whether to prepare and use an autoregressive cache. The ``decode``
        flag passed into the call method will take precedence over the ``decode``
        flag passed into the constructor.

    Returns:
      output of shape `[batch_sizes..., length, features]`.
    """"""
    if rngs is None:
      rngs = self.rngs

    if inputs_k is None:
      if inputs_v is not None:
        raise ValueError(
          '`inputs_k` cannot be None if `inputs_v` is not None. '
          'To have both `inputs_k` and `inputs_v` be the same value, pass in the '
          'value to `inputs_k` and leave `inputs_v` as None.'
        )
      inputs_k = inputs_q
    if inputs_v is None:
      inputs_v = inputs_k

    if inputs_q.shape[-1] != self.in_features:
      raise ValueError(
        f'Incompatible input dimension, got {inputs_q.shape[-1]} '
        f'but module expects {self.in_features}.'
      )

    query = self.query(inputs_q)
    key = self.key(inputs_k)
    value = self.value(inputs_v)

    if self.normalize_qk:
      assert self.query_ln is not None and self.key_ln is not None
      # Normalizing query and key projections stabilizes training with higher
      # LR. See ViT-22B paper http://arxiv.org/abs/2302.05442 for analysis.
      query = self.query_ln(query)
      key = self.key_ln(key)

    # During fast autoregressive decoding, we feed one position at a time,
    # and cache the keys and values step by step.
    decode = first_from(
      decode,
      self.decode,
      error_msg=""""""No `decode` argument was provided to MultiHeadAttention
        as either a __call__ argument, class attribute, or nnx.flag."""""",
    )

    if decode:
      if (
        self.cached_key is None
        or self.cached_value is None
        or self.cache_index is None
      ):
        raise ValueError(
          'Autoregressive cache not initialized, call ``init_cache`` first.'
        )
      (
        *batch_dims,
        max_length,
        num_heads,
        depth_per_head,
      ) = self.cached_key.value.shape
      # shape check of cached keys against query input
      expected_shape = tuple(batch_dims) + (1, num_heads, depth_per_head)
      if expected_shape != query.shape:
        raise ValueError(
          'Autoregressive cache shape error, '
          'expected query shape %s instead got %s.'
          % (expected_shape, query.shape)
        )
      # update key, value caches with our new 1d spatial slices
      cur_index = self.cache_index[...]
      zero = jnp.array(0, dtype=lax.dtype(cur_index.dtype))
      indices = (zero,) * len(batch_dims) + (cur_index, zero, zero)
      key = lax.dynamic_update_slice(self.cached_key[...], key, indices)
      value = lax.dynamic_update_slice(self.cached_value[...], value, indices)
      self.cached_key[...] = key
      self.cached_value[...] = value
      self.cache_index[...] += 1
      # causal mask for cached decoder self-attention:
      # our single query position should only attend to those key
      # positions that have already been generated and cached,
      # not the remaining zero elements.
      mask = combine_masks(
        mask,
        jnp.broadcast_to(
          jnp.arange(max_length) <= cur_index,
          tuple(batch_dims) + (1, 1, max_length),
        ),
      )

    if (
      self.dropout_rate > 0.0
    ):  # Require `deterministic` only if using dropout.
      deterministic = first_from(
        deterministic,
        self.deterministic,
        error_msg=""""""No `deterministic` argument was provided to MultiHeadAttention
          as either a __call__ argument, class attribute, or nnx.flag."""""",
      )
      if not deterministic:
        if rngs is None:
          raise ValueError(
            ""'rngs' must be provided to __call__ method if ""
            ""MultiHeadAttention instance is defined with keep_rngs=False.""
          )
        dropout_rng = rngs.dropout()
      else:
        dropout_rng = None
    else:
      deterministic = True
      dropout_rng = None

    # apply attention
    x = self.attention_fn(
      query,
      key,
      value,
      mask=mask,
      dropout_rng=dropout_rng,
      dropout_rate=self.dropout_rate,
      broadcast_dropout=self.broadcast_dropout,
      deterministic=deterministic,
      dtype=self.dtype,
      precision=self.precision,
      module=self if sow_weights else None,
    )
    # back to the original inputs dimensions
    out = self.out(x)
    return out

  def init_cache(self, input_shape: Shape, dtype: Dtype = jnp.float32):
    """"""Initializes cache for fast autoregressive decoding. When
    ``decode=True``, this method must be called first before performing
    forward inference. When in decode mode, only one token must be passed
    at a time.

    Example usage::

      >>> from flax import nnx
      >>> import jax.numpy as jnp
      ...
      >>> batch_size = 5
      >>> embed_dim = 3
      >>> x = jnp.ones((batch_size, 1, embed_dim)) # single token
      ...
      >>> model_nnx = nnx.MultiHeadAttention(
      ...   num_heads=2,
      ...   in_features=3,
      ...   qkv_features=6,
      ...   out_features=6,
      ...   decode=True,
      ...   rngs=nnx.Rngs(42),
      ... )
      ...
      >>> # out_nnx = model_nnx(x)  <-- throws an error because cache isn't initialized
      ...
      >>> model_nnx.init_cache(x.shape)
      >>> out_nnx = model_nnx(x)
    """"""
    cache_shape = (*input_shape[:-1], self.num_heads, self.head_dim)
    self.cached_key = nnx.Cache(jnp.zeros(cache_shape, dtype))
    self.cached_value = nnx.Cache(jnp.zeros(cache_shape, dtype))
    self.cache_index = nnx.Cache(jnp.array(0, dtype=jnp.int32))


# mask-making utility functions


def make_attention_mask(
  query_input: Array,
  key_input: Array,
  pairwise_fn: Callable[..., Any] = jnp.multiply,
  extra_batch_dims: int = 0,
  dtype: Dtype = jnp.float32,
):
  """"""Mask-making helper for attention weights.

  In case of 1d inputs (i.e., `[batch..., len_q]`, `[batch..., len_kv]`, the
  attention weights will be `[batch..., heads, len_q, len_kv]` and this
  function will produce `[batch..., 1, len_q, len_kv]`.

  Args:
    query_input: a batched, flat input of query_length size
    key_input: a batched, flat input of key_length size
    pairwise_fn: broadcasting elementwise comparison function
    extra_batch_dims: number of extra batch dims to add singleton axes for, none
      by default
    dtype: mask return dtype

  Returns:
    A `[batch..., 1, len_q, len_kv]` shaped mask for 1d attention.
  """"""
  mask = pairwise_fn(
    jnp.expand_dims(query_input, axis=-1), jnp.expand_dims(key_input, axis=-2)
  )
  mask = jnp.expand_dims(mask, axis=-3)
  mask = jnp.expand_dims(mask, axis=tuple(range(extra_batch_dims)))
  return mask.astype(dtype)


def make_causal_mask(
  x: Array, extra_batch_dims: int = 0, dtype: Dtype = jnp.float32
) -> Array:
  """"""Make a causal mask for self-attention.

  In case of 1d inputs (i.e., `[batch..., len]`, the self-attention weights
  will be `[batch..., heads, len, len]` and this function will produce a
  causal mask of shape `[batch..., 1, len, len]`.

  Args:
    x: input array of shape `[batch..., len]`
    extra_batch_dims: number of batch dims to add singleton axes for, none by
      default
    dtype: mask return dtype

  Returns:
    A `[batch..., 1, len, len]` shaped causal mask for 1d attention.
  """"""
  idxs = jnp.broadcast_to(jnp.arange(x.shape[-1], dtype=jnp.int32), x.shape)
  return make_attention_mask(
    idxs,
    idxs,
    jnp.greater_equal,
    extra_batch_dims=extra_batch_dims,
    dtype=dtype,
  )


def combine_masks(
  *masks: Array | None, dtype: Dtype = jnp.float32
) -> Array | None:
  """"""Combine attention masks.

  Args:
    *masks: set of attention mask arguments to combine, some can be None.
    dtype: dtype for the returned mask.

  Returns:
    Combined mask, reduced by logical and, returns None if no masks given.
  """"""
  masks_list = [m for m in masks if m is not None]
  if not masks_list:
    return None
  assert all(
    map(lambda x: x.ndim == masks_list[0].ndim, masks_list)
  ), f'masks must have same rank: {tuple(map(lambda x: x.ndim, masks_list))}'
  mask, *other_masks = masks_list
  for other_mask in other_masks:
    mask = jnp.logical_and(mask, other_mask)
  return mask.astype(dtype)
",python,tab
4109,12125242,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",9464,0,"",python,selection_command
4110,12126205,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",12517,0,"",python,selection_command
4111,12130371,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",13195,0,"",python,selection_command
4112,12130522,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",14311,0,"",python,selection_command
4113,12130834,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",14320,0,"",python,selection_command
4114,12131148,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",17149,0,"",python,selection_command
4115,12132069,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",18670,0,"",python,selection_command
4116,12132407,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",18736,0,"",python,selection_command
4117,12132826,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",18818,0,"",python,selection_command
4118,12133228,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20111,0,"",python,selection_command
4119,12134774,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20138,0,"",python,selection_command
4120,12134972,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20157,0,"",python,selection_command
4121,12135157,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20188,0,"",python,selection_command
4122,12135645,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20327,0,"",python,selection_command
4123,12136124,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20320,14,"    if decode:",python,selection_command
4124,12137257,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20320,1625,"    if decode:
      if (
        self.cached_key is None
        or self.cached_value is None
        or self.cache_index is None
      ):
        raise ValueError(
          'Autoregressive cache not initialized, call ``init_cache`` first.'
        )
      (
        *batch_dims,
        max_length,
        num_heads,
        depth_per_head,
      ) = self.cached_key.value.shape
      # shape check of cached keys against query input
      expected_shape = tuple(batch_dims) + (1, num_heads, depth_per_head)
      if expected_shape != query.shape:
        raise ValueError(
          'Autoregressive cache shape error, '
          'expected query shape %s instead got %s.'
          % (expected_shape, query.shape)
        )
      # update key, value caches with our new 1d spatial slices
      cur_index = self.cache_index[...]
      zero = jnp.array(0, dtype=lax.dtype(cur_index.dtype))
      indices = (zero,) * len(batch_dims) + (cur_index, zero, zero)
      key = lax.dynamic_update_slice(self.cached_key[...], key, indices)
      value = lax.dynamic_update_slice(self.cached_value[...], value, indices)
      self.cached_key[...] = key
      self.cached_value[...] = value
      self.cache_index[...] += 1
      # causal mask for cached decoder self-attention:
      # our single query position should only attend to those key
      # positions that have already been generated and cached,
      # not the remaining zero elements.
      mask = combine_masks(
        mask,
        jnp.broadcast_to(
          jnp.arange(max_length) <= cur_index,
          tuple(batch_dims) + (1, 1, max_length),
        ),
      )
",python,selection_command
4125,12284342,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20319,0,"",python,selection_mouse
4126,12288514,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20328,0,"",python,selection_mouse
4127,12295051,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20653,0,"",python,selection_mouse
4128,12296271,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",20771,0,"",python,selection_mouse
4129,12297504,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",21124,0,"",python,selection_mouse
4130,12307669,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",24045,0,"",python,selection_mouse
4131,12313645,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",24061,0,"",python,selection_mouse
4132,12315472,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",24045,0,"",python,selection_mouse
4133,12360741,".venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py",24133,0,"",python,selection_mouse
4134,12367518,".venv/lib/python3.10/site-packages/flax/nnx/variablelib.py",0,0,"# Copyright 2024 The Flax Authors.
#
# Licensed under the Apache License, Version 2.0 (the ""License"");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an ""AS IS"" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# pytype: skip-file
from __future__ import annotations

import dataclasses
import functools
from functools import partial
import typing as tp
from typing import Any
from flax import config

import jax
import treescope  # type: ignore[import-untyped]

from flax import errors
from flax.nnx import filterlib, reprlib, tracers, visualization
from flax.typing import Missing, PathParts, SizeBytes
import jax.tree_util as jtu
import jax.numpy as jnp
from jax._src.core import mutable_array, MutableArray
from jax._src.state.types import AbstractRef

A = tp.TypeVar('A')
B = tp.TypeVar('B')
F = tp.TypeVar('F', bound=tp.Callable[..., tp.Any])
V = tp.TypeVar('V', bound='Variable[Any]')
GetValueHook = tp.Callable[['Variable[A]', A], A]
SetValueHook = tp.Callable[['Variable[A]', A], A]
CreateValueHook = tp.Callable[['Variable[A]', A], A]
AxisName = str
AxisIndex = int
AddAxisHook = tp.Callable[[V, AxisIndex, AxisName | None], None]
RemoveAxisHook = tp.Callable[[V, AxisIndex, AxisName | None], None]

MUTABLE_DEFAULT = True if config.flax_mutable_array else None


def is_mutable_array(x) -> tp.TypeGuard[MutableArray]:
  return isinstance(x, jax.Array | AbstractRef | MutableArray) and isinstance(
    jax.typeof(x), AbstractRef | MutableArray
  )


@dataclasses.dataclass
class VariableMetadata(tp.Generic[A]):
  raw_value: A
  set_value_hooks: tuple[SetValueHook[A], ...] = ()
  get_value_hooks: tuple[GetValueHook[A], ...] = ()
  create_value_hooks: tuple[CreateValueHook[A], ...] = ()
  add_axis_hooks: tuple[AddAxisHook[Variable[A]], ...] = ()
  remove_axis_hooks: tuple[RemoveAxisHook[Variable[A]], ...] = ()
  metadata: tp.Mapping[str, tp.Any] = dataclasses.field(default_factory=dict)


class Variable(tp.Generic[A], reprlib.Representable):
  """"""The base class for all ``Variable`` types. Create custom ``Variable``
  types by subclassing this class. Numerous NNX graph functions can filter
  for specific ``Variable`` types, for example, :func:`split`, :func:`state`,
  :func:`pop`, and :func:`State.filter`.

  Example usage::

    >>> from flax import nnx
    >>> import jax, jax.numpy as jnp

    >>> class CustomVariable(nnx.Variable):
    ...   pass

    >>> class Model(nnx.Module):
    ...   def __init__(self, rngs):
    ...     self.linear = nnx.Linear(2, 3, rngs=rngs)
    ...     self.custom_variable = CustomVariable(jnp.ones((1, 3)))
    ...   def __call__(self, x):
    ...     return self.linear(x) + self.custom_variable
    >>> model = Model(rngs=nnx.Rngs(0))

    >>> linear_variables = nnx.state(model, nnx.Param)
    >>> jax.tree.map(jnp.shape, linear_variables)
    State({
      'linear': {
        'bias': VariableState(
          type=Param,
          value=(3,)
        ),
        'kernel': VariableState(
          type=Param,
          value=(2, 3)
        )
      }
    })

    >>> custom_variable = nnx.state(model, CustomVariable)
    >>> jax.tree.map(jnp.shape, custom_variable)
    State({
      'custom_variable': VariableState(
        type=CustomVariable,
        value=(1, 3)
      )
    })

    >>> variables = nnx.state(model)
    >>> jax.tree.map(jnp.shape, variables)
    State({
      'custom_variable': VariableState(
        type=CustomVariable,
        value=(1, 3)
      ),
      'linear': {
        'bias': VariableState(
          type=Param,
          value=(3,)
        ),
        'kernel': VariableState(
          type=Param,
          value=(2, 3)
        )
      }
    })
  """"""

  __slots__ = ('raw_value', '_trace_state', '_var_metadata')

  raw_value: A
  _trace_state: tracers.TraceState
  _var_metadata: dict[str, tp.Any]

  def __init__(
    self,
    value: tp.Union[A, VariableMetadata[A]],
    *,
    mutable: bool | None = MUTABLE_DEFAULT,
    **metadata: tp.Any,
  ):
    var_t = type(self)
    object.__setattr__(self, '_trace_state', tracers.TraceState())

    if isinstance(value, VariableMetadata):
      metadata.update(value.metadata)
      value = tp.cast(A, value.raw_value)

    if mutable is None:
      _value = value
    elif mutable:
      if is_mutable_array(value):
        _value = tp.cast(A, value)
      else:
        _value = mutable_array(jnp.asarray(value))
    else:
      _value = tp.cast(A, jnp.asarray(value))

    object.__setattr__(self, 'raw_value', _value)

    if hasattr(var_t, 'on_get_value') and 'on_get_value' not in metadata:
      metadata['on_get_value'] = var_t.on_get_value

    if hasattr(var_t, 'on_set_value') and 'on_set_value' not in metadata:
      metadata['on_set_value'] = var_t.on_set_value

    if hasattr(var_t, 'on_create_value') and 'on_create_value' not in metadata:
      metadata['on_create_value'] = var_t.on_create_value

    if hasattr(var_t, 'on_add_axis') and 'on_add_axis' not in metadata:
      metadata['on_add_axis'] = var_t.on_add_axis

    if hasattr(var_t, 'on_remove_axis') and 'on_remove_axis' not in metadata:
      metadata['on_remove_axis'] = var_t.on_remove_axis

    object.__setattr__(self, '_var_metadata', metadata)
    # run create_value hooks
    object.__setattr__(self, 'raw_value', self.create_value(self.raw_value))

  def __getattr__(self, name: str) -> tp.Any:
    if name in object.__getattribute__(self, '_var_metadata'):
      return self._var_metadata[name]
    return getattr(self.raw_value, name)

  def __setattr__(self, name: str, value: tp.Any):
    if not self._trace_state.is_valid() and (
      name != 'value' or not self.mutable
    ):
      raise errors.TraceContextError(
        f'Cannot mutate {type(self).__name__} from a different trace level'
      )
    if (
      name == 'value'
      or name == 'raw_value'
      or name == '_var_metadata'
      or name == '_trace_state'
    ):
      object.__setattr__(self, name, value)
    else:
      self._var_metadata[name] = value

  def __delattr__(self, name: str):
    if not self._trace_state.is_valid():
      raise errors.TraceContextError(
        f'Cannot mutate {type(self).__name__} from a different trace level'
      )

    if (
      name == 'value'
      or name == 'raw_value'
      or name == '_var_metadata'
      or name == '_trace_state'
    ):
      object.__delattr__(self, name)
    else:
      del self._var_metadata[name]

  @classmethod
  def state(cls, value: A, **metadata) -> VariableState[A]:
    return cls(value, **metadata).to_state()

  @property
  def mutable(self) -> bool | None:
    if is_mutable_array(self.raw_value):
      return True
    elif isinstance(self.raw_value, jax.Array):
      return False
    else:
      return None

  def get_metadata(self):
    return self._var_metadata

  def copy_from(self, other: Variable[A]) -> None:
    if type(self) is not type(other):
      raise ValueError(
        f'Cannot copy from incompatible container, '
        f'expected {type(self).__name__}, got {type(other).__name__}'
      )
    if self is other:
      return
    self.raw_value = other.raw_value
    self._var_metadata.clear()
    self._var_metadata.update(other.get_metadata())

  def update_from_state(self, variable_state: VariableState[A]):
    object.__setattr__(self, 'raw_value', variable_state.raw_value)
    object.__setattr__(
        self, '_var_metadata', variable_state._var_metadata.copy()
    )

  @property
  def value(self) -> A:
    value = self.raw_value
    if is_mutable_array(value):
      value = value[...]

    if 'on_get_value' in self._var_metadata:
      value = self._var_metadata['on_get_value'](self, value)
    return value

  @value.setter
  def value(self, value: A):
    if isinstance(value, Variable):
      raise ValueError(
        'Cannot set value to a Variable, ' 'use `copy_from` method instead'
      )
    if 'on_set_value' in self._var_metadata:
      value = self._var_metadata['on_set_value'](self, value)
    if config.flax_mutable_array:
      self.raw_value[...] = value  # type: ignore
    else:
      object.__setattr__(self, 'raw_value', value)

  def create_value(self, value: A):
    if 'on_create_value' in self._var_metadata:
      value = self._var_metadata['on_create_value'](self, value)
    return value

  def add_axis(self, axis_index: AxisIndex, axis_name: AxisName | None):
    if 'on_add_axis' in self._var_metadata:
      self._var_metadata['on_add_axis'](self, axis_index, axis_name)

  def remove_axis(self, axis_index: AxisIndex, axis_name: AxisName | None):
    if 'on_remove_axis' in self._var_metadata:
      self._var_metadata['on_remove_axis'](self, axis_index, axis_name)

  @tp.overload
  def replace(self, value: B, **kwargs) -> Variable[B]: ...

  @tp.overload
  def replace(self, **kwargs) -> Variable[A]: ...

  def replace(self, value: tp.Any = Missing, **kwargs) -> Variable[tp.Any]:
    if value is not Missing:
      kwargs['raw_value'] = value

    # rename `value` to `raw_value`
    if 'value' in kwargs:
      kwargs['raw_value'] = kwargs.pop('value')

    # return `value` if it is a Variable
    if 'raw_value' in kwargs and isinstance(
      value := kwargs['raw_value'], Variable
    ):
      # remove value from kwargs
      kwargs.pop('raw_value')
      if type(self) is not type(value):
        raise ValueError(
          'Cannot replace value from incompatible container, '
          f'expected {type(self).__name__}, got {type(value).__name__}'
        )
      # if kwargs aren't empty, recursively call replace
      # else return variable value
      if kwargs:
        return value.replace(**kwargs)
      else:
        return value

    # get and update attributes
    # return new instance with updated attributes
    obj = object.__new__(type(self))
    object.__setattr__(obj, '_trace_state', self._trace_state)
    object.__setattr__(obj, 'raw_value', kwargs.pop('raw_value'))
    object.__setattr__(obj, '_var_metadata', self.get_metadata() | kwargs)
    return obj

  @classmethod
  def from_metadata(cls, value: A, attributes: dict[str, tp.Any]):
    obj = object.__new__(cls)
    object.__setattr__(obj, '_trace_state', tracers.TraceState())
    object.__setattr__(obj, 'raw_value', value)
    object.__setattr__(obj, '_var_metadata', attributes)
    return obj

  def copy(self: Variable[A]) -> Variable[A]:
    obj = object.__new__(type(self))
    object.__setattr__(obj, '_trace_state', self._trace_state)
    object.__setattr__(obj, 'raw_value', self.raw_value)
    object.__setattr__(obj, '_var_metadata', self.get_metadata().copy())
    return obj

  def to_state(self: Variable[A]) -> VariableState[A]:
    return VariableState(type(self), self.raw_value, **self._var_metadata)

  def __nnx_repr__(self):
    stats = SizeBytes.from_any(self.raw_value)
    if stats:
      comment = f' # {stats}'
    else:
      comment = ''

    yield reprlib.Object(type=type(self).__name__, comment=comment)
    yield reprlib.Attr('value', self.raw_value)
    for name, value in self._var_metadata.items():
      yield reprlib.Attr(name, repr(value))

  def __treescope_repr__(self, path, subtree_renderer):
    size_bytes = SizeBytes.from_any(self.value)
    if size_bytes:
      stats_repr = f' # {size_bytes}'
      first_line_annotation = treescope.rendering_parts.comment_color(
        treescope.rendering_parts.text(f'{stats_repr}')
      )
    else:
      first_line_annotation = None

    children = {'value': self.raw_value, **self._var_metadata}
    return visualization.render_object_constructor(
      object_type=type(self),
      attributes=children,
      path=path,
      subtree_renderer=subtree_renderer,
      first_line_annotation=first_line_annotation,
    )

  # hooks API
  if tp.TYPE_CHECKING:

    def on_get_value(self, value: A) -> A: ...

    def on_set_value(self, value: A) -> A: ...

    def on_create_value(self, value: A) -> A: ...

    def on_add_axis(
      self: V, axis_index: AxisIndex, axis_name: AxisName | None
    ) -> V: ...

    def on_remove_axis(
      self: V, axis_index: AxisIndex, axis_name: AxisName | None
    ) -> V: ...

  def __jax_array__(self):
    return self.value

  # pickle support
  def __getstate__(self):
    return {
        'raw_value': self.raw_value,
        '_trace_state': self._trace_state,
        '_var_metadata': self._var_metadata,
    }

  def __setstate__(self, state):
    object.__setattr__(self, 'raw_value', state['raw_value'])
    object.__setattr__(self, '_trace_state', state['_trace_state'])
    object.__setattr__(self, '_var_metadata', state['_var_metadata'])

  # --------------------------------------------
  # proxy methods
  # --------------------------------------------

  def __getitem__(self, key) -> jax.Array:
    return self.value[key]  # type: ignore

  def __setitem__(self, key, value) -> None:
    if config.flax_mutable_array:
      self.raw_value[key] = value  # type: ignore
    else:
      if not is_mutable_array(self.raw_value):
        if not self._trace_state.is_valid():
          raise errors.TraceContextError(
            f'Cannot mutate {type(self).__name__} from a different trace level'
          )
      if isinstance(self.raw_value, jax.Array):
        self.raw_value = self.raw_value.at[key].set(value)  # type: ignore
      else:
        self.raw_value[key] = value  # type: ignore

  def __call__(self, *args, **kwargs) -> tp.Any:
    return self.value(*args, **kwargs)  # type: ignore

  def __len__(self) -> int:
    return len(self.value)  # type: ignore

  def __iter__(self) -> tp.Iterator:
    return iter(self.value)  # type: ignore

  def __contains__(self, item) -> bool:
    return item in self.value  # type: ignore

  def __add__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__add__(other)  # type: ignore

  def __sub__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__sub__(other)  # type: ignore

  def __mul__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__mul__(other)  # type: ignore

  def __matmul__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__matmul__(other)  # type: ignore

  def __truediv__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__truediv__(other)  # type: ignore

  def __floordiv__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__floordiv__(other)  # type: ignore

  def __mod__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__mod__(other)  # type: ignore

  def __divmod__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__divmod__(other)  # type: ignore

  def __pow__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__pow__(other)  # type: ignore

  def __lshift__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__lshift__(other)  # type: ignore

  def __rshift__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rshift__(other)  # type: ignore

  def __and__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__and__(other)  # type: ignore

  def __xor__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__xor__(other)  # type: ignore

  def __or__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__or__(other)  # type: ignore

  def __radd__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__radd__(other)  # type: ignore

  def __rsub__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rsub__(other)  # type: ignore

  def __rmul__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rmul__(other)  # type: ignore

  def __rmatmul__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rmatmul__(other)  # type: ignore

  def __rtruediv__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rtruediv__(other)  # type: ignore

  def __rfloordiv__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rfloordiv__(other)  # type: ignore

  def __rmod__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rmod__(other)  # type: ignore

  def __rdivmod__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rdivmod__(other)  # type: ignore

  def __rpow__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rpow__(other)  # type: ignore

  def __rlshift__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rlshift__(other)  # type: ignore

  def __rrshift__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rrshift__(other)  # type: ignore

  def __rand__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rand__(other)  # type: ignore

  def __rxor__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__rxor__(other)  # type: ignore

  def __ror__(self, other) -> A:
    if isinstance(other, Variable):
      other = other.value
    return self.value.__ror__(other)  # type: ignore

  def __iadd__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__iadd__'):
      value.__iadd__(other)
    else:
      self.value = value.__add__(other)
    return self

  def __isub__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__isub__'):
      value.__isub__(other)
    else:
      self.value = value.__sub__(other)
    return self

  def __imul__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__imul__'):
      value.__imul__(other)
    else:
      self.value = value.__mul__(other)
    return self

  def __imatmul__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__imatmul__'):
      value.__imatmul__(other)
    else:
      self.value = value.__matmul__(other)
    return self

  def __itruediv__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__itruediv__'):
      value.__itruediv__(other)
    else:
      self.value = value.__truediv__(other)
    return self

  def __ifloordiv__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__ifloordiv__'):
      value.__ifloordiv__(other)
    else:
      self.value = value.__floordiv__(other)
    return self

  def __imod__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__imod__'):
      value.__imod__(other)
    else:
      self.value = value.__mod__(other)
    return self

  def __ipow__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__ipow__'):
      value.__ipow__(other)
    else:
      self.value = value.__pow__(other)
    return self

  def __ilshift__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__ilshift__'):
      value.__ilshift__(other)
    else:
      self.value = value.__lshift__(other)
    return self

  def __irshift__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__irshift__'):
      value.__irshift__(other)
    else:
      self.value = value.__rshift__(other)
    return self

  def __iand__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__iand__'):
      value.__iand__(other)
    else:
      self.value = value.__and__(other)
    return self

  def __ixor__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__ixor__'):
      value.__ixor__(other)
    else:
      self.value = value.__xor__(other)
    return self

  def __ior__(self: V, other) -> V:
    if isinstance(other, Variable):
      other = other.value
    value = self.value
    if hasattr(value, '__ior__'):
      value.__ior__(other)
    else:
      self.value = value.__or__(other)
    return self

  def __neg__(self) -> A:
    return self.value.__neg__()  # type: ignore

  def __pos__(self) -> A:
    return self.value.__pos__()  # type: ignore

  def __abs__(self) -> A:
    return self.value.__abs__()  # type: ignore

  def __invert__(self) -> A:
    return self.value.__invert__()  # type: ignore

  def __complex__(self) -> A:
    return self.value.__complex__()  # type: ignore

  def __int__(self) -> A:
    return self.value.__int__()  # type: ignore

  def __float__(self) -> A:
    return self.value.__float__()  # type: ignore

  def __index__(self) -> A:
    return self.value.__index__()  # type: ignore

  def __round__(self, ndigits: int) -> A:
    return self.value.__round__(ndigits)  # type: ignore

  def __trunc__(self) -> A:
    return self.value.__trunc__()  # type: ignore

  def __floor__(self) -> A:
    return self.value.__floor__()  # type: ignore

  def __ceil__(self) -> A:
    return self.value.__ceil__()  # type: ignore

  # --------------------------------------------

  def __init_subclass__(cls) -> None:
    super().__init_subclass__()

    jax.tree_util.register_pytree_with_keys(
      cls,
      flatten_with_keys=_variable_flatten_with_keys,
      unflatten_func=partial(_variable_unflatten, cls),  # type: ignore
      flatten_func=_variable_flatten,
    )


def _variable_flatten_with_keys(x: Variable[tp.Any]):
  metadata = tuple(x.get_metadata().items())
  node = (jtu.GetAttrKey('value'), x.raw_value)
  return (node,), metadata

def _variable_flatten(x: Variable[tp.Any]):
  metadata = tuple(x.get_metadata().items())
  return (x.raw_value,), metadata


def _variable_unflatten(
  cls: type[Variable[tp.Any]],
  static: tuple[tuple[str, tp.Any], ...],
  children: tuple[tp.Any],
):
  return cls.from_metadata(value=children[0], attributes=dict(static))

jax.tree_util.register_pytree_with_keys(
  Variable,
  flatten_with_keys=_variable_flatten_with_keys,
  unflatten_func=partial(_variable_unflatten, Variable),  # type: ignore
  flatten_func=_variable_flatten,
)


class Param(Variable[A]):
  """"""The canonical learnable parameter. All learnable parameters
  in NNX layer modules will have the ``Param`` :class:`Variable`
  type::

    >>> from flax import nnx
    >>> import jax, jax.numpy as jnp

    >>> layer = nnx.Linear(2, 3, rngs=nnx.Rngs(0))
    >>> jax.tree.map(jnp.shape, nnx.state(layer))
    State({
      'bias': VariableState(
        type=Param,
        value=(3,)
      ),
      'kernel': VariableState(
        type=Param,
        value=(2, 3)
      )
    })
  """"""

  pass


class BatchStat(Variable[A]):
  """"""The mean and variance batch statistics stored in
  the :class:`BatchNorm` layer. Note, these are not the
  learnable scale and bias parameters, but rather the
  running average statistics that are typically used
  during post-training inference::

    >>> from flax import nnx
    >>> import jax, jax.numpy as jnp

    >>> layer = nnx.BatchNorm(3, rngs=nnx.Rngs(0))
    >>> jax.tree.map(jnp.shape, nnx.state(layer))
    State({
      'bias': VariableState(
        type=Param,
        value=(3,)
      ),
      'mean': VariableState(
        type=BatchStat,
        value=(3,)
      ),
      'scale': VariableState(
        type=Param,
        value=(3,)
      ),
      'var': VariableState(
        type=BatchStat,
        value=(3,)
      )
    })
  """"""

  pass


class Cache(Variable[A]):
  """"""Autoregressive cache in :class:`MultiHeadAttention`::

  >>> from flax import nnx
  >>> import jax, jax.numpy as jnp

  >>> layer = nnx.MultiHeadAttention(
  ...   num_heads=2,
  ...   in_features=3,
  ...   qkv_features=6,
  ...   out_features=6,
  ...   decode=True,
  ...   rngs=nnx.Rngs(0),
  ... )

  >>> layer.init_cache((1, 3))
  >>> jax.tree.map(jnp.shape, nnx.state(layer, nnx.Cache))
  State({
    'cache_index': VariableState(
      type=Cache,
      value=()
    ),
    'cached_key': VariableState(
      type=Cache,
      value=(1, 2, 3)
    ),
    'cached_value': VariableState(
      type=Cache,
      value=(1, 2, 3)
    )
  })
  """"""

  pass


class Intermediate(Variable[A]):
  """""":class:`Variable` type that is typically used for
  :func:`Module.sow`::

    >>> from flax import nnx
    >>> import jax, jax.numpy as jnp

    >>> class Model(nnx.Module):
    ...   def __init__(self, rngs):
    ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
    ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
    ...   def __call__(self, x):
    ...     x = self.linear1(x)
    ...     self.sow(nnx.Intermediate, 'i', x)
    ...     x = self.linear2(x)
    ...     return x
    >>> model = Model(rngs=nnx.Rngs(0))

    >>> x = jnp.ones((1, 2))
    >>> y = model(x)
    >>> jax.tree.map(jnp.shape, nnx.state(model, nnx.Intermediate))
    State({
      'i': VariableState(
        type=Intermediate,
        value=((1, 3),)
      )
    })
  """"""

  pass


class Perturbation(Intermediate[A]):
  """""":class:`Variable` type that is typically used for
  :func:`Module.perturb`::

    >>> from flax import nnx
    >>> import jax, jax.numpy as jnp

    >>> class Model(nnx.Module):
    ...   def __init__(self, rngs):
    ...     self.linear1 = nnx.Linear(2, 3, rngs=rngs)
    ...     self.linear2 = nnx.Linear(3, 4, rngs=rngs)
    ...   def __call__(self, x):
    ...     x = self.linear1(x)
    ...     x = self.perturb('i', x)
    ...     x = self.linear2(x)
    ...     return x
    >>> model = Model(rngs=nnx.Rngs(0))

    >>> x = jnp.ones((1, 2))
    >>> y = model(x)
    >>> jax.tree.map(jnp.shape, nnx.state(model, nnx.Perturbation))
    State({
      'i': VariableState(
        type=Perturbation,
        value=(1, 3)
      )
    })
  """"""

  pass


class VariableState(tp.Generic[A], reprlib.Representable):
  __slots__ = ('type', 'value', '_var_metadata')
  type: type[Variable[A]]
  value: A
  _var_metadata: dict[str, tp.Any]

  def __init__(
    self,
    type: type[Variable[A]],  # type: ignore [valid-type]
    value: A,
    **metadata,
  ):
    object.__setattr__(self, 'type', type)
    object.__setattr__(self, 'value', value)
    object.__setattr__(self, '_var_metadata', metadata)

  @property
  def raw_value(self) -> A:
    return object.__getattribute__(self, 'value')

  @raw_value.setter
  def raw_value(self, value: A) -> None:
    object.__setattr__(self, 'value', value)

  def __getattribute__(self, name: str) -> None:
    if name == 'value':
      value = object.__getattribute__(self, 'value')
      if is_mutable_array(value):
        value = value[...]
      return value
    return object.__getattribute__(self, name)

  def __getattr__(self, name: str) -> None:
    var_metadata = object.__getattribute__(self, '_var_metadata')
    if name not in var_metadata:
      raise AttributeError(f""'VariableState' object has no attribute '{name}'"")
    return var_metadata[name]

  def __setattr__(self, name: str, value: Any) -> None:
    if name in ('type', 'value', '_var_metadata', 'raw_value'):
      object.__setattr__(self, name, value)
    else:
      self._var_metadata[name] = value

  def __delattr__(self, name: str) -> None:
    if name in ('type', 'value', '_var_metadata', 'raw_value'):
      object.__delattr__(self, name)
    else:
      del self._var_metadata[name]

  def __getitem__(self, key: Any) -> jax.Array:
    return self.raw_value[key]  # type: ignore

  def __setitem__(self, key: Any, value: Any) -> None:
    self.raw_value[key] = value  # type: ignore

  def __nnx_repr__(self):
    stats = SizeBytes.from_any(self.raw_value)
    if stats:
      comment = f' # {stats}'
    else:
      comment = ''

    yield reprlib.Object(type=type(self), comment=comment)
    yield reprlib.Attr('type', self.type)
    yield reprlib.Attr('value', self.raw_value)

    for name, value in self._var_metadata.items():
      yield reprlib.Attr(name, value)

  def __treescope_repr__(self, path, subtree_renderer):
    size_bytes = SizeBytes.from_any(self.raw_value)
    if size_bytes:
      stats_repr = f' # {size_bytes}'
      first_line_annotation = treescope.rendering_parts.comment_color(
        treescope.rendering_parts.text(f'{stats_repr}')
      )
    else:
      first_line_annotation = None
    children = {'type': self.type, 'value': self.value, **self._var_metadata}
    return visualization.render_object_constructor(
      object_type=type(self),
      attributes=children,
      path=path,
      subtree_renderer=subtree_renderer,
      first_line_annotation=first_line_annotation,
    )

  def replace(self, value: B) -> VariableState[B]:
    return VariableState(self.type, value, **self.get_metadata())

  def to_variable(self) -> Variable[A]:
    # we use object.__new__ to avoid calling __init__ and bypass the
    # __init__ logic which should not be called twice
    variable = object.__new__(self.type)
    object.__setattr__(variable, '_trace_state', tracers.TraceState())
    object.__setattr__(variable, 'raw_value', self.raw_value)
    object.__setattr__(variable, '_var_metadata', self.get_metadata().copy())
    return variable

  def copy(self: VariableState[A]) -> VariableState[A]:
    return jax.tree.map(lambda x: x, self)

  def get_metadata(self) -> dict[str, tp.Any]:
    return self._var_metadata

  def add_axis(self, axis_index: AxisIndex, axis_name: AxisName | None):
    if 'on_add_axis' in self._var_metadata:
      self._var_metadata['on_add_axis'](self, axis_index, axis_name)

  def remove_axis(self, axis_index: AxisIndex, axis_name: AxisName | None):
    if 'on_remove_axis' in self._var_metadata:
      self._var_metadata['on_remove_axis'](self, axis_index, axis_name)

GraphVariableState = VariableState[VariableState[tp.Any]]

def _variable_state_flatten(x: VariableState[tp.Any], *, with_keys: bool):
  metadata = tuple(x.get_metadata().items())
  if with_keys:
    node = (jtu.GetAttrKey('value'), x.raw_value)
  else:
    node = x.raw_value

  return (node,), (x.type, metadata)


def _variable_state_unflatten(
  static: tuple[type[Variable[A]], tuple[tuple[str, tp.Any], ...]],
  children: tuple[A],
) -> VariableState[A]:
  return VariableState(
    type=static[0],
    value=children[0],
    **dict(static[1]),
  )


jtu.register_pytree_with_keys(
  VariableState,
  partial(_variable_state_flatten, with_keys=True),  # type: ignore
  _variable_state_unflatten,  # type: ignore
  flatten_func=partial(_variable_state_flatten, with_keys=False),  # type: ignore
)


def with_metadata(
  initializer: F,
  set_value_hooks: tp.Union[SetValueHook[A], tp.Sequence[SetValueHook[A]]] = (),
  get_value_hooks: tp.Union[SetValueHook[A], tp.Sequence[SetValueHook[A]]] = (),
  create_value_hooks: tp.Union[
    CreateValueHook[A], tp.Sequence[CreateValueHook[A]]
  ] = (),
  add_axis_hooks: tp.Union[
    AddAxisHook[Variable[A]], tp.Sequence[AddAxisHook[Variable[A]]]
  ] = (),
  remove_axis_hooks: tp.Union[
    RemoveAxisHook[Variable[A]],
    tp.Sequence[RemoveAxisHook[Variable[A]]],
  ] = (),
  **metadata: tp.Any,
) -> F:
  if set_value_hooks:
    if callable(set_value_hooks):
      set_value_hooks = (set_value_hooks,)
    else:
      set_value_hooks = tuple(set_value_hooks)
  else:
    set_value_hooks = ()

  if get_value_hooks:
    if callable(get_value_hooks):
      get_value_hooks = (get_value_hooks,)
    else:
      get_value_hooks = tuple(get_value_hooks)
  else:
    get_value_hooks = ()

  if create_value_hooks:
    if callable(create_value_hooks):
      create_value_hooks = (create_value_hooks,)
    else:
      create_value_hooks = tuple(create_value_hooks)
  else:
    create_value_hooks = ()

  if add_axis_hooks:
    if callable(add_axis_hooks):
      add_axis_hooks = (add_axis_hooks,)
    else:
      add_axis_hooks = tuple(add_axis_hooks)
  else:
    add_axis_hooks = ()

  if remove_axis_hooks:
    if callable(remove_axis_hooks):
      remove_axis_hooks = (remove_axis_hooks,)
    else:
      remove_axis_hooks = tuple(remove_axis_hooks)
  else:
    remove_axis_hooks = ()

  @functools.wraps(initializer)
  def wrapper(*args):
    return VariableMetadata(
      initializer(*args),
      set_value_hooks=set_value_hooks,
      get_value_hooks=get_value_hooks,
      create_value_hooks=create_value_hooks,
      add_axis_hooks=add_axis_hooks,
      remove_axis_hooks=remove_axis_hooks,
      metadata=metadata,
    )

  return wrapper  # type: ignore


def split_flat_state(
    flat_state: tp.Iterable[tuple[PathParts, Variable | VariableState]],
    filters: tuple[filterlib.Filter, ...],
) -> tuple[list[tuple[PathParts, Variable | VariableState]], ...]:
  predicates = filterlib.filters_to_predicates(filters)
  # we have n + 1 states, where n is the number of predicates
  # the last state is for values that don't match any predicate
  flat_states: tuple[list[tuple[PathParts, Variable | VariableState]], ...] = (
    tuple([] for _ in predicates)
  )

  for path, value in flat_state:
    for i, predicate in enumerate(predicates):
      if predicate(path, value):
        flat_states[i].append((path, value))
        break
    else:
      raise ValueError(
          'Non-exhaustive filters, got a non-empty remainder: '
          f'{path} -> {value}.'
          '
Use `...` to match all remaining elements.'
      )

  return flat_states


###################################################
### Variable type/class <-> string name mapping ###
###################################################
# Assumption: the mapping is 1-1 and unique.

VariableTypeCache: dict[str, tp.Type[Variable[tp.Any]]] = {}


def variable_type_from_name(
  name: str,
  /,
  *,
  base: type[Variable[tp.Any]] = Variable,
  allow_register: bool = False,
) -> tp.Type[Variable[tp.Any]]:
  """"""Given a Linen-style collection name, get or create its NNX Variable class.""""""
  if name not in VariableTypeCache:
    if not allow_register:
      raise ValueError(
        f'Name {name} is not registered in the registry. '
        'To register a new name, use register_variable_name() '
        'or set allow_register=True.'
      )
    VariableTypeCache[name] = type(name, (base,), {})
  return VariableTypeCache[name]


def variable_name_from_type(
  typ: tp.Type[Variable[tp.Any]], /, *, allow_register: bool = False
) -> str:
  """"""Given an NNX Variable type, get its Linen-style collection name.

  Should output the exact inversed result of `variable_type_from_name()`.""""""
  for name, t in VariableTypeCache.items():
    if typ == t:
      return name

  if not allow_register:
    raise ValueError(
      f'Type {typ} is not registered in the registry. '
      'To register a new type, use register_variable_name() '
      'or set allow_register=True.'
    )
  name = typ.__name__
  if name in VariableTypeCache:
    raise ValueError(
      'Name {name} is already registered in the registry as {VariableTypeCache[name]}. '
      'It cannot be linked with this type {typ}.'
    )
  register_variable_name(name, typ)
  return name


class _Missing:
  pass


_MISSING = _Missing()


@tp.overload
def register_variable_name(
    name: str,
    typ: type[Variable[tp.Any]],
    *,
    overwrite: bool = False,
) -> type[Variable[tp.Any]]:
  ...


@tp.overload
def register_variable_name(
    name: str,
    *,
    overwrite: bool = False,
) -> tp.Callable[[type[Variable[tp.Any]]], type[Variable[tp.Any]]]:
  ...


def register_variable_name(
    name: str,
    typ: type[Variable[A]] | _Missing = _MISSING,
    *,
    overwrite=False,
) -> type[Variable[A]] | tp.Callable[[type[Variable[A]]], type[Variable[A]]]:
  """"""Register a pair of Linen collection name and its NNX type.""""""
  if typ is _MISSING:
    return partial(register_variable_name, name, overwrite=overwrite)
  typ = tp.cast(type[Variable[A]], typ)
  if not overwrite and name in VariableTypeCache:
    raise ValueError(f'Name {name} already mapped to type {VariableTypeCache[name]}. '
                     'To overwrite, call set_variable_name() with `overwrite=True`.')
  VariableTypeCache[name] = typ
  return typ


# add known variable type names
register_variable_name('params', Param)
register_variable_name('batch_stats', BatchStat)
register_variable_name('cache', Cache)
register_variable_name('intermediates', Intermediate)
register_variable_name('perturbations', Perturbation)
",python,tab
4135,12367519,".venv/lib/python3.10/site-packages/flax/nnx/variablelib.py",24983,0,"",python,selection_command
4136,12486618,".venv/lib/python3.10/site-packages/flax/nnx/variablelib.py",0,0,"",python,tab
4137,13396178,"sample.py",0,0,"",python,tab
4138,13398892,"sample.py",5581,0,"",python,selection_mouse
4139,13398892,"sample.py",5580,0,"",python,selection_command
4140,13400167,"sample.py",5513,0,"",python,selection_mouse
4141,13424369,"genie.py",0,0,"",python,tab
4142,13425259,"genie.py",17025,0,"",python,selection_mouse
4143,13425265,"genie.py",17024,0,"",python,selection_command
4144,13429884,"genie.py",14320,0,"",python,selection_mouse
4145,13431876,"genie.py",14432,0,"",python,selection_mouse
4146,13434666,"genie.py",14336,0,"",python,selection_mouse
4147,13439833,"genie.py",14434,0,"",python,selection_command
4148,13440052,"genie.py",14435,0,"",python,selection_command
4149,13440353,"genie.py",14436,0,"",python,selection_command
4150,13440664,"genie.py",14435,0,"",python,selection_command
4151,13441094,"genie.py",14472,0,"",python,selection_command
4152,13446883,"genie.py",14373,0,"",python,selection_command
4153,13448215,"genie.py",14431,0,"",python,selection_command
4154,13449004,"genie.py",14435,0,"",python,selection_command
4155,13450105,"genie.py",14435,1,"S",python,content
4156,13450710,"genie.py",14337,0,"",python,selection_command
4157,13450742,"genie.py",14336,0,"",python,selection_command
4158,13451544,"genie.py",14336,1,"",python,content
4159,13454413,"genie.py",14336,0,".",python,content
4160,13454414,"genie.py",14337,0,"",python,selection_keyboard
4161,13454531,"genie.py",14337,0,"0",python,content
4162,13454531,"genie.py",14338,0,"",python,selection_keyboard
4163,13454992,"genie.py",14337,1,"",python,content
4164,13455128,"genie.py",14336,1,"",python,content
4165,13455457,"genie.py",14336,0,"()",python,content
4166,13455457,"genie.py",14337,0,"",python,selection_keyboard
4167,13455543,"genie.py",14337,1,")",python,content
4168,13455544,"genie.py",14338,0,"",python,selection_keyboard
4169,13456007,"genie.py",14337,0,"",python,selection_command
4170,13456572,"genie.py",14337,0,"N",python,content
4171,13456572,"genie.py",14338,0,"",python,selection_keyboard
4172,13456764,"genie.py",14338,0," ",python,content
4173,13456764,"genie.py",14339,0,"",python,selection_keyboard
4174,13457019,"genie.py",14339,0,"=",python,content
4175,13457019,"genie.py",14340,0,"",python,selection_keyboard
4176,13457381,"genie.py",14340,0," ",python,content
4177,13457381,"genie.py",14341,0,"",python,selection_keyboard
4178,13457581,"genie.py",14340,1,"",python,content
4179,13457777,"genie.py",14339,1,"",python,content
4180,13457904,"genie.py",14338,1,"",python,content
4181,13458396,"genie.py",14338,0," ",python,content
4182,13458397,"genie.py",14339,0,"",python,selection_keyboard
4183,13458697,"genie.py",14339,0,"+",python,content
4184,13458697,"genie.py",14340,0,"",python,selection_keyboard
4185,13459067,"genie.py",14340,0,"1",python,content
4186,13459068,"genie.py",14341,0,"",python,selection_keyboard
4187,13459492,"genie.py",14340,1,"",python,content
4188,13459680,"genie.py",14340,0," ",python,content
4189,13459681,"genie.py",14341,0,"",python,selection_keyboard
4190,13459897,"genie.py",14341,0,"1",python,content
4191,13459898,"genie.py",14342,0,"",python,selection_keyboard
4192,13460166,"genie.py",14341,0,"",python,selection_command
4193,13461097,"genie.py",14445,0,"",python,selection_command
4194,13461401,"genie.py",14444,0,"",python,selection_command
4195,13461568,"genie.py",14442,0,"",python,selection_command
4196,13461722,"genie.py",14441,0,"",python,selection_command
4197,13462165,"genie.py",14441,1,"",python,content
4198,13462616,"genie.py",14441,0,"s",python,content
4199,13462617,"genie.py",14442,0,"",python,selection_keyboard
4200,13462621,"genie.py",14442,0,"e",python,content
4201,13462621,"genie.py",14443,0,"",python,selection_keyboard
4202,13462904,"genie.py",14443,0,"q",python,content
4203,13462904,"genie.py",14444,0,"",python,selection_keyboard
4204,13463361,"genie.py",14444,0,"_",python,content
4205,13463361,"genie.py",14445,0,"",python,selection_keyboard
4206,13463667,"genie.py",14445,0,"l",python,content
4207,13463668,"genie.py",14446,0,"",python,selection_keyboard
4208,13463674,"genie.py",14446,0,"e",python,content
4209,13463674,"genie.py",14447,0,"",python,selection_keyboard
4210,13463770,"genie.py",14447,0,"n",python,content
4211,13463770,"genie.py",14448,0,"",python,selection_keyboard
4212,13463961,"genie.py",14447,0,"",python,selection_command
4213,13465618,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4214,13466235,"TERMINAL",0,0,"

",,terminal_output
4215,13468621,"genie.py",14484,0,"",python,selection_command
4216,13477452,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4217,13485948,"TERMINAL",0,0,"2025-08-01 13:19:54.639352: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4218,13490333,"TERMINAL",0,0,"2025-08-01 13:19:59.078386: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4219,13498446,"TERMINAL",0,0,"2025-08-01 13:20:07.188448: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4220,13502026,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in sample_causal
    final_carry_causal, _ = jax.lax.scan(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 126, in __call__
    out = self.f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 373, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 323, in __call__
    z_F1M = self.spatial_attention(z_F1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 551, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py"", line 1204, in dot_product_attention
    out = cudnn_dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 1981, in dot_product_attention
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 332, in check_layout
    raise ValueError(
ValueError: Bias must have same seq length as QKV, got 4 and 4
",,terminal_output
4221,13503149,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4222,13525412,"utils/nn.py",0,0,"",python,tab
4223,13525764,"utils/nn.py",18961,0,"",python,selection_command
4224,13528024,"utils/nn.py",18443,0,"",python,selection_mouse
4225,13534877,"utils/nn.py",18510,0,"",python,selection_mouse
4226,13650261,"utils/nn.py",18445,0,"",python,selection_mouse
4227,13658184,"utils/nn.py",18510,0,"",python,selection_command
4228,13658727,"utils/nn.py",18524,0,"",python,selection_command
4229,13658910,"utils/nn.py",18510,0,"",python,selection_command
4230,13659079,"utils/nn.py",18445,0,"",python,selection_command
4231,13670002,"/fast/home/franz.srambical/jafar/utils/nn.py",0,0,"import math
from typing import Tuple, Callable, List

from flax import nnx
import jax
import jax.numpy as jnp
import einops


class PositionalEncoding(nnx.Module):
    """"""https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/JAX/tutorial6/Transformers_and_MHAttention.html""""""

    def __init__(self, d_model: int, max_len: int = 5000):
        self.d_model = d_model
        self.max_len = max_len

        pe = jnp.zeros((self.max_len, self.d_model))
        position = jnp.arange(0, self.max_len, dtype=jnp.float32)[:, None]
        div_term = jnp.exp(
            jnp.arange(0, self.d_model, 2) * (-math.log(10000.0) / self.d_model)
        )
        pe = pe.at[:, 0::2].set(jnp.sin(position * div_term))
        pe = pe.at[:, 1::2].set(jnp.cos(position * div_term))
        self.pe = nnx.Variable(pe)

    def __call__(self, x: jax.Array) -> jax.Array:
        x = x + self.pe[: x.shape[2]]
        return x


class STBlock(nnx.Module):
    def __init__(
        self,
        dim: int,
        ffn_dim: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        rngs: nnx.Rngs,
    ):
        self.dim = dim
        self.ffn_dim = ffn_dim
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.spatial_pos_enc = PositionalEncoding(self.dim)
        self.spatial_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.spatial_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.dim,
            qkv_features=self.dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=False
            ),
            rngs=rngs,
            decode=False,
        )

        self.temporal_pos_enc = PositionalEncoding(self.dim)
        self.temporal_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.temporal_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.dim,
            qkv_features=self.dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=False,
        )

        self.ffn_norm = nnx.LayerNorm(
            num_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense1 = nnx.Linear(
            in_features=self.dim,
            out_features=self.ffn_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense2 = nnx.Linear(
            in_features=self.ffn_dim,
            out_features=self.dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    @nnx.remat
    def __call__(self, x_BTNM: jax.Array) -> jax.Array:
        # --- Spatial attention ---
        z_BTNM = self.spatial_pos_enc(x_BTNM)
        z_BTNM = self.spatial_norm(z_BTNM)
        z_BTNM = self.spatial_attention(z_BTNM)
        x_BTNM = x_BTNM + z_BTNM

        # --- Temporal attention ---
        x_BNTM = x_BTNM.swapaxes(1, 2)
        z_BNTM = self.temporal_pos_enc(x_BNTM)
        z_BNTM = self.temporal_norm(z_BNTM)
        z_BNTM = self.temporal_attention(z_BNTM)
        x_BNTM = x_BNTM + z_BNTM
        x_BTNM = x_BNTM.swapaxes(1, 2)

        # --- Feedforward ---
        z_BTNM = self.ffn_norm(x_BTNM)
        z_BTND = self.ffn_dense1(z_BTNM)
        z_BTND = jax.nn.gelu(z_BTND)
        z_BTNM = self.ffn_dense2(z_BTND)
        x_BTNM = x_BTNM + z_BTNM

        return x_BTNM


class STTransformer(nnx.Module):
    """"""
    Dimension keys:
        B: batch size
        T: number of frames
        N: number of patches per frame
        I: number of input features
        M: model dimension
        D: FFN dimension
        O: number of output features
    """"""
    def __init__(
        self,
        input_dim: int,
        model_dim: int,
        ffn_dim: int,
        out_dim: int,
        num_blocks: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        rngs: nnx.Rngs,
    ):
        self.input_dim = input_dim
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.out_dim = out_dim
        self.num_blocks = num_blocks
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.input_norm1 = nnx.LayerNorm(
            num_features=self.input_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_dense = nnx.Linear(
            in_features=self.input_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_norm2 = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

        self.blocks = []
        for _ in range(self.num_blocks):
            self.blocks.append(
                STBlock(
                    dim=self.model_dim,
                    ffn_dim=self.ffn_dim,
                    num_heads=self.num_heads,
                    dropout=self.dropout,
                    param_dtype=self.param_dtype,
                    dtype=self.dtype,
                    use_flash_attention=self.use_flash_attention,
                    rngs=rngs,
                )
            )

        self.output_dense = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.out_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    def __call__(self, x_BTNI: jax.Array) -> jax.Array:
        x_BTNI = self.input_norm1(x_BTNI)
        x_BTNM = self.input_dense(x_BTNI)
        x_BTNM = self.input_norm2(x_BTNM)

        for block in self.blocks:
            x_BTNM = block(x_BTNM)

        x_BTNO = self.output_dense(x_BTNM)
        return x_BTNO

class TransformerBlock(nnx.Module):
    def __init__(
        self,
        model_dim: int,
        ffn_dim: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
    ):
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention
        self.decode = decode

        self.temporal_pos_enc = PositionalEncoding(self.model_dim)
        self.spatial_pos_enc = PositionalEncoding(self.model_dim)
        self.temporal_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.spatial_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_norm = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.temporal_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.model_dim,
            qkv_features=self.model_dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=self.decode,
        )
        self.spatial_attention = nnx.MultiHeadAttention(
            num_heads=self.num_heads,
            in_features=self.model_dim,
            qkv_features=self.model_dim,
            dropout_rate=self.dropout,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            attention_fn=_create_flash_attention_fn(
                self.use_flash_attention, is_causal=True
            ),
            rngs=rngs,
            decode=self.decode,
        )
        self.ffn_dense1 = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.ffn_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.ffn_dense2 = nnx.Linear(
            in_features=self.ffn_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    @nnx.remat
    def __call__(self, x_BTNM: jax.Array, pos_index: Tuple[jax.Array, jax.Array] | None = None) -> jax.Array:
        # --- Spatial attention ---
        B, T, N, M = x_BTNM.shape
        z_FNM = einops.rearrange(x_BTNM, ""b t n m -> (b t) n m"")
        z_FNM = self.spatial_norm(z_FNM)
        if self.decode:
            assert pos_index is not None
            z_FM = z_FNM[:, pos_index[1]]
            z_F1M = jnp.reshape(z_FM, (B * T, 1, M))
            z_F1M = self.spatial_attention(z_F1M)
            z_FM = jnp.reshape(z_F1M, (B * T, M))
            z_FNM = z_FNM.at[:, pos_index[1], :].set(z_FM)
        else:
            z_FNM = self.spatial_attention(z_FNM)
        z_BTNM = einops.rearrange(z_FNM, ""(b t) n m -> b t n m"", t=T)
        x_BTNM = x_BTNM + z_BTNM
        # --- Temporal attention ---
        z_PTM = einops.rearrange(x_BTNM, ""b t n m -> (b n) t m"")
        z_PTM = self.temporal_norm(z_PTM)
        if self.decode:
            assert pos_index is not None
            z_PM = z_PTM[:, pos_index[0]]
            z_P1M = jnp.reshape(z_PM, (B * N, 1, M))
            z_P1M = self.temporal_attention(z_P1M)
            z_PM = jnp.reshape(z_P1M, (B * N, M))
            z_PTM = z_PTM.at[:, pos_index[0], :].set(z_PM)
        else:
            z_PTM = self.temporal_attention(z_PTM)
        z_BTNM = einops.rearrange(z_PTM, ""(b n) t m -> b t n m"", n=N)
        x_BTNM = x_BTNM + z_BTNM
        # --- Feedforward ---
        z_BTNM = self.ffn_norm(x_BTNM)
        z_BTND = self.ffn_dense1(z_BTNM)
        z_BTND = jax.nn.gelu(z_BTND)
        z_BTNM = self.ffn_dense2(z_BTND)
        x_BTNM = x_BTNM + z_BTNM

        return x_BTNM

class Transformer(nnx.Module):
    """"""
    Dimension keys:
        B: batch size
        T: number of frames
        N: number of patches per frame
        I: number of input features
        M: model dimension
        D: FFN dimension
        O: number of output features
        F: number of frames in batch
        P: number of patch positions in batch
    """"""
    def __init__(
        self,
        input_dim: int,
        model_dim: int,
        ffn_dim: int,
        out_dim: int,
        num_blocks: int,
        num_heads: int,
        dropout: float,
        param_dtype: jnp.dtype,
        dtype: jnp.dtype,
        use_flash_attention: bool,
        decode: bool,
        rngs: nnx.Rngs,
    ):
        self.input_dim = input_dim
        self.model_dim = model_dim
        self.ffn_dim = ffn_dim
        self.out_dim = out_dim
        self.num_blocks = num_blocks
        self.num_heads = num_heads
        self.dropout = dropout
        self.param_dtype = param_dtype
        self.dtype = dtype
        self.use_flash_attention = use_flash_attention

        self.pos_enc = PositionalEncoding(self.model_dim)
        self.input_norm1 = nnx.LayerNorm(
            num_features=self.input_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_dense = nnx.Linear(
            in_features=self.input_dim,
            out_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )
        self.input_norm2 = nnx.LayerNorm(
            num_features=self.model_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

        self.blocks: List[TransformerBlock] = []
        for _ in range(self.num_blocks):
            self.blocks.append(
                TransformerBlock(
                    model_dim=self.model_dim,
                    ffn_dim=self.ffn_dim,
                    num_heads=self.num_heads,
                    dropout=self.dropout,
                    param_dtype=self.param_dtype,
                    dtype=self.dtype,
                    use_flash_attention=self.use_flash_attention,
                    decode=decode,
                    rngs=rngs,
                )
            )
        self.output_dense = nnx.Linear(
            in_features=self.model_dim,
            out_features=self.out_dim,
            param_dtype=self.param_dtype,
            dtype=self.dtype,
            rngs=rngs,
        )

    def __call__(self, x_BTNI: jax.Array, pos_index: Tuple[jax.Array, jax.Array] | None = None) -> jax.Array:
        x_BTNI = self.input_norm1(x_BTNI)
        x_BTNM = self.input_dense(x_BTNI)
        x_BTNM = self.input_norm2(x_BTNM)
        x_BTNM = self.pos_enc(x_BTNM)

        for block in self.blocks:
            x_BTNM = block(x_BTNM, pos_index)

        x_BTNV = self.output_dense(x_BTNM)
        return x_BTNV

def normalize(x: jax.Array) -> jax.Array:
    return x / (jnp.linalg.norm(x, ord=2, axis=-1, keepdims=True) + 1e-8)


class VectorQuantizer(nnx.Module):
    """"""
    Dimension keys:
        D: B * T * N
        K: number of latents
        L: latent dimension
    """"""
    def __init__(
        self, latent_dim: int, num_latents: int, dropout: float, rngs: nnx.Rngs
    ):
        self.latent_dim = latent_dim
        self.num_latents = num_latents
        self.dropout = dropout

        self.codebook = nnx.Param(
            normalize(
                nnx.initializers.lecun_uniform()(
                    rngs.params(), (self.num_latents, self.latent_dim)
                )
            )
        )
        self.drop = nnx.Dropout(self.dropout, rngs=rngs)

    def __call__(
        self, x_DL: jax.Array, training: bool
    ) -> Tuple[jax.Array, jax.Array, jax.Array, jax.Array]:
        # --- Compute distances ---
        x_DL = normalize(x_DL)
        normalized_codebook_KL = normalize(self.codebook.value)
        distance_DK = -jnp.matmul(x_DL, normalized_codebook_KL.T)
        if training:
            distance_DK = self.drop(distance_DK)

        # --- Get indices and embeddings ---
        indices_D = jnp.argmin(distance_DK, axis=-1)
        z_DL = self.codebook[indices_D]

        # --- Straight through estimator ---
        z_q_DL = x_DL + jax.lax.stop_gradient(z_DL - x_DL)
        return z_q_DL, z_DL, x_DL, indices_D

    def get_codes(self, indices_E: jax.Array) -> jax.Array:
        return self.codebook[indices_E]


def _create_flash_attention_fn(use_flash_attention: bool, is_causal: bool) -> Callable:
    """"""
    Create an attention function that uses flash attention if enabled.

    flax.nnx.MultiHeadAttention provides tensors with shape (batch..., length, num_heads, head_dim),
    but jax.nn.dot_product_attention expects (batch, length, num_heads, head_dim). We reshape to
    ensure compatibility. cuDNN's flash attention additionally requires a sequence length that
    is a multiple of 4. We pad the sequence length to the nearest multiple of 4 and mask
    accordingly. Note that cuDNN requires the mask to be broadcast before calling the attention
    function due to strict shape checking.
    """"""

    # FIXME (f.srambical): keys and values could have different dimensionalities
    def attention_fn(query_BSHD, key_BSHD, value_BSHD, bias=None, mask_B111=None, **kwargs):
        implementation = ""cudnn"" if use_flash_attention else None

        def _merge_batch_dims(x):
            return einops.rearrange(x, ""... l h k -> (...) l h k"")

        def _pad(x):
            return jnp.pad(x, ((0, 0), (0, pad_size), (0, 0), (0, 0)))

        original_shape = query_BSHD.shape
        original_seq_len = query_BSHD.shape[-3]

        # Pad to nearest multiple of 4
        T = ((original_seq_len + 3) // 4) * 4
        pad_size = T - original_seq_len

        query_BTHD = _pad(_merge_batch_dims(query_BSHD))
        key_BTHD = _pad(_merge_batch_dims(key_BSHD))
        value_BTHD = _pad(_merge_batch_dims(value_BSHD))
        B = query_BTHD.shape[0]

        attention_mask = jnp.ones((T, T), dtype=jnp.bool_)
        attention_mask = attention_mask.at[original_seq_len:, :].set(False)
        attention_mask = attention_mask.at[:, original_seq_len:].set(False)

        # Handle causal mask for cached decoder self-attention (from nnx.MultiHeadAttention)
        if mask_B111 is not None:
            mask_B111 = _merge_batch_dims(mask_B111)
            # We need to broadcast T and S dimensions to target_seq_len since cudnn attention strictly checks the mask shape
            # https://github.com/jax-ml/jax/issues/28974
            # https://github.com/jax-ml/jax/blob/08c7677393672ccb85c10f1ed0bd506905c3c994/jax/_src/cudnn/fused_attention_stablehlo.py#L1830
            # https://github.com/jax-ml/jax/blob/08c7677393672ccb85c10f1ed0bd506905c3c994/jax/_src/cudnn/fused_attention_stablehlo.py#L337
            mask_B1QK = einops.repeat(mask_B111, ""... 1 1 -> ... t s"", t=T, s=T)
            mask_B1QK = mask_B111.astype(jnp.bool)
        else:
            mask_11QK = attention_mask[jnp.newaxis, jnp.newaxis, :, :]
            mask_B1QK = jnp.broadcast_to(mask_11QK, (B, 1, T, T))

        bias_4d = _pad(_merge_batch_dims(bias)) if bias is not None else None

        # NOTE: jax.nn.dot_product_attention does not support dropout
        output_4d = jax.nn.dot_product_attention(
            query=query_BTHD,
            key=key_BTHD,
            value=value_BTHD,
            bias=bias_4d,
            mask=mask_B1QK,
            implementation=implementation,
            is_causal=is_causal,
        )
        return output_4d[..., :original_seq_len, :, :].reshape(original_shape)

    return attention_fn
",python,tab
4232,13670002,"/fast/home/franz.srambical/jafar/utils/nn.py",18581,0,"",python,selection_command
4233,13672106,"/fast/home/franz.srambical/jafar/utils/nn.py",18580,0,"
        ",python,content
4234,13672300,"/fast/home/franz.srambical/jafar/utils/nn.py",18589,0,"b",python,content
4235,13672301,"/fast/home/franz.srambical/jafar/utils/nn.py",18590,0,"",python,selection_keyboard
4236,13672505,"/fast/home/franz.srambical/jafar/utils/nn.py",18590,0,"r",python,content
4237,13672506,"/fast/home/franz.srambical/jafar/utils/nn.py",18591,0,"",python,selection_keyboard
4238,13672552,"/fast/home/franz.srambical/jafar/utils/nn.py",18591,0,"e",python,content
4239,13672552,"/fast/home/franz.srambical/jafar/utils/nn.py",18592,0,"",python,selection_keyboard
4240,13672788,"/fast/home/franz.srambical/jafar/utils/nn.py",18592,0,"a",python,content
4241,13672789,"/fast/home/franz.srambical/jafar/utils/nn.py",18593,0,"",python,selection_keyboard
4242,13672807,"/fast/home/franz.srambical/jafar/utils/nn.py",18593,0,"k",python,content
4243,13672807,"/fast/home/franz.srambical/jafar/utils/nn.py",18594,0,"",python,selection_keyboard
4244,13672870,"/fast/home/franz.srambical/jafar/utils/nn.py",18594,0,"p",python,content
4245,13672870,"/fast/home/franz.srambical/jafar/utils/nn.py",18595,0,"",python,selection_keyboard
4246,13672939,"/fast/home/franz.srambical/jafar/utils/nn.py",18595,0,"o",python,content
4247,13672939,"/fast/home/franz.srambical/jafar/utils/nn.py",18596,0,"",python,selection_keyboard
4248,13673125,"/fast/home/franz.srambical/jafar/utils/nn.py",18596,0,"i",python,content
4249,13673125,"/fast/home/franz.srambical/jafar/utils/nn.py",18597,0,"",python,selection_keyboard
4250,13673183,"/fast/home/franz.srambical/jafar/utils/nn.py",18597,0,"n",python,content
4251,13673183,"/fast/home/franz.srambical/jafar/utils/nn.py",18598,0,"",python,selection_keyboard
4252,13673270,"/fast/home/franz.srambical/jafar/utils/nn.py",18598,0,"t",python,content
4253,13673271,"/fast/home/franz.srambical/jafar/utils/nn.py",18599,0,"",python,selection_keyboard
4254,13673847,"/fast/home/franz.srambical/jafar/utils/nn.py",18599,0,"()",python,content
4255,13673847,"/fast/home/franz.srambical/jafar/utils/nn.py",18600,0,"",python,selection_keyboard
4256,13673922,"/fast/home/franz.srambical/jafar/utils/nn.py",18600,1,")",python,content
4257,13673922,"/fast/home/franz.srambical/jafar/utils/nn.py",18601,0,"",python,selection_keyboard
4258,13674474,"/fast/home/franz.srambical/jafar/utils/nn.py",18600,0,"",python,selection_command
4259,13674908,"/fast/home/franz.srambical/jafar/utils/nn.py",18599,0,"",python,selection_command
4260,13676419,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4261,13676625,"TERMINAL",0,0,"

",,terminal_output
4262,13688086,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4263,13697709,"TERMINAL",0,0,"> /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4264,13698640,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",0,0,"# Copyright 2019 The JAX Authors.
#
# Licensed under the Apache License, Version 2.0 (the ""License"");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an ""AS IS"" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

""""""Shared neural network activations and other functions.""""""

from __future__ import annotations

from collections.abc import Sequence
from functools import partial
import operator
import math
import numpy as np
from typing import Any, List, Literal
import warnings

import jax
import jax.numpy as jnp
from jax import custom_jvp
from jax import lax
from jax._src import config
from jax._src import core
from jax._src import deprecations
from jax._src import dtypes
from jax._src import util
from jax._src.core import AxisName
from jax._src.sharding_impls import NamedSharding, PartitionSpec as P
from jax._src.cudnn.fused_attention_stablehlo import (
    dot_product_attention as cudnn_dot_product_attention, MaskType)
from jax._src.cudnn.scaled_matmul_stablehlo import (
    scaled_matmul_wrapper as cudnn_scaled_matmul,
    scaled_dot_general_wrapper as cudnn_scaled_dot_general,
    BlockScaleConfig)
from jax._src.interpreters import batching
from jax._src.interpreters import mlir
from jax._src.numpy import util as numpy_util
from jax._src.typing import Array, ArrayLike, DType, DTypeLike
from jax._src.ops.special import logsumexp as _logsumexp


# activations
@jax.jit
def identity(x: ArrayLike) -> Array:
  r""""""Identity activation function.

  Returns the argument unmodified.

  Args:
    x : input array

  Returns:
    The argument `x` unmodified.

  Examples:
    >>> jax.nn.identity(jax.numpy.array([-2., -1., -0.5, 0, 0.5, 1., 2.]))
    Array([-2. , -1. , -0.5, 0. , 0.5, 1. , 2. ], dtype=float32)

  """"""
  numpy_util.check_arraylike(""identity"", x)
  return jnp.asarray(x)

@custom_jvp
@jax.jit
def relu(x: ArrayLike) -> Array:
  r""""""Rectified linear unit activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{relu}(x) = \max(x, 0)

  except under differentiation, we take:

  .. math::
    
abla \mathrm{relu}(0) = 0

  For more information see
  `Numerical influence of ReLU’(0) on backpropagation
  <https://dl.acm.org/doi/10.5555/3540261.3540297>`_.

  Args:
    x : input array

  Returns:
    An array.

  Examples:
    >>> jax.nn.relu(jax.numpy.array([-2., -1., -0.5, 0, 0.5, 1., 2.]))
    Array([0. , 0. , 0. , 0. , 0.5, 1. , 2. ], dtype=float32)

  See also:
    :func:`relu6`

  """"""
  return jnp.maximum(x, 0)
# For behavior at 0, see https://dl.acm.org/doi/10.5555/3540261.3540297
relu.defjvps(lambda g, ans, x: lax.select(x > 0, g, lax.full_like(g, 0)))

@jax.jit
def squareplus(x: ArrayLike, b: ArrayLike = 4) -> Array:
  r""""""Squareplus activation function.

  Computes the element-wise function

  .. math::
    \mathrm{squareplus}(x) = \frac{x + \sqrt{x^2 + b}}{2}

  as described in https://arxiv.org/abs/2112.11687.

  Args:
    x : input array
    b : smoothness parameter
  """"""
  numpy_util.check_arraylike(""squareplus"", x)
  numpy_util.check_arraylike(""squareplus"", b)
  x = jnp.asarray(x)
  b = jnp.asarray(b)
  y = x + jnp.sqrt(jnp.square(x) + b)
  return y / 2

@jax.jit
def softplus(x: ArrayLike) -> Array:
  r""""""Softplus activation function.

  Computes the element-wise function

  .. math::
    \mathrm{softplus}(x) = \log(1 + e^x)

  Args:
    x : input array
  """"""
  return jnp.logaddexp(x, 0)

@jax.jit
def sparse_plus(x: ArrayLike) -> Array:
  r""""""Sparse plus function.

  Computes the function:

  .. math::

    \mathrm{sparse\_plus}(x) = \begin{cases}
      0, & x \leq -1\\
      \frac{1}{4}(x+1)^2, & -1 < x < 1 \\
      x, & 1 \leq x
    \end{cases}

  This is the twin function of the softplus activation ensuring a zero output
  for inputs less than -1 and a linear output for inputs greater than 1,
  while remaining smooth, convex, monotonic by an adequate definition between
  -1 and 1.

  Args:
    x: input (float)
  """"""
  numpy_util.check_arraylike(""sparse_plus"", x)
  x = jnp.asarray(x)
  return jnp.where(x <= -1.0, 0.0, jnp.where(x >= 1.0, x, (x + 1.0)**2/4))

@jax.jit
def soft_sign(x: ArrayLike) -> Array:
  r""""""Soft-sign activation function.

  Computes the element-wise function

  .. math::
    \mathrm{soft\_sign}(x) = \frac{x}{|x| + 1}

  Args:
    x : input array
  """"""
  numpy_util.check_arraylike(""soft_sign"", x)
  x_arr = jnp.asarray(x)
  return x_arr / (jnp.abs(x_arr) + 1)

@partial(jax.jit, inline=True)
def sigmoid(x: ArrayLike) -> Array:
  r""""""Sigmoid activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{sigmoid}(x) = \frac{1}{1 + e^{-x}}

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`log_sigmoid`

  """"""
  return lax.logistic(x)

@jax.jit
def sparse_sigmoid(x: ArrayLike) -> Array:
  r""""""Sparse sigmoid activation function.

  Computes the function:

  .. math::

    \mathrm{sparse\_sigmoid}(x) = \begin{cases}
      0, & x \leq -1\\
      \frac{1}{2}(x+1), & -1 < x < 1 \\
      1, & 1 \leq x
    \end{cases}

  This is the twin function of the ``sigmoid`` activation ensuring a zero output
  for inputs less than -1, a 1 output for inputs greater than 1, and a linear
  output for inputs between -1 and 1. It is the derivative of ``sparse_plus``.

  For more information, see `Learning with Fenchel-Young Losses (section 6.2)
  <https://arxiv.org/abs/1901.02324>`_.

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`sigmoid`
  """"""
  return 0.5 * jnp.clip(x + 1.0, 0.0, 2.0)

@jax.jit
def silu(x: ArrayLike) -> Array:
  r""""""SiLU (aka swish) activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{silu}(x) = x \cdot \mathrm{sigmoid}(x) = \frac{x}{1 + e^{-x}}

  :func:`swish` and :func:`silu` are both aliases for the same function.

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`sigmoid`
  """"""
  numpy_util.check_arraylike(""silu"", x)
  x_arr = jnp.asarray(x)
  return x_arr * sigmoid(x_arr)

swish = silu

@jax.jit
def mish(x: ArrayLike) -> Array:
  r""""""Mish activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{mish}(x) = x \cdot \mathrm{tanh}(\mathrm{softplus}(x))

  For more information, see
  `Mish: A Self Regularized Non-Monotonic Activation Function
  <https://arxiv.org/abs/1908.08681>`_.

  Args:
    x : input array

  Returns:
    An array.
  """"""
  numpy_util.check_arraylike(""mish"", x)
  x_arr = jnp.asarray(x)
  return x_arr * jnp.tanh(softplus(x_arr))

@jax.jit
def log_sigmoid(x: ArrayLike) -> Array:
  r""""""Log-sigmoid activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{log\_sigmoid}(x) = \log(\mathrm{sigmoid}(x)) = -\log(1 + e^{-x})

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`sigmoid`
  """"""
  numpy_util.check_arraylike(""log_sigmoid"", x)
  x_arr = jnp.asarray(x)
  return -softplus(-x_arr)

@jax.jit
def elu(x: ArrayLike, alpha: ArrayLike = 1.0) -> Array:
  r""""""Exponential linear unit activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{elu}(x) = \begin{cases}
      x, & x > 0\\
      \alpha \left(\exp(x) - 1
ight), & x \le 0
    \end{cases}

  Args:
    x : input array
    alpha : scalar or array of alpha values (default: 1.0)

  Returns:
    An array.

  See also:
    :func:`selu`
  """"""
  numpy_util.check_arraylike(""elu"", x)
  x_arr = jnp.asarray(x)
  return jnp.where(x_arr > 0,
                   x_arr,
                   alpha * jnp.expm1(jnp.where(x_arr > 0, 0., x_arr)))

@jax.jit
def leaky_relu(x: ArrayLike, negative_slope: ArrayLike = 1e-2) -> Array:
  r""""""Leaky rectified linear unit activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{leaky\_relu}(x) = \begin{cases}
      x, & x \ge 0\\
      \alpha x, & x < 0
    \end{cases}

  where :math:`\alpha` = :code:`negative_slope`.

  Args:
    x : input array
    negative_slope : array or scalar specifying the negative slope (default: 0.01)

  Returns:
    An array.

  See also:
    :func:`relu`
  """"""
  numpy_util.check_arraylike(""leaky_relu"", x)
  x_arr = jnp.asarray(x)
  return jnp.where(x_arr >= 0, x_arr, negative_slope * x_arr)

@jax.jit
def hard_tanh(x: ArrayLike) -> Array:
  r""""""Hard :math:`\mathrm{tanh}` activation function.

  Computes the element-wise function:

  .. math::
    \mathrm{hard\_tanh}(x) = \begin{cases}
      -1, & x < -1\\
      x, & -1 \le x \le 1\\
      1, & 1 < x
    \end{cases}

  Args:
    x : input array

  Returns:
    An array.
  """"""
  numpy_util.check_arraylike(""hard_tanh"", x)
  x_arr = jnp.asarray(x)
  return jnp.where(x_arr > 1, 1, jnp.where(x_arr < -1, -1, x_arr))

@jax.jit
def celu(x: ArrayLike, alpha: ArrayLike = 1.0) -> Array:
  r""""""Continuously-differentiable exponential linear unit activation.

  Computes the element-wise function:

  .. math::
    \mathrm{celu}(x) = \begin{cases}
      x, & x > 0\\
      \alpha \left(\exp(\frac{x}{\alpha}) - 1
ight), & x \le 0
    \end{cases}

  For more information, see
  `Continuously Differentiable Exponential Linear Units
  <https://arxiv.org/abs/1704.07483>`_.

  Args:
    x : input array
    alpha : array or scalar (default: 1.0)

  Returns:
    An array.
  """"""
  return jnp.maximum(x, 0.0) + alpha * jnp.expm1(jnp.minimum(x, 0.0) / alpha)

@jax.jit
def selu(x: ArrayLike) -> Array:
  r""""""Scaled exponential linear unit activation.

  Computes the element-wise function:

  .. math::
    \mathrm{selu}(x) = \lambda \begin{cases}
      x, & x > 0\\
      \alpha e^x - \alpha, & x \le 0
    \end{cases}

  where :math:`\lambda = 1.0507009873554804934193349852946` and
  :math:`\alpha = 1.6732632423543772848170429916717`.

  For more information, see
  `Self-Normalizing Neural Networks
  <https://arxiv.org/abs/1706.02515>`_.

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`elu`
  """"""
  alpha = 1.6732632423543772848170429916717
  scale = 1.0507009873554804934193349852946
  return scale * elu(x, alpha)

# TODO(phawkins): this jit was found to change numerics in a test. Debug this.
# @partial(jax.jit, static_argnames=(""approximate"",))
def gelu(x: ArrayLike, approximate: bool = True) -> Array:
  r""""""Gaussian error linear unit activation function.

  If ``approximate=False``, computes the element-wise function:

  .. math::
    \mathrm{gelu}(x) = \frac{x}{2} \left(\mathrm{erfc} \left(
      \frac{-x}{\sqrt{2}} 
ight) 
ight)

  If ``approximate=True``, uses the approximate formulation of GELU:

  .. math::
    \mathrm{gelu}(x) = \frac{x}{2} \left(1 + \mathrm{tanh} \left(
      \sqrt{\frac{2}{\pi}} \left(x + 0.044715 x^3 
ight) 
ight) 
ight)

  For more information, see `Gaussian Error Linear Units (GELUs)
  <https://arxiv.org/abs/1606.08415>`_, section 2.

  Args:
    x: input array
    approximate: whether to use the approximate or exact formulation.
  """"""
  [x_arr] = numpy_util.promote_args_inexact(""gelu"", x)

  if approximate:
    sqrt_2_over_pi = np.sqrt(2 / np.pi).astype(x_arr.dtype)
    cdf = 0.5 * (1.0 + jnp.tanh(sqrt_2_over_pi * (x_arr + 0.044715 * (x_arr ** 3))))
    return x_arr * cdf
  else:
    sqrt_half = np.sqrt(0.5).astype(x_arr.dtype)
    return jnp.array(
        0.5 * x_arr * (lax.erfc(-x_arr * sqrt_half)), dtype=x_arr.dtype
    )

@partial(jax.jit, static_argnames=(""axis"",))
def glu(x: ArrayLike, axis: int = -1) -> Array:
  r""""""Gated linear unit activation function.

  Computes the function:

  .. math::
    \mathrm{glu}(x) =  x\left[\ldots, 0:\frac{n}{2}, \ldots
ight] \cdot
      \mathrm{sigmoid} \left( x\left[\ldots, \frac{n}{2}:n, \ldots
ight]
        
ight)

  where the array is split into two along ``axis``. The size of the ``axis``
  dimension must be divisible by two.

  Args:
    x : input array
    axis: the axis along which the split should be computed (default: -1)

  Returns:
    An array.

  See also:
    :func:`sigmoid`
  """"""
  numpy_util.check_arraylike(""glu"", x)
  x_arr = jnp.asarray(x)
  size = x_arr.shape[axis]
  assert size % 2 == 0, ""axis size must be divisible by 2""
  x1, x2 = jnp.split(x_arr, 2, axis)
  return x1 * sigmoid(x2)

# other functions

logsumexp = _logsumexp


@partial(jax.jit, static_argnames=(""axis"",))
def log_softmax(x: ArrayLike,
                axis: int | tuple[int, ...] | None = -1,
                where: ArrayLike | None = None) -> Array:
  r""""""Log-Softmax function.

  Computes the logarithm of the :code:`softmax` function, which rescales
  elements to the range :math:`[-\infty, 0)`.

  .. math ::
    \mathrm{log\_softmax}(x)_i = \log \left( \frac{\exp(x_i)}{\sum_j \exp(x_j)}
    
ight)

  Args:
    x : input array
    axis: the axis or axes along which the :code:`log_softmax` should be
      computed. Either an integer or a tuple of integers.
    where: Elements to include in the :code:`log_softmax`.

  Returns:
    An array.

  Note:
    If any input values are ``+inf``, the result will be all ``NaN``: this reflects the
    fact that ``inf / inf`` is not well-defined in the context of floating-point math.

  See also:
    :func:`softmax`
  """"""
  numpy_util.check_arraylike(""log_softmax"", x)
  x_arr = jnp.asarray(x)
  x_max = jnp.max(x_arr, axis, where=where, initial=-jnp.inf, keepdims=True)
  x_safe = x_arr if where is None else jnp.where(where, x_arr, -jnp.inf)
  shifted = x_safe - lax.stop_gradient(x_max)
  shifted_logsumexp = jnp.log(
      jnp.sum(jnp.exp(shifted), axis, where=where, keepdims=True))
  result = shifted - shifted_logsumexp
  if where is not None:
    return jnp.where(where, result, -jnp.inf)
  return result


# TODO(phawkins): this jit was found to change numerics in a test. Debug this.
# @partial(jax.jit, static_argnames=(""axis"",))
def softmax(x: ArrayLike,
            axis: int | tuple[int, ...] | None = -1,
            where: ArrayLike | None = None) -> Array:
  r""""""Softmax function.

  Computes the function which rescales elements to the range :math:`[0, 1]`
  such that the elements along :code:`axis` sum to :math:`1`.

  .. math ::
    \mathrm{softmax}(x) = \frac{\exp(x_i)}{\sum_j \exp(x_j)}

  Args:
    x : input array
    axis: the axis or axes along which the softmax should be computed. The
      softmax output summed across these dimensions should sum to :math:`1`.
      Either an integer or a tuple of integers.
    where: Elements to include in the :code:`softmax`.

  Returns:
    An array.

  Note:
    If any input values are ``+inf``, the result will be all ``NaN``: this reflects the
    fact that ``inf / inf`` is not well-defined in the context of floating-point math.

  See also:
    :func:`log_softmax`
  """"""
  if config.softmax_custom_jvp.value:
    # mypy is confused by the `functools.partial` application in the definition
    # of `_softmax` and incorrectly concludes that `_softmax` returns
    # `ReturnValue` -- the unsubstituted type parameter of `custom_jvp`.
    return _softmax(x, axis, where)
  else:
    return _softmax_deprecated(x, axis, where)

# TODO(mattjj): replace softmax with _softmax when deprecation flag is removed
@partial(jax.custom_jvp, nondiff_argnums=(1,))
def _softmax(
    x: ArrayLike,
    axis: int | tuple[int, ...] | None = -1,
    where: ArrayLike | None = None,
    initial: ArrayLike | None = -jnp.inf) -> Array:
  x_max = jnp.max(x, axis, where=where, initial=initial, keepdims=True)
  x_safe = x if where is None else jnp.where(where, x, initial)
  unnormalized = jnp.exp(x_safe - x_max)
  result = unnormalized / jnp.sum(unnormalized, axis, where=where, keepdims=True)
  if where is not None:
    result = jnp.where(where, result, 0)
  return result

@_softmax.defjvp
def _softmax_jvp(axis, primals, tangents):
  (x, where, initial), (x_dot, _, _) = primals, tangents
  y = _softmax(x, axis, where, initial)
  return y, y * (x_dot - (y * x_dot).sum(axis, where=where, keepdims=True))

def _softmax_deprecated(
    x: ArrayLike,
    axis: int | tuple[int, ...] | None = -1,
    where: ArrayLike | None = None,
    initial: ArrayLike | None = -jnp.inf) -> Array:
  x_max = jnp.max(x, axis, where=where, initial=initial, keepdims=True)
  x_safe = x if where is None else jnp.where(where, x, initial)
  unnormalized = jnp.exp(x_safe - lax.stop_gradient(x_max))
  result = unnormalized / jnp.sum(unnormalized, axis, where=where, keepdims=True)
  if where is not None:
    result = jnp.where(where, result, 0)
  return result


@partial(jax.jit, static_argnames=(""axis"",))
def standardize(x: ArrayLike,
                axis: int | tuple[int, ...] | None = -1,
                mean: ArrayLike | None = None,
                variance: ArrayLike | None = None,
                epsilon: ArrayLike = 1e-5,
                where: ArrayLike | None = None) -> Array:
  r""""""Standardizes input to zero mean and unit variance.

  The standardization is given by:

  .. math::

     x_{std} = \frac{x - \langle x
angle}{\sqrt{\langle(x - \langle x
angle)^2
angle + \epsilon}}

  where :math:`\langle x
angle` indicates the mean of :math:`x`, and :math:`\epsilon` is
  a small correction factor introduced to avoid division by zero.

  Args:
    x: input array to be standardized.
    axis: integer or tuple of integers representing the axes along which
      to standardize. Defaults to the last axis (``-1``).
    mean: optionally specify the mean used for standardization. If not specified,
      then ``x.mean(axis, where=where)`` will be used.
    variance: optionally specify the variance used for standardization. If not
      specified, then ``x.var(axis, where=where)`` will be used.
    epsilon: correction factor added to variance to avoid division by zero; defaults
      to ``1E-5``.
    where: optional boolean mask specifying which elements to use when computing
      the mean and variance.

  Returns:
    An array of the same shape as ``x`` containing the standardized input.
  """"""
  numpy_util.check_arraylike(""standardize"", x)
  numpy_util.check_arraylike_or_none(""standardize"", mean, variance, where)
  if mean is None:
    mean = jnp.mean(x, axis, keepdims=True, where=where)
  if variance is None:
    # this definition is traditionally seen as less accurate than jnp.var's
    # mean((x - mean(x))**2) but may be faster and even, given typical
    # activation distributions and low-precision arithmetic, more accurate
    # when used in neural network normalization layers
    variance = jnp.mean(
        jnp.square(x), axis, keepdims=True, where=where) - jnp.square(mean)
  return jnp.subtract(x, jnp.asarray(mean)) * lax.rsqrt(jnp.asarray(variance) + epsilon)

# TODO(slebedev): Change the type of `x` to `ArrayLike`.
@partial(jax.jit, static_argnames=(""num_classes"", ""dtype"", ""axis""))
def _one_hot(x: Array, num_classes: int, *,
             dtype: Any, axis: int | AxisName) -> Array:
  num_classes = core.concrete_dim_or_error(
      num_classes,
      ""The error arose in jax.nn.one_hot argument `num_classes`."")
  dtype = dtypes.canonicalize_dtype(dtype)
  try:
    output_pos_axis = util.canonicalize_axis(axis, x.ndim + 1)  # type: ignore[arg-type]
  except TypeError:
    axis_size = lax.axis_size(axis)
    if num_classes != axis_size:
      raise ValueError(f""Expected num_classes to match the size of axis {axis}, ""
                       f""but {num_classes} != {axis_size}"") from None
    axis_idx = lax.axis_index(axis)
    return jnp.asarray(_dot_product_attention_xla == axis_idx, dtype=dtype)
  axis = operator.index(axis)  # type: ignore[arg-type]
  lhs = lax.expand_dims(x, (axis,))
  rhs_shape = [1] * x.ndim
  rhs_shape.insert(output_pos_axis, num_classes)
  # TODO(yashkatariya): Maybe expose `out_sharding` on `one_hot` too?
  rhs_sharding = NamedSharding(x.aval.sharding.mesh, P(*[None] * len(rhs_shape)))  # pytype: disable=attribute-error
  rhs = lax.broadcasted_iota(x.dtype, rhs_shape, output_pos_axis,
                             out_sharding=rhs_sharding)
  return (lhs == rhs).astype(dtype)

# TODO(slebedev): Change the type of `x` to `ArrayLike`.
def one_hot(x: Any, num_classes: int, *,
            dtype: Any = jnp.float_, axis: int | AxisName = -1) -> Array:
  """"""One-hot encodes the given indices.

  Each index in the input ``x`` is encoded as a vector of zeros of length
  ``num_classes`` with the element at ``index`` set to one::

    >>> jax.nn.one_hot(jnp.array([0, 1, 2]), 3)
    Array([[1., 0., 0.],
           [0., 1., 0.],
           [0., 0., 1.]], dtype=float32)

  Indices outside the range [0, num_classes) will be encoded as zeros::

    >>> jax.nn.one_hot(jnp.array([-1, 3]), 3)
    Array([[0., 0., 0.],
           [0., 0., 0.]], dtype=float32)

  Args:
    x: A tensor of indices.
    num_classes: Number of classes in the one-hot dimension.
    dtype: optional, a float dtype for the returned values (default :obj:`jnp.float_`).
    axis: the axis or axes along which the function should be
      computed.
  """"""
  num_classes = core.concrete_dim_or_error(
      num_classes,
      ""The error arose in jax.nn.one_hot argument `num_classes`."")
  x_arr = jnp.asarray(x)
  if not jnp.isdtype(x_arr.dtype, ""integral""):
    # Deprecated 2024-12-18
    deprecations.warn(
      'jax-nn-one-hot-float-input',
      f""jax.nn.one_hot input should be integer-typed; got dtype={x_arr.dtype}"",
      stacklevel=1)
  return _one_hot(x_arr, num_classes, dtype=dtype, axis=axis)


@jax.custom_jvp
@jax.jit
def relu6(x: ArrayLike) -> Array:
  r""""""Rectified Linear Unit 6 activation function.

  Computes the element-wise function

  .. math::
    \mathrm{relu6}(x) = \min(\max(x, 0), 6)

  except under differentiation, we take:

  .. math::
    
abla \mathrm{relu}(0) = 0

  and

  .. math::
    
abla \mathrm{relu}(6) = 0

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`relu`
  """"""
  return jnp.minimum(jnp.maximum(x, 0), 6.)
relu6.defjvps(lambda g, ans, x:
              lax.select((x > 0) & (x < 6), g, lax.full_like(g, 0)))

@jax.jit
def hard_sigmoid(x: ArrayLike) -> Array:
  r""""""Hard Sigmoid activation function.

  Computes the element-wise function

  .. math::
    \mathrm{hard\_sigmoid}(x) = \frac{\mathrm{relu6}(x + 3)}{6}

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`relu6`
  """"""
  return relu6(x + 3.) / 6.

@jax.jit
def hard_silu(x: ArrayLike) -> Array:
  r""""""Hard SiLU (swish) activation function

  Computes the element-wise function

  .. math::
    \mathrm{hard\_silu}(x) = x \cdot \mathrm{hard\_sigmoid}(x)

  Both :func:`hard_silu` and :func:`hard_swish` are aliases for the same
  function.

  Args:
    x : input array

  Returns:
    An array.

  See also:
    :func:`hard_sigmoid`
  """"""
  numpy_util.check_arraylike(""hard_silu"", x)
  x_arr = jnp.asarray(x)
  return x_arr * hard_sigmoid(x_arr)

hard_swish = hard_silu

def _get_large_negative(dtype):
  dtype_max = jnp.finfo(dtype).max
  return jnp.asarray(-0.7 * dtype_max, dtype=dtype)

def _get_causal_mask(T, S):
  mask = jnp.tril(jnp.ones((T, S), dtype=jnp.bool_))
  return mask[None, None, :, :]

def _get_window_mask(T: int, S: int, local_window_size: tuple[int, int]):
  query_pos = jnp.array(range(T))
  key_pos = jnp.array(range(S))
  left_window, right_window = local_window_size
  left_mask = query_pos[..., None] <= key_pos[..., None, :] + left_window
  right_mask = query_pos[..., None] >= key_pos[..., None, :] - right_window
  return jnp.logical_and(right_mask, left_mask)[None, None, :, :]

def _get_padding_mask_logits(T, S, q_seqlen, kv_seqlen):
  q_mask = True
  kv_mask = True
  if q_seqlen is not None:
    q_indices = jnp.arange(0, T)[None, :, None]
    q_mask = q_indices < q_seqlen[:, None, None]
  if kv_seqlen is not None:
    kv_indices = jnp.arange(0, S)[None, None, :]
    kv_mask = kv_indices < kv_seqlen[:, None, None]
  mask = jnp.logical_and(q_mask, kv_mask)
  return mask[:, None, :, :]

def _get_padding_mask_encoded(T, q_seqlen):
  q_indices = jnp.arange(0, T)[None, :]
  mask = q_indices < q_seqlen[:, None]
  return mask[:, :, None, None]

def _apply_masks(logits, mask, is_causal, q_seqlen, kv_seqlen,
                 local_window_size):
  if mask is None and not is_causal and q_seqlen is None and kv_seqlen is None:
    return logits

  combined_mask = jnp.ones_like(logits, dtype=jnp.bool_)
  if mask is not None:
    assert mask.dtype == jnp.bool_
    combined_mask = jnp.logical_and(combined_mask, mask)

  T, S = logits.shape[2], logits.shape[3]

  if is_causal:
    mask = _get_causal_mask(T, S)
    combined_mask = jnp.logical_and(combined_mask, mask)

  if local_window_size is not None:
    mask = _get_window_mask(T, S, local_window_size)
    combined_mask = jnp.logical_and(combined_mask, mask)

  if q_seqlen is not None or kv_seqlen is not None:
    mask = _get_padding_mask_logits(T, S, q_seqlen, kv_seqlen)
    combined_mask = jnp.logical_and(combined_mask, mask)

  large_negative_number = _get_large_negative(logits.dtype)
  padded_logits = jnp.where(combined_mask, logits, large_negative_number)
  return padded_logits

def _dot_product_attention_core(query, key, value, bias, mask, is_causal,
                                scale, q_seqlen, kv_seqlen, local_window_size):
  logits_dtype = jnp.promote_types(query.dtype, jnp.float32)

  # If the query and logits dtypes are different, then the default precision
  # can use inconsistent types in the backwards pass
  # (see https://github.com/jax-ml/jax/issues/24047).
  if query.dtype == jnp.bfloat16:
    precision = jax.lax.DotAlgorithmPreset.BF16_BF16_F32
  elif query.dtype == jnp.float16:
    precision = jax.lax.DotAlgorithmPreset.F16_F16_F32
  # TODO(sbodenstein): Implement this fix for all dtypes.
  else:
    precision = None

  # Explicit precision will fail on platforms that don't support it. For example,
  # some GPUs do not support BF16_BF16_F32, and TPU does not support F16_F16_F32.
  # Use the default precision as a fallback in these cases.
  try:
    logits = jnp.einsum(
        ""BTNH,BSNH->BNTS"",
        query,
        key,
        precision=precision,
        preferred_element_type=logits_dtype,
    )
  except:  # pylint: disable=bare-except
    logits = jnp.einsum(
        ""BTNH,BSNH->BNTS"",
        query,
        key,
        precision=None,
        preferred_element_type=logits_dtype,
    )

  logits *= jnp.array(scale, dtype=logits.dtype)

  if bias is not None:
    logits = (logits + bias).astype(logits.dtype)

  padded_logits = _apply_masks(logits, mask, is_causal, q_seqlen, kv_seqlen,
                               local_window_size)

  # Softmax and it is always carried out in fp32.
  padded_logits = padded_logits.astype(jnp.float32)
  probs = jax.nn.softmax(padded_logits, axis=-1).astype(key.dtype)

  encoded = jnp.einsum('BNTS,BSNH->BTNH', probs, value)
  if q_seqlen is not None and kv_seqlen is not None:
    mask = _get_padding_mask_encoded(encoded.shape[1], q_seqlen)
    encoded *= mask.astype(encoded.dtype)
  return encoded

def _dot_product_attention_xla(
    query: Array,
    key: Array,
    value: Array,
    bias: Array | None,
    mask: Array | None,
    is_causal: bool,
    scale: float,
    q_seqlen: Array | None,
    kv_seqlen: Array | None,
    local_window_size: tuple[int, int] | None):

  B, T, N, H = query.shape
  _, S, K, _ = key.shape
  G = N // K

  query = jnp.reshape(query, (B, T, K, G, H))
  def _reshape_to_grouped(t):
    if t is not None:
      tB, tN, tT, tS = t.shape
      if tN == 1:
        t = jnp.broadcast_to(t[:, :, None, :, :], (tB, tN, G, tT, tS))
      else:
        assert tN == N
        t = jnp.reshape(t, (tB, K, G, tT, tS))
    return t
  bias = _reshape_to_grouped(bias)
  mask = _reshape_to_grouped(mask)
  vmapped_fn = jax.vmap(
      _dot_product_attention_core,
      in_axes=(3, None, None, 2, 2, None, None, None, None, None),
      out_axes=3,
  )
  encoded = vmapped_fn(query, key, value, bias, mask, is_causal, scale,
                       q_seqlen, kv_seqlen, local_window_size)
  encoded = jnp.reshape(encoded, (B, T, N, H))
  return encoded

def bias_fwd_rule(a, query_head_num):
  return bias_fwd_p.bind(a, query_head_num), a
def bias_bwd_rule(query_head_num, res, g):
  a = res
  if a.shape[0] > 1 or a.shape[-3] != query_head_num:
    raise ValueError(""cuDNN only supports bias gradient when the batch size is ""
                     f""1 and the head number matches the query, but got ""
                     f""B={a.shape[0]}, N={a.shape[-3]}."")
  return (bias_bwd_p.bind(g, a, query_head_num),)

# This function uses two custom primitives, `bias_fwd` and `bias_bwd`, to work
# around a cuDNN issue where bias gradients are only supported when the batch
# size is 1 and the number of heads matches the query.
# TODO(kaixih@nvidia): Remove this workaround once cuDNN resolves the issue.
@partial(jax.custom_vjp, nondiff_argnums=(1,))
def check_valid_bias_batch(x, query_head_num):
  output, _ = bias_fwd_rule(x, query_head_num)
  return output
check_valid_bias_batch.defvjp(bias_fwd_rule, bias_bwd_rule)

bias_fwd_p = core.Primitive('bias_fwd')
bias_fwd_p.multiple_results = False
bias_bwd_p = core.Primitive('bias_bwd')
bias_bwd_p.multiple_results = False

def bias_fwd_impl(a, query_head_num):
  return a
def bias_bwd_impl(g, a, query_head_num):
  return g
bias_fwd_p.def_impl(bias_fwd_impl)
bias_bwd_p.def_impl(bias_bwd_impl)

def bias_fwd_abstract_eval(a, query_head_num):
  return core.ShapedArray(a.shape, a.dtype)
def bias_bwd_abstract_eval(g, a, query_head_num):
  return core.ShapedArray(g.shape, g.dtype)
bias_fwd_p.def_abstract_eval(bias_fwd_abstract_eval)
bias_bwd_p.def_abstract_eval(bias_bwd_abstract_eval)

def bias_fwd_lowering(ctx, a, query_head_num):
  return [a]
def bias_bwd_lowering(ctx, g, a, query_head_num):
  return [g]
mlir.register_lowering(bias_fwd_p, bias_fwd_lowering)
mlir.register_lowering(bias_bwd_p, bias_bwd_lowering)

def bias_fwd_batch_rule(batched_args, batch_dims):
  x, query_head_num = batched_args
  a = batch_dims[0]
  output, _ = bias_fwd_rule(x, query_head_num)
  return output, a
def bias_bwd_batch_rule(batched_args, batch_dims):
  g, x, query_head_num = batched_args
  b = batch_dims[0]
  *Bs, _, _, _ = x.shape
  B = math.prod(Bs)
  x = jnp.reshape(x, (B,) + x.shape[-3:])
  output, = bias_bwd_rule(query_head_num, x, g)
  return output, b
batching.primitive_batchers[bias_fwd_p] = bias_fwd_batch_rule
batching.primitive_batchers[bias_bwd_p] = bias_bwd_batch_rule

def dot_product_attention(
    query: ArrayLike,
    key: ArrayLike,
    value: ArrayLike,
    bias: ArrayLike | None = None,
    mask: ArrayLike | None = None,
    *,
    scale: float | None = None,
    is_causal: bool = False,
    query_seq_lengths: ArrayLike | None = None,
    key_value_seq_lengths: ArrayLike | None = None,
    local_window_size: int | tuple[int, int] | None = None,
    implementation: Literal['xla', 'cudnn'] | None = None) -> Array:
  r""""""Scaled dot product attention function.

  Computes the attention function on Query, Key, and Value tensors:

  .. math::

    \mathrm{Attention}(Q, K, V)=\mathrm{softmax}(\frac{QK^T}{\sqrt{d_k}})V

  If we define :code:`logits` as the output of :math:`QK^T` and the
  :code:`probs` as the output of :math:`softmax`.

  Throughout this function, we utilize the following uppercase letters to
  represent the shape of array::

    B = batch size
    S = length of the key/value (source)
    T = length of the query (target)
    N = number of attention heads
    H = dimensions of each attention head
    K = number of key/value heads
    G = number of groups, which equals to N // K

  Args:
    query: query array; shape :code:`(BTNH|TNH)`
    key: key array: shape :code:`(BSKH|SKH)`. When `K` equals `N`, multi-headed
      attention (MHA https://arxiv.org/abs/1706.03762) is performed. Otherwise,
      grouped query attention (GQA https://arxiv.org/abs/2305.13245) is
      performed if `N` is a multiple of `K`, and multi-query attention (MQA
      https://arxiv.org/abs/1911.02150) is performed if `K == 1` (a special case
      of GQA).
    value: value array, should have the same shape as the `key` array.
    bias: optional, bias array to be added to logits; The shape must be 4D and
      be broadcastable to :code:`(BNTS|NTS)`.
    mask: optional, mask array used to filter out logits. It is a boolean mask
      where `True` indicates the element should take part in attention. For an
      additive mask, users should pass it to `bias`. The shape must be 4D and be
      broadcastable to :code:`(BNTS|NTS)`.
    scale: scale for the logits. If None, the scale will be set to 1 divided by
      the square root of query's head dimension (i.e. H).
    is_causal: If true, causal attention will be applied. Note, some
      implementations like `xla` will generate a mask tensor and apply it to the
      logits to mask out the non-causal parts of the attention matrix, but other
      implementations like `cudnn` will avoid computing the non-causal regions,
      providing speedups.
    query_seq_lengths: `int32` array of sequence lengths for query; shape
      :code:`(B)`
    key_value_seq_lengths: `int32` array of sequence lengths for key and value;
      shape :code:`(B)`
    local_window_size: Window sizes to make self attention to attend to each
      token's local window. If set, this specifies the (left_window_size,
      right_window_size) for each token. E.g., if local_window_size == (3, 2)
      and the sequence is [0, 1, 2, 3, 4, 5, c, 7, 8, 9], token `c` can attend
      to [3, 4, 5, c, 7, 8]. If a single int is given, it will be interpreted as
      a symmetric window (window_size, window_size).
    implementation: A string to control which implementation backend to use.
      Supported strings are `xla`, `cudnn` (cuDNN flash attention). It defaults
      to `None`, which will automatically select the best available backend.
      Note, `cudnn` supports only a subset of shapes/dtypes, and an exception
      will be thrown if its not supported.

  Returns:
    An array of the attention output with the same shape as :code:`query`.
  """"""
  output_shape = jnp.asarray(query).shape
  def _ensure_4d(t):
    t = jnp.asarray(t)
    dims_to_add = 4 - t.ndim
    if dims_to_add > 0:
      return jnp.expand_dims(t, axis=tuple(range(dims_to_add)))
    return t

  query_arr = _ensure_4d(query)
  key_arr = _ensure_4d(key)
  value_arr = _ensure_4d(value)
  bias = _ensure_4d(bias) if bias is not None else None
  mask = _ensure_4d(mask) if mask is not None else None
  if query_seq_lengths is not None:
    query_seq_lengths = jnp.asarray(query_seq_lengths)
  if key_value_seq_lengths is not None:
    key_value_seq_lengths = jnp.asarray(key_value_seq_lengths)
  if isinstance(local_window_size, int):
    local_window_size = (local_window_size, local_window_size)

  def _check_shape_and_dtype(t: Array | None, shape: Sequence[int],
                             dtype: DType | None, name: str) -> None:
    if t is None:
      return
    if t.ndim != len(shape):
      raise ValueError(f""{name} ndim should be {len(shape)}, but got {t.ndim}"")
    if dtype is not None and t.dtype != dtype:
      raise ValueError(f""{name} dtype should be {dtype}, but got {t.dtype}"")
    for i in range(t.ndim):
      if shape[i] != -1 and t.shape[i] != shape[i]:
        raise ValueError(f""{name} shape should be {shape}: but got {t.shape}"")

  B, S, K, H = key_arr.shape
  _check_shape_and_dtype(value_arr, [B, S, K, H], key_arr.dtype, 'value')
  _check_shape_and_dtype(query_arr, [B, -1, -1, H], key_arr.dtype, 'query')
  _check_shape_and_dtype(mask, [-1] * 4, jnp.bool_, 'mask')
  _check_shape_and_dtype(bias, [-1] * 4, None, 'bias')
  _check_shape_and_dtype(query_seq_lengths, [B], jnp.int32,
                         'query_seq_lengths')
  _check_shape_and_dtype(key_value_seq_lengths, [B], jnp.int32,
                         'key_value_seq_lengths')
  if query_arr.shape[-2] % K != 0:
    raise ValueError(f""The number of query heads must be a multiple of ""
                     f""key/value heads, but got {query_arr.shape[-2]} vs {K}"")

  scale_val = (1.0 / np.sqrt(H)) if scale is None else scale

  match implementation:
    case 'xla':
      out = _dot_product_attention_xla(
          query_arr, key_arr, value_arr, bias, mask, is_causal=is_causal,
          scale=scale_val, q_seqlen=query_seq_lengths,
          kv_seqlen=key_value_seq_lengths,
          local_window_size=local_window_size,
      )
    case 'cudnn':
      if bias is not None:
        bias = check_valid_bias_batch(bias, query_arr.shape[-2])
        bias = jnp.asarray(bias)
      use_padding = (
           query_seq_lengths is not None or key_value_seq_lengths is not None
      )
      if use_padding:
        if query_seq_lengths is None:
          T = query_arr.shape[1]
          query_seq_lengths = jnp.full((B,), T, dtype=jnp.int32)
        if key_value_seq_lengths is None:
          key_value_seq_lengths = jnp.full((B,), S, dtype=jnp.int32)

      mask_type = MaskType.NO_MASK
      if use_padding and is_causal:
        mask_type = MaskType.PADDING_CAUSAL
      elif is_causal:
        mask_type = MaskType.CAUSAL
      elif use_padding:
        mask_type = MaskType.PADDING
      # CuDNN supports only the left window with an exclusive boundary when
      # causal mask is enabled.
      sliding_window = None
      if local_window_size is not None:
        l_window, r_window = local_window_size
        if r_window == 0 or mask_type == MaskType.CAUSAL:
          sliding_window = l_window + 1
        else:
          raise ValueError(f""cuDNN doesn't support right window: {r_window} ""
                           ""when causal mask is not used."")

      out = cudnn_dot_product_attention(
          query_arr, key_arr, value_arr, bias, mask, query_seq_lengths,
          key_value_seq_lengths, scale=scale_val, mask_type=mask_type,
          sliding_window_length=sliding_window,
      )
    case None:
      # TODO(kaixih@nvidia) Defaults to XLA for now. Will automatically select
      # best backend.
      out = _dot_product_attention_xla(
          query_arr, key_arr, value_arr, bias, mask, is_causal=is_causal,
          scale=scale_val, q_seqlen=query_seq_lengths,
          kv_seqlen=key_value_seq_lengths,
          local_window_size=local_window_size,
      )
    case _:
      raise ValueError(f""Unsupported implementation option: {implementation}"")

  return jnp.reshape(out, output_shape)

def scaled_matmul(
    lhs: Array,
    rhs: Array,
    lhs_scales: Array,
    rhs_scales: Array,
    preferred_element_type: DTypeLike = jnp.float32,
) -> Array:
    r""""""Scaled matrix multiplication function.

    Performs block-scaled matmul of `a` and `b` using `a_scales` and `b_scales`.
    The last dim is the contracting dim, and block size is inferred.

    Mathematically, this operation is equivalent to::

      a_block_size = a.shape[-1] // a_scales.shape[-1]
      b_block_size = b.shape[-1] // b_scales.shape[-1]
      a_scaled = a * jnp.repeat(a_scales, a_block_size, axis=-1)
      b_scaled = b * jnp.repeat(b_scales, b_block_size, axis=-1)
      jnp.einsum('BMK,BNK->BMN', a_scaled, b_scaled)

    Args:
      lhs (Array): Operand a, shape (B, M, K).
      rhs (Array): Operand b, shape (B, N, K).
      lhs_scales (Array): Shape (B, M, K_a), where `K % K_a == 0`.
      rhs_scales (Array): Shape (B, N, K_b), where `K % K_b == 0`.
      preferred_element_type (DTypeLike, optional): Defaults to `jnp.float32`.

    Returns:
      Array of shape (B, M, N).

    Notes:
      - We currently do not support user-defined `precision` for customizing the
        compute data type. It is fixed to `jnp.float32`.
      - Block size is inferred as `K // K_a` for `a` and `K // K_b` for `b`.
      - To use cuDNN with Nvidia Blackwell GPUs, inputs must match::

          # mxfp8
          a, b: jnp.float8_e4m3fn | jnp.float8_e5m2
          a_scales, b_scales: jnp.float8_e8m0fnu
          block_size: 32
          # nvfp4
          a, b: jnp.float4_e2m1fn
          a_scales, b_scales: jnp.float8_e4m3fn
          block_size: 16

    Examples:

      Basic case:

      >>> a = jnp.array([1, 2, 3]).reshape((1, 1, 3))
      >>> b = jnp.array([4, 5, 6]).reshape((1, 1, 3))
      >>> a_scales = jnp.array([0.5]).reshape((1, 1, 1))
      >>> b_scales = jnp.array([0.5]).reshape((1, 1, 1))
      >>> scaled_matmul(a, b, a_scales, b_scales)  # doctest: +SKIP
      Array([[[8.]]], dtype=float32)

      Using fused cuDNN call on Blackwell GPUs:

      >>> dtype = jnp.float8_e4m3fn
      >>> a = jax.random.normal(jax.random.PRNGKey(1), (3, 128, 64), dtype=dtype)
      >>> b = jax.random.normal(jax.random.PRNGKey(2), (3, 128, 64), dtype=dtype)
      >>> a_scales = jnp.ones((3, 128, 4), dtype=jnp.float8_e8m0fnu)
      >>> b_scales = jnp.ones((3, 128, 4), dtype=jnp.float8_e8m0fnu)
      >>> scaled_matmul(a, b, a_scales, b_scales)  # doctest: +SKIP
    """"""
    a, b, a_scales, b_scales = lhs, rhs, lhs_scales, rhs_scales
    if not all(x.ndim == 3 for x in (a, b, a_scales, b_scales)):
        raise ValueError(
            ""scaled_matmul requires all inputs to be 3-dimensional arrays""
        )

    B_a, M_a, K_a = a.shape
    B_b, N_b, K_b = b.shape
    if K_a != K_b or B_a != B_b:
        raise ValueError(
            ""scaled_matmul requires inputs a and b to have matching batch (B) ""
            f""and contract (K) dimensions, but got shapes {a.shape} and ""
            f""{b.shape}""
        )

    B_as, M_as, K_as = a_scales.shape
    B_bs, N_bs, K_bs = b_scales.shape
    if K_as != K_bs or B_as != B_bs:
        raise ValueError(
            ""scaled_matmul requires scales to have matching batch (B) and ""
            f""contract (K) dimensions, but got shapes {a_scales.shape} and ""
            f""{b_scales.shape}""
        )

    if M_as != M_a or N_bs != N_b:
        raise ValueError(
            ""scaled_matmul requires scales to match non-contract dimensions of ""
            f""inputs, but got shapes a: {a.shape}, b: {b.shape}, a_scales: ""
            f""{a_scales.shape}, b_scales: {b_scales.shape}""
        )

    preferred_element_type = dtypes.canonicalize_dtype(
        np.dtype(preferred_element_type)
    )
    out = cudnn_scaled_matmul(
        a,
        b,
        a_scales,
        b_scales,
        preferred_element_type=preferred_element_type,
    )
    return out

def get_scaled_dot_general_config(mode: Literal['nvfp4', 'mxfp8'],
                                  global_scale: Array | None = None):
    r""""""Get quantization configs for scaled_dot_general.

    Create quantization configs for the `jax.nn.scaled_dot_general`.

    See Also:
      - :func:`jax.nn.scaled_dot_general`: Scaled dot general function.
    """"""

    if mode == 'nvfp4':
        one = jnp.ones((1,), dtype=jnp.float32)
        return BlockScaleConfig(
            mode='nvfp4',
            block_size=16,
            data_type=jnp.float4_e2m1fn,
            scale_type=jnp.float8_e4m3fn,
            global_scale=one if global_scale is None else global_scale,
            infer_only=False
        )
    elif mode == 'mxfp8':
        return BlockScaleConfig(
            mode='mxfp8',
            block_size=32,
            data_type=jnp.float8_e4m3fn,
            scale_type=jnp.float8_e8m0fnu,
            global_scale=None,
            infer_only=False
        )
    else:
        raise ValueError(f""Unsupported mode: {mode}"")

def scaled_dot_general(
    lhs, rhs,
    dimension_numbers,
    preferred_element_type=jnp.float32,
    configs: List[BlockScaleConfig] | None = None,
    implementation: Literal['cudnn'] | None = None,
  ):
  r""""""Scaled dot general operation.

  Performs a generalized dot product with block-scaled quantization on the
  lhs and rhs inputs. This operation extends `lax.dot_general` to support
  user-defined scaling configurations.

  Essentially, the operation follows::

      a, a_scales = quantize(lhs, configs[0])
      b, b_scales = quantize(rhs, configs[1])
      c = jax.nn.scaled_matmul(a, b, a_scales, b_scales)

  Args:
    lhs (ArrayLike): Input array.
    rhs (ArrayLike): Input array.
    dimension_numbers (DotDimensionNumbers): A tuple of two tuples specifying
      the contraction and batch dimensions:
      `((lhs_contracting_dims, rhs_contracting_dims), (lhs_batch_dims, rhs_batch_dims))`.
    preferred_element_type (DTypeLike, optional): Output data type of the dot
      product. Defaults to `jnp.float32`. Other valid types include
      `jnp.bfloat16` and `jnp.float16`.
    configs (list of BlockScaleConfig, optional): Scaling configurations for
      lhs, rhs, and gradients. Users can obtain valid configurations via
      `jax.nn.get_scaled_dot_general_config`. Currently, `nvfp4` and `mxfp8`
      are supported. If `None`, falls back to `lax.dot_general`.
    implementation: str
      (Deprecated) Backend selector, now ignored. The system chooses the backend
      automatically. Scheduled for removal in future releases.

  Returns:
    Array: The resulting tensor, with batch dimensions first, followed by
    non-contracting/non-batch dimensions of lhs, and then those of rhs.

  See Also:
    - :func:`jax.nn.scaled_matmul`: Scaled matmul function.
    - :func:`jax.lax.dot_general`: General dot product operator.

  Notes:
    - Unlike `nn.scaled_matmul`, which assumes quantized low-precision
      inputs with explicit scaling factors, this operator takes high-precision
      inputs, applies quantization internally, and handles the backward pass.

  Examples:

    Creating config for mxfp8:

    >>> configs = [jax.nn.get_scaled_dot_general_config('mxfp8')] * 3

    Creating config for nvfp4:

    >>> global_scale = jnp.array([0.5], jnp.float32)
    >>> configs = [jax.nn.get_scaled_dot_general_config('nvfp4', global_scale)] * 3

    Using scaled_dot_general with the configs:

    >>> import functools
    >>> scaled_dot_general_fn = functools.partial(jax.nn.scaled_dot_general, configs=configs)
    >>> lhs = jax.random.normal(jax.random.PRNGKey(1), (3, 128, 64))
    >>> rhs = jax.random.normal(jax.random.PRNGKey(2), (3, 128, 64))
    >>> out = scaled_dot_general_fn(lhs, rhs, (((2,), (2,)), ((0,), (0,))))  # doctest: +SKIP
  """"""
  if implementation is not None:
    warnings.warn(""Backend selector, now ignored. The system chooses the ""
                  ""backend automatically."", DeprecationWarning)

  if configs is None:
    return lax.dot_general(lhs, rhs, dimension_numbers,
                           preferred_element_type=preferred_element_type)

  out = cudnn_scaled_dot_general(
      lhs, rhs, dimension_numbers,
      preferred_element_type=preferred_element_type,
      configs=configs
  )

  return out
",python,tab
4265,13698641,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",37837,0,"",python,selection_command
4266,13701216,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"# Copyright 2024 The JAX Authors.
#
# Licensed under the Apache License, Version 2.0 (the ""License"");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an ""AS IS"" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import enum
import functools
import json
import math
from typing import TypedDict

import jax
from jax import dtypes
from jax._src import core
from jax._src import dispatch
from jax._src.custom_partitioning import custom_partitioning
from jax._src.interpreters import batching
from jax._src.interpreters import mlir
from jax._src.lib import cuda_versions
from jax._src import xla_bridge
from jax._src.lib.mlir import ir
from jax._src.lib.mlir.dialects import hlo
import jax.numpy as jnp
from jax.sharding import NamedSharding, PartitionSpec

Array = jnp.ndarray

class FP8Params(TypedDict):
  amax_dQ: float  # Amax of gradient of query
  amax_dK: float  # Amax of gradient of key
  amax_dV: float  # Amax of gradient of value
  amax_dP: float  # Amax of gradient of state
  descale_q: float  # Descaling factor of query
  descale_k: float  # Descaling factor of key
  descale_v: float  # Descaling factor of value
  descale_s: float  # Descaling factor of attention score
  scale_s: float  # Scale factor for S tensor
  scale_o: float  # Scale factor for output
  descale_o: float  # Descale factor for output (bwd)
  descale_dO: float  # Descale factor for output gradient (bwd)
  descale_dP: float  # Descale factor for P gradient tensor (bwd)
  scale_dQ: float  # Scale factor for query gradient (bwd)
  scale_dK: float  # Scale factor for key gradient (bwd)
  scale_dV: float  # Scale factor for value gradient (bwd)
  scale_dP: float  # Scale factor for state gradient (bwd)


class AttentionLayout(enum.Enum):
  BTNH = 0
  BNTH = 1


class MaskType(enum.Enum):
  NO_MASK = 0
  PADDING = 1
  CAUSAL = 2
  PADDING_CAUSAL = 3
  ALIBI = 4


def convert_mask_type_to_string(mask_type: MaskType) -> str:
  if mask_type == MaskType.NO_MASK:
    return ""NO_MASK""
  elif mask_type == MaskType.PADDING:
    return ""PADDING""
  elif mask_type == MaskType.CAUSAL:
    return ""CAUSAL""
  elif mask_type == MaskType.PADDING_CAUSAL:
    return ""PADDING_CAUSAL""
  elif mask_type == MaskType.ALIBI:
    return ""ALIBI""
  else:
    raise ValueError(f""Unexpected mask type: {mask_type}"")

def has_padding(mask_type: MaskType) -> bool:
  return mask_type == MaskType.PADDING or mask_type == MaskType.PADDING_CAUSAL

def should_export_dbias(bias_shape, query_shape, layout) -> bool:
  b_B, b_N, _, _ = bias_shape
  if layout == AttentionLayout.BNTH.value:
    _, q_N, _, _ = query_shape
  else:
    _, _, q_N, _ = query_shape
  return b_B == 1 and b_N == q_N

def get_large_negative_number(dtype):
  # temp WAR as cuDNN has a bug for subtraction between two large negative value
  if dtype == jnp.bfloat16:
    return jnp.asarray(-2 << 40, dtype=dtype)
  elif dtype == jnp.float16:
    return jnp.asarray(-2 << 14, dtype=dtype)
  else:
    raise ValueError(""Unsupported dtype for inputs."")

def _normalize_layout(layout: str) -> AttentionLayout:
  layout_upper = layout.upper()
  if layout_upper in [""BSNH"", ""BNSH"", ""BTNH"", ""BNTH""]:
    return AttentionLayout[layout_upper.replace(""S"", ""T"")]
  else:
    raise ValueError(f""Unsupported qkv_layout: {layout}"")

def element_type_to_backend_config_type_mapping(dtype):
  _element_type_to_backend_config_type_mapping = {
    ir.BF16Type.get(): ""BF16"",
    ir.F16Type.get(): ""F16"",
  }
  return _element_type_to_backend_config_type_mapping[dtype]

def default_layouts(*shapes):
  return [range(len(shape) - 1, -1, -1) for shape in shapes]

def get_max_seg_per_batch(q_offsets):
  return q_offsets.shape[1] - 1 if len(q_offsets.shape) == 2 else 1

def check_is_paged_attention(page_table_k):
  return len(page_table_k.shape) == 4

def create_dot_product_attention_backend_config_base(
    batch, num_heads, seq_q, seq_kv, dtype, fmha_scale, mask_type, layout, is_bwd
):
  # Q, K, V: query, key, value in shape of BT(S)NH or BNT(S)H
  # P: BMM1 output in shape of BNTS
  # O: BMM2 output in the same shape with Q
  # BMM1: Q @ K -> P
  # BMM2: P @ V -> O
  # BMM1Grad1: dP @ Q -> dK
  # BMM1Grad2: dP @ K -> dQ
  # BMM2Grad1: P @ dO -> dV
  # BMM2Grad2: dO @ V -> dP
  cudnn_fmha_backend_config = {
    ""algorithm"": {
      ""algo_id"": ""0"",
      ""math_type"": ""TENSOR_OP_MATH"",
      ""tuning_knobs"": {""17"": ""1"", ""24"": ""0""},
      ""is_cudnn_frontend"": True,
      ""workspace_size"": ""0"",
    },
    ""fmha_scale"": fmha_scale,
    ""intermediate_tensor_shape"": {
      ""element_type"": element_type_to_backend_config_type_mapping(dtype),
      ""dimensions"": [str(batch), str(num_heads), str(seq_q), str(seq_kv)],
      ""tuple_shapes"": [],
      ""layout"": {
        ""dim_level_types"": [],
        ""dim_unique"": [],
        ""dim_ordered"": [],
        ""minor_to_major"": [""3"", ""2"", ""1"", ""0""],
        ""tiles"": [],
        ""element_size_in_bits"": ""0"",
        ""memory_space"": ""0"",
        ""index_primitive_type"": ""PRIMITIVE_TYPE_INVALID"",
        ""pointer_primitive_type"": ""PRIMITIVE_TYPE_INVALID"",
        ""dynamic_shape_metadata_prefix_bytes"": ""0"",
      },
      ""is_dynamic_dimension"": [False, False, False, False],
    },
    ""is_flash_attention"": True,
    ""mask_type"": convert_mask_type_to_string(mask_type),
  }

  # We define the contracting and batch dims in the format of
  # ((lhs_contracting_dims, rhs_contracting_dims), (lhs_batch_dims,
  # rhs_batch_dims)).
  if layout == AttentionLayout.BNTH.value:
    dims = [
        ((3, 3), ((0, 1), (0, 1))), # BMM1: BNTH,BNSH->BNTS
        ((3, 2), ((0, 1), (0, 1))), # BMM2: BNTS,BNSH->BNTH
        ((2, 2), ((0, 1), (0, 1))), # BMM1_grad_1: BNTS,BNTH->BNSH
        ((3, 2), ((0, 1), (0, 1))), # BMM1_grad_2: BNTS,BNSH->BNTH
        ((2, 2), ((0, 1), (0, 1))), # BMM2_grad_1: BNTS,BNTH->BNSH
        ((3, 3), ((0, 1), (0, 1))), # BMM2_grad_2: BNTH,BNSH->BNTS
    ]
  else:
    dims = [
        ((3, 3), ((0, 2), (0, 2))), # BMM1: BTNH,BSNH->BNTS
        ((3, 1), ((0, 1), (0, 2))), # BMM2: BNTS,BSNH->BTNH
        ((2, 1), ((0, 1), (0, 2))), # BMM1_grad_1: BNTS,BTNH->BSNH
        ((3, 1), ((0, 1), (0, 2))), # BMM1_grad_2: BNTS,BSNH->BTNH
        ((2, 1), ((0, 1), (0, 2))), # BMM2_grad_1: BNTS,BTNH->BSNH
        ((3, 3), ((0, 2), (0, 2))), # BMM2_grad_2: BTNH,BSNH->BNTS
    ]
  keys = [
      ""bmm1_dot_dimension_numbers"",
      ""bmm2_dot_dimension_numbers"",
      ""bmm1_grad_gemm1_dot_dimension_numbers"",
      ""bmm1_grad_gemm2_dot_dimension_numbers"",
      ""bmm2_grad_gemm1_dot_dimension_numbers"",
      ""bmm2_grad_gemm2_dot_dimension_numbers"",
  ]
  fwd_dot_number = {}
  bwd_dot_number = {}
  for idx, (key, ((lc, rc), (lb, rb))) in enumerate(zip(keys, dims)):
    dims_to_write = fwd_dot_number if idx < 2 else bwd_dot_number
    dims_to_write[key] = {
        ""lhs_contracting_dimensions"": [str(lc)],
        ""rhs_contracting_dimensions"": [str(rc)],
        ""lhs_batch_dimensions"": [str(i) for i in lb],
        ""rhs_batch_dimensions"": [str(i) for i in rb],
    }

  if is_bwd:
    cudnn_fmha_backend_config = {**cudnn_fmha_backend_config, **bwd_dot_number}
  else:
    cudnn_fmha_backend_config = {**cudnn_fmha_backend_config, **fwd_dot_number}
  backend_config = {
    ""operation_queue_id"":""0"",
    ""wait_on_operation_queues"":[],
    ""cudnn_fmha_backend_config"": cudnn_fmha_backend_config
  }
  return backend_config

def create_dot_product_attention_backend_config(
    batch,
    num_heads,
    seq_q,
    seq_kv,
    dtype,
    fmha_scale,
    seed,
    dropout_rate,
    mask_type,
    layout,
    sliding_window_length,
    max_seg_per_batch,
    is_paged_attention,
    is_bwd
):
  backend_config = create_dot_product_attention_backend_config_base(
      batch, num_heads, seq_q, seq_kv, dtype,
      fmha_scale, mask_type, layout, is_bwd
  )
  if sliding_window_length is None:
    sliding_window_length = 0
  backend_config['cudnn_fmha_backend_config'][""dropout_rate""] = dropout_rate
  backend_config['cudnn_fmha_backend_config'][""seed""] = seed
  backend_config['cudnn_fmha_backend_config'][""sliding_window_length""] = sliding_window_length
  backend_config['cudnn_fmha_backend_config'][""max_seg_per_batch""] = max_seg_per_batch
  backend_config['cudnn_fmha_backend_config'][""is_paged_attention""] = is_paged_attention
  return json.dumps(backend_config)

def create_dot_product_attention_fp8_backend_config(
    batch, num_heads, seq_q, seq_kv, dtype, fmha_scale, mask_type, layout, is_bwd):
  backend_config = create_dot_product_attention_backend_config_base(
      batch, num_heads, seq_q, seq_kv, dtype, fmha_scale, mask_type, layout, is_bwd)
  return json.dumps(backend_config)

# mapping from (is_bwd, has_dropout, has_bias) to custom call name
_custom_name_maps = {
  # fMHA forward call targets.
  (False, False, False, False): ""__cudnn$fmhaSoftmax"",
  (False, False, True, False): ""__cudnn$fmhaScaleBiasSoftmax"",
  (False, True, False, False): ""__cudnn$fmhaSoftmaxDropout"",
  (False, True, True, False): ""__cudnn$fmhaScaleBiasSoftmaxDropout"",
  (False, False, False, True): ""__cudnn$fmhaSoftmaxF8"",
  # fMHA backward call targets.
  (True, False, False, False): ""__cudnn$fmhaSoftmaxBackward"",
  (True, False, True, False): ""__cudnn$fmhaScaleBiasSoftmaxBackward"",
  (True, True, False, False): ""__cudnn$fmhaSoftmaxDropoutBackward"",
  (True, True, True, False): ""__cudnn$fmhaScaleBiasSoftmaxDropoutBackward"",
  (True, False, False, True): ""__cudnn$fmhaSoftmaxBackwardF8"",
}

def get_custom_call_name(has_bias, has_dropout, is_bwd, is_fp8=False):
  return _custom_name_maps[(is_bwd, has_dropout, has_bias, is_fp8)]

get_fp8_custom_call_name = functools.partial(
    get_custom_call_name, has_bias=False, has_dropout=False, is_fp8=True
)

def check_layout(query, key, value, bias, q_seqlen, kv_seqlen,
  q_offsets, kv_offsets, page_table_k, page_table_v, layout):
  def check_eq(a, b, c, msg):
    if not (a == b == c):
      raise ValueError(f""{msg} must be same, got {a}, {b}, {b}"")

  q_rank, k_rank, v_rank = len(query.shape), len(key.shape), len(value.shape)
  if q_rank != 4:
    raise ValueError(f""Q must have a rank of 4, got {q_rank}"")
  check_eq(q_rank, k_rank, v_rank, ""QKV rank"")

  q_dtype, k_dtype, v_dtype = query.dtype, key.dtype, value.dtype
  if q_dtype not in [jnp.bfloat16, jnp.float16, jnp.float8_e4m3fn, jnp.float8_e5m2]:
    raise NotImplementedError(f""Q must be fp16/bf16/fp8_e4m3fn/fp8_e5m2, got {q_dtype}"")
  check_eq(q_dtype, k_dtype, v_dtype, ""QKV dtype"")

  if layout == AttentionLayout.BNTH:
    qB, qN, qT, qH = query.shape
    kB, kN, kS, kH = key.shape
    vB, vN, vS, vH = value.shape
  else:
    assert layout == AttentionLayout.BTNH
    qB, qT, qN, qH = query.shape
    kB, kS, kN, kH = key.shape
    vB, vS, vN, vH = value.shape

  if page_table_k is not None and page_table_v is not None:
    k_blocks, k_block_size = kB, kS
    v_blocks, v_block_size = vB, vS
    kB, _, k_blocks_per_batch, _ = page_table_k.shape
    vB, _, v_blocks_per_batch, _ = page_table_v.shape
    kS = k_blocks_per_batch * k_block_size
    vS = v_blocks_per_batch * v_block_size
    if kB * k_blocks_per_batch != k_blocks:
      raise ValueError(
        f""Key and page_table_k must have same number of blocks, ""
        f""got {k_blocks} vs {kB * k_blocks_per_batch}"")
    if vB * v_blocks_per_batch != v_blocks:
      raise ValueError(
        f""Value and page_table_v must have same number of blocks, ""
        f""got {v_blocks} vs {vB * v_blocks_per_batch}"")

  check_eq(qB, kB, vB, ""QKV batch"")
  check_eq(qH, kH, vH, ""QKV dim_per_head"")
  if kN != vN:
    raise ValueError(f""KV must have same number of heads, got {kN} vs {vN}"")
  if kS != vS:
    raise ValueError(f""KV must have same seq length, got {kS} vs {vS}"")

  # check bias
  if bias is not None:
    _, _, bT, bS = bias.shape
    if bT != qT or bS != vS:
      raise ValueError(
        f""Bias must have same seq length as QKV, got {bT} and {bS}"")

  # check q_seqlen/kv_seqlen/q_offsets/kv_offsets
  expected_rank = 2 if q_offsets is not None else 1
  def check_seqlen_offsets(tensor, name):
    if tensor is not None:
      dtype = tensor.dtype
      rank = len(tensor.shape)
      if dtype != jnp.int32:
        raise ValueError(f""{name} must have int32 datatype, got {dtype}"")
      if rank != expected_rank:
        raise ValueError(f""{name} must have a rank of {expected_rank}, got {rank}"")
      b = tensor.shape[0]
      if b != qB:
        raise ValueError(f""{name} must have same batch as Q, got {b}"")

  check_seqlen_offsets(q_seqlen, ""q_seqlen"")
  check_seqlen_offsets(kv_seqlen, ""kv_seqlen"")
  check_seqlen_offsets(q_offsets, ""q_offsets"")
  check_seqlen_offsets(kv_offsets, ""kv_offsets"")


def check_is_flash_attention(
    query, key, layout: int, cudnn_version, has_bias, is_training, is_packed=False,
    is_paged_attention=False, is_fp8=False):
    # Extract sequence length (T) and head dim (H) based on layout
    if layout == AttentionLayout.BNTH.value:
        _, _, T, H = query.shape
        _, _, S, _ = key.shape
    else:
        _, T, _, H = query.shape
        _, S, _, _ = key.shape

    # Flash attention conditions
    if is_fp8:
        # FP8 specific conditions
        if not ((is_training and H == 128 and T % 128 == 0 and S % 128 == 0) or
                (not is_training and H <= 256 and H % 16 == 0)):
            raise NotImplementedError(
                f""Unsupported sequence length Q {T}, KV {S} and head dim {H} for FP8.""
            )
    else:
        # bf16/fp16 attention conditions
        # Check the head dim.
        is_on_hopper = is_cuda_compute_capability_equal(""9.0"")
        H_max = 256 if cudnn_version >= 90500 and is_on_hopper else 128
        if not (H <= H_max and H % 8 == 0):
          raise NotImplementedError(
              f""The head dim must be <= {H_max} and a multiple of 8, ""
              f""but got {H}.""
          )

        # Check patterns with bias, seqlen should be divisible by 2
        if (is_training and has_bias and (T % 2 != 0 or S % 2 != 0)):
          raise NotImplementedError(
              f""Unsupported sequence length Q {T}, KV {S}.""
          )

        if is_packed and (cudnn_version < 90600 or not check_compute_capability(""9.0"")):
          raise NotImplementedError(
            ""Packed layout requires cudnn version >= 9.6 and at least hopper arch."")
        if is_paged_attention and cudnn_version < 90500:
          raise NotImplementedError(""Page attention requires cudnn version >= 9.5."")

def check_cudnn_version():
  # check if cuDNN is installed
  if cuda_versions is None:
    raise RuntimeError(""cuDNN is not detected."")
  return cuda_versions.cudnn_get_version()

def check_compute_capability(capability):
  if not 'cuda' in xla_bridge.get_backend().platform_version:
    return False
  d, *_ = jax.local_devices(backend=""gpu"")
  target = tuple(int(x) for x in capability.split("".""))
  current = tuple(int(x) for x in d.compute_capability.split("".""))
  return current >= target

def is_cuda_compute_capability_equal(capability):
  if not 'cuda' in xla_bridge.get_backend().platform_version:
    return False
  d, *_ = jax.local_devices(backend=""gpu"")
  target = tuple(int(x) for x in capability.split("".""))
  current = tuple(int(x) for x in d.compute_capability.split("".""))
  return current == target

def _dot_product_attention_fwd(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v,
    scale, seed, dropout_rate, variadic_args, mask_type, layout,
    sliding_window_length, cudnn_version, return_residual):
  # check if flash attention is supported for this attention pattern
  check_is_flash_attention(
      query, key, layout, cudnn_version, bias is not None, False,
      get_max_seg_per_batch(q_offsets) > 1, check_is_paged_attention(page_table_k))
  outputs = _dot_product_attention_fwd_p_wrapper.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, scale=scale, seed=seed, dropout_rate=dropout_rate,
      variadic_args=variadic_args, mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length, is_training=False or return_residual)
  if return_residual:
    return tuple(outputs)
  else:
    return outputs[0]

def _dot_product_attention_fwd_rule(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, scale, seed, dropout_rate, variadic_args,
    mask_type, layout, sliding_window_length, cudnn_version,
    return_residual):
  # check if flash attention is supported for this attention pattern
  check_is_flash_attention(
      query, key, layout, cudnn_version, bias is not None, True,
      get_max_seg_per_batch(q_offsets) > 1)
  outputs = _dot_product_attention_fwd_p_wrapper.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, scale=scale, seed=seed, dropout_rate=dropout_rate,
      variadic_args=variadic_args, mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length, is_training=True)
  res = (query, key, value, bias, q_seqlen, kv_seqlen, q_offsets,
         kv_offsets, page_table_k, page_table_v, outputs[1], outputs[0])
  if return_residual:
    return tuple(outputs), res
  else:
    return outputs[0], res

def _dot_product_attention_bwd_rule(
    scale, seed, dropout_rate, variadic_args, mask_type, layout,
    sliding_window_length, is_training, return_residual, res, grad_output):
  (query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
   page_table_k, page_table_v, activation, fwd_output) = res
  if return_residual:
    grad_output = grad_output[0]
  grads = _dot_product_attention_bwd_p_wrapper.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, activation, fwd_output, grad_output,
      scale=scale, seed=seed, dropout_rate=dropout_rate, variadic_args=variadic_args,
      mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length
  )
  grads = (*grads,) + (None,) * (10 - len(grads))
  return grads

def _fix_seqlen_offsets(q_seqlen, kv_seqlen, q_offsets, kv_offsets, query, key):
  # fix seqlen and offsets to what cuDNN expects in sequence packing.
  # cuDNN expects seqlen to have shape [S] where S is the total number of segments
  # while the SDPA API accetps seqlen with shape [B, M] where B is the batch and M
  # is the maximum number of segments of one batch. B x M is larger than S and seqlen
  # is filled with -1 for padded regions. Therefore, we need to shift all non negative
  # values to left side to form a correct seqlen. Similar layout is required for
  # offsets tensors.
  # cuDNN expects offsets to have offset for each segment starting from first segment
  # while SDPA API accetps offsets to have offset for each segment starting from
  # current batch, therefore we need to calculate accumulative offset of each segment
  # starting from first segment.
  def _shift_to_left(x, fill_value):
    # shift any non-negative value to left
    # [[1, 3, -1, -1], [2, 3, 4, -1]]
    # -> [[1, 3, 2, 3], [4, -1, -1, -1]]
    x_shape = x.shape
    x = x.flatten()
    size = x.size
    indices = jnp.nonzero(x >= 0, size=size, fill_value=size)[0]
    y = jnp.take(x, indices, fill_value=fill_value)
    return jnp.reshape(y, x_shape)

  def _cu_offset(offsets, max_seq):
    # calculate accumulative offset by batch
    # [[1, 3, 5, 7], [4, 5, -1, -1]], max_seq = 8
    # -> [[1, 3, 5, 7], [12, 13, -1, -1]]
    batch = offsets.shape[0]
    offsets = jnp.where(
        offsets >= 0,
        offsets + (jnp.arange(batch, dtype=offsets.dtype) * max_seq)[..., jnp.newaxis],
        offsets,
    )
    return offsets

  if get_max_seg_per_batch(q_offsets) > 1:
    B, T, N, H = query.shape
    _, S, _, _ = key.shape

    q_seqlen = _shift_to_left(q_seqlen, -1)
    kv_seqlen = _shift_to_left(kv_seqlen, -1)

    q_offsets = _cu_offset(q_offsets, T)
    kv_offsets = _cu_offset(kv_offsets, S)
    q_offsets = _shift_to_left(q_offsets, -1)
    kv_offsets = _shift_to_left(kv_offsets, -1)

    # mark any invalid entries as maximum offset
    q_offsets = jnp.where(q_offsets < 0, B * T, q_offsets)
    kv_offsets = jnp.where(kv_offsets < 0, B * S, kv_offsets)

    # multiply by stride_per_token to get correct offsets
    # do it here because real stride changes after sharding
    q_offsets = q_offsets * N * H
    kv_offsets = kv_offsets * N * H

  return q_seqlen, kv_seqlen, q_offsets, kv_offsets

def _dot_product_attention_fwd_impl(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, scale, seed, dropout_rate, variadic_args,
    mask_type, layout, sliding_window_length, is_training):
  # args: {Q, K, V, mask*, bias*}
  q_seqlen, kv_seqlen, q_offsets, kv_offsets = \
      _fix_seqlen_offsets(q_seqlen, kv_seqlen, q_offsets, kv_offsets, query, key)
  outputs = _dot_product_attention_fwd_p.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, scale=scale, seed=seed, dropout_rate=dropout_rate,
      variadic_args=variadic_args, mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length, is_training=is_training)
  return outputs

def _dot_product_attention_bwd_impl(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, activation, fwd_output, grad_output, scale,
    seed, dropout_rate, variadic_args, mask_type, layout, sliding_window_length):
  q_seqlen, kv_seqlen, q_offsets, kv_offsets = \
      _fix_seqlen_offsets(q_seqlen, kv_seqlen, q_offsets, kv_offsets, query, key)
  grads = _dot_product_attention_bwd_p.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, activation, fwd_output, grad_output,
      scale=scale, seed=seed,
      dropout_rate=dropout_rate, variadic_args=variadic_args,
      mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length)
  return grads

def _dot_product_attention_fwd_abstract(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, *, scale, seed, dropout_rate, variadic_args,
    mask_type, layout, sliding_window_length, is_training):
  query_dtype = dtypes.canonicalize_dtype(query.dtype)
  if layout == AttentionLayout.BNTH.value:
    B, N, T, _ = query.shape
    _, _, S, _ = key.shape
  else:
    B, T, N, _ = query.shape
    _, S, _, _ = key.shape
  output_shape = query.shape

  max_seg_per_batch = get_max_seg_per_batch(q_offsets)
  softmax_stat_shape = (B * max_seg_per_batch, N, T)

  if is_training:
    return (
      core.ShapedArray(output_shape, query_dtype),  # output
      core.ShapedArray(softmax_stat_shape, jnp.float32),  # softmax_stat
    )
  else:
    return (
      core.ShapedArray(output_shape, query_dtype),  # output
    )

def _dot_product_attention_bwd_abstract(
    query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, activation, fwd_output, grad_output, *,
    scale, seed, dropout_rate, variadic_args, mask_type, layout, sliding_window_length):
  query_dtype = dtypes.canonicalize_dtype(query.dtype)
  key_dtype = dtypes.canonicalize_dtype(key.dtype)
  value_dtype = dtypes.canonicalize_dtype(value.dtype)

  _, has_dbias = variadic_args
  if has_dbias:
    # cuDNN supports bias for this case
    bias_dtype = dtypes.canonicalize_dtype(bias.dtype)
    return (
      core.ShapedArray(
          query.shape, query_dtype
      ),  # grad query
      core.ShapedArray(
          key.shape, key_dtype
      ),  # grad key
      core.ShapedArray(
          value.shape, value_dtype
      ),  # grad value
      core.ShapedArray(
          bias.shape, bias_dtype
      ),  # grad bias
    )
  else:
    return (
      core.ShapedArray(
          query.shape, query_dtype
      ),  # grad query
      core.ShapedArray(
          key.shape, key_dtype
      ),  # grad key
      core.ShapedArray(
          value.shape, value_dtype
      ),  # grad value
    )

def _dot_product_attention_fwd_cuda_lowering(
    ctx, query, key, value, bias, q_seqlen, kv_seqlen, q_offsets,
    kv_offsets, page_table_k, page_table_v, scale, seed, dropout_rate,
    variadic_args, mask_type, layout, sliding_window_length, is_training):
  query_type = ir.RankedTensorType(query.type)
  query_shape = query_type.shape
  key_type = ir.RankedTensorType(key.type)
  key_shape = key_type.shape

  if layout == AttentionLayout.BNTH.value:
    B, N, T, H = query_shape
    _, _, S, _ = key_shape
    output_layout = (3, 2, 1, 0)
    output_transpose_perm = mlir.dense_int_array((0, 1, 2, 3))
  else:
    B, T, N, H = query_shape
    _, S, _, _ = key_shape
    output_layout = (3, 1, 2, 0)
    output_transpose_perm = mlir.dense_int_array((0, 2, 1, 3))

  max_seg_per_batch = get_max_seg_per_batch(ir.RankedTensorType(q_offsets.type))
  is_paged_attention = check_is_paged_attention(ir.RankedTensorType(page_table_k.type))

  output_shape = (B, N, T, H)
  softmax_stat_shape = (B * max_seg_per_batch, N, T)
  workspace_shape = (0,)
  workspace_type = ir.IntegerType.get_unsigned(8)

  has_bias, _ = variadic_args
  backend_config = create_dot_product_attention_backend_config(
      B, N, T, S, query_type.element_type, scale, seed, dropout_rate,
      mask_type, layout, sliding_window_length, max_seg_per_batch,
      is_paged_attention, is_bwd=False)
  # {Q, K, V, bias*, q_seqlen*, kv_seqlen*,  q_offsets*, kv_offsets*}}
  # {output, activation*, workspace}
  has_dropout = dropout_rate > 0
  operands = [query, key, value]
  if has_bias:
    operands.append(bias)
  if has_padding(mask_type) or max_seg_per_batch > 1 or is_paged_attention:
    operands.append(q_seqlen)
    operands.append(kv_seqlen)
  if max_seg_per_batch > 1:
    operands.append(q_offsets)
    operands.append(kv_offsets)
  if is_paged_attention:
    operands.append(page_table_k)
    operands.append(page_table_v)

  custom_call_name = get_custom_call_name(has_bias, has_dropout, False)

  if is_training:
    result_types = [
      ir.RankedTensorType.get(output_shape, query_type.element_type),
      ir.RankedTensorType.get(softmax_stat_shape, ir.F32Type.get()),
      ir.RankedTensorType.get(workspace_shape, workspace_type),
    ]
    result_layouts = [output_layout] + default_layouts(softmax_stat_shape, workspace_shape)
  else:
    result_types = [
      ir.RankedTensorType.get(output_shape, query_type.element_type),
      ir.RankedTensorType.get(workspace_shape, workspace_type)
    ]
    result_layouts = [output_layout] + default_layouts(workspace_shape)
  # create custom call here
  out = mlir.custom_call(
    custom_call_name,
    result_types=result_types,
    operands=operands,
    backend_config=backend_config,
    operand_layouts=default_layouts(
      *[ir.RankedTensorType(operand.type).shape for operand in operands]),
    result_layouts=result_layouts,
  )
  # drop workspace memory
  # output should be (B, T, N, H) instead of (B, N, T, H)
  if is_training:
    return [hlo.transpose(out.results[0], output_transpose_perm), out.results[1]]
  else:
    return [hlo.transpose(out.results[0], output_transpose_perm)]

def _dot_product_attention_bwd_cuda_lowering(
    ctx, query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
    page_table_k, page_table_v, activation, fwd_output, grad_output,
    scale, seed, dropout_rate, variadic_args, mask_type, layout, sliding_window_length):
  query_type = ir.RankedTensorType(query.type)
  query_shape = query_type.shape
  key_type = ir.RankedTensorType(key.type)
  key_shape = key_type.shape
  value_type = ir.RankedTensorType(value.type)

  if layout == AttentionLayout.BNTH.value:
    B, q_N, T, H = query_shape
    _, k_N, S, _ = key_shape
    grad_layout = (3, 2, 1, 0)
    grad_transpose_perm = mlir.dense_int_array((0, 1, 2, 3))
  else:
    B, T, q_N, H = query_shape
    _, S, k_N, _ = key_shape
    grad_layout = (3, 1, 2, 0)
    grad_transpose_perm = mlir.dense_int_array((0, 2, 1, 3))

  workspace_shape = (0,)
  workspace_type = ir.IntegerType.get_unsigned(8)

  grad_query_shape = (B, q_N, T, H)
  grad_key_shape = (B, k_N, S, H)
  grad_value_shape = (B, k_N, S, H)

  has_bias, has_dbias = variadic_args
  max_seg_per_batch = get_max_seg_per_batch(ir.RankedTensorType(q_offsets.type))
  backend_config = create_dot_product_attention_backend_config(
      B, q_N, T, S, query_type.element_type, scale, seed, dropout_rate,
      mask_type, layout, sliding_window_length, max_seg_per_batch,
      False, is_bwd=True)
  # {Q, K, V, activation, dO, bias*, O, q_seqlen*, kv_seqlen*,
  #  q_offsets*, kv_offsets*}
  # {dQ, dK, dV, dbias*, workspace}
  has_dropout = dropout_rate > 0
  # create operands
  operands = [query, key, value, activation, grad_output]
  if has_bias:
    # flash attention requires bias in the bwd for remat
    operands.append(bias)
  operands.append(fwd_output)
  if has_padding(mask_type) or max_seg_per_batch > 1:
    operands.append(q_seqlen)
    operands.append(kv_seqlen)
  if max_seg_per_batch > 1:
    operands.append(q_offsets)
    operands.append(kv_offsets)
  # get custom call name
  custom_call_name = get_custom_call_name(has_bias, has_dropout, True)

  # create output types and layouts
  # grad_query, grad_key, grad_value
  result_types = [
    ir.RankedTensorType.get(grad_query_shape, query_type.element_type),
    ir.RankedTensorType.get(grad_key_shape, key_type.element_type),
    ir.RankedTensorType.get(grad_value_shape, value_type.element_type),
  ]
  result_layouts = [grad_layout, grad_layout, grad_layout]
  bias_type = ir.RankedTensorType(bias.type)
  bias_shape = bias_type.shape
  if has_dbias:
    # cuDNN supports bias for this case
    result_types.append(
      ir.RankedTensorType.get(bias_shape, bias_type.element_type))
    result_layouts = result_layouts + default_layouts(bias_shape)
  # workspace
  result_types.append(ir.RankedTensorType.get(workspace_shape, workspace_type))
  result_layouts = result_layouts + default_layouts(workspace_shape)
  out = mlir.custom_call(
    custom_call_name,
    result_types=result_types,
    operands=operands,
    backend_config=backend_config,
    operand_layouts=default_layouts(
      *[ir.RankedTensorType(operand.type).shape for operand in operands]),
    result_layouts=result_layouts,
  )
  dqkv = (hlo.transpose(out.results[0], grad_transpose_perm),
          hlo.transpose(out.results[1], grad_transpose_perm),
          hlo.transpose(out.results[2], grad_transpose_perm))
  # Only keep dQ, dK, dV and dBias here
  if has_dbias:
    return dqkv + (out.results[3],)
  else:
    return dqkv

# batcher
def _check_valid_batch_dims(bdims):
  for dim in bdims:
    if dim not in [0, None]:
      raise NotImplementedError(
        f""Currently only support batch_dim in [0, None], but got {dim=}"")

def _dot_product_attention_fwd_batcher(
    batched_args, batch_dims, *, scale, seed, dropout_rate, variadic_args,
    mask_type, layout, sliding_window_length, is_training):
  _check_valid_batch_dims(batch_dims)
  query, key, value, bias, q_seqlen, kv_seqlen, \
    q_offsets, kv_offsets, page_table_k, page_table_v = batched_args
  query_bdim = batch_dims[0]
  if is_training:
    out_bdims = query_bdim, query_bdim
  else:
    out_bdims = (query_bdim,)

  if layout == AttentionLayout.BNTH.value:
    *Bs, N, T, _ = query.shape
    *_, _, S, _ = key.shape
  else:
    *Bs, T, N, _ = query.shape
    *_, S, _, _ = key.shape
  B = math.prod(Bs)
  has_bias, _ = variadic_args
  original_shape = query.shape
  # reshape to 4D shape
  query = jnp.reshape(query, (B,) + query.shape[-3:])
  key = jnp.reshape(key, (B,) + key.shape[-3:])
  value = jnp.reshape(value, (B,) + key.shape[-3:])
  if has_bias and batch_dims[3] is not None:
    bias = jnp.reshape(bias, (B, N, T, S))
  if has_padding(mask_type):
    q_seqlen = jnp.reshape(q_seqlen, (B, ))
    kv_seqlen = jnp.reshape(kv_seqlen, (B, ))

  outputs = _dot_product_attention_fwd_p_wrapper.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, scale=scale, seed=seed, dropout_rate=dropout_rate,
      variadic_args=variadic_args, mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length, is_training=is_training)

  # reshape to original shape
  output = outputs[0]
  output = jnp.reshape(output, original_shape)
  if is_training:
    activation = outputs[1]
    activation = jnp.reshape(activation, (*Bs, N, T))
    return (output, activation), out_bdims
  else:
    return (output,), out_bdims

def _dot_product_attention_bwd_batcher(
     batched_args, batch_dims, *, scale, seed, dropout_rate, variadic_args,
     mask_type, layout, sliding_window_length):
  _check_valid_batch_dims(batch_dims)
  query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets, \
    page_table_k, page_table_v, activation, fwd_output, grad_output = batched_args
  query_bdim = batch_dims[0]
  out_bdims = query_bdim, query_bdim, query_bdim

  if layout == AttentionLayout.BNTH.value:
    *Bs, N, T, _ = query.shape
    *_, _, S, _ = key.shape
  else:
    *Bs, T, N, _ = query.shape
    *_, S, _, _ = key.shape
  B = math.prod(Bs)
  has_bias, has_dbias = variadic_args
  # Reset the has_dbias if the combined batch size is not 1, because cuDNN only
  # supports dbias with a single batch. In this case, an all-zero dbias will be
  # appended instead.
  if B > 1:
    variadic_args = (has_bias, False)
  original_query_shape = query.shape
  original_key_shape = key.shape
  original_value_shape = value.shape
  original_bias_shape = bias.shape if has_bias else None
  # reshape to 4D shape
  query = jnp.reshape(query, (B,) + query.shape[-3:])
  key = jnp.reshape(key, (B,) + key.shape[-3:])
  value = jnp.reshape(value, (B,) + key.shape[-3:])
  if has_bias and batch_dims[3] is not None:
    bias = jnp.reshape(bias, (B, N, T, S))
  if has_padding(mask_type):
    q_seqlen = jnp.reshape(q_seqlen, (B, ))
    kv_seqlen = jnp.reshape(kv_seqlen, (B, ))

  activation = jnp.reshape(activation, (B, N, T))
  fwd_output = jnp.reshape(fwd_output, (B,) + query.shape[-3:])
  grad_output = jnp.reshape(grad_output, (B,) + query.shape[-3:])

  grads = _dot_product_attention_bwd_p_wrapper.bind(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, activation, fwd_output, grad_output,
      scale=scale, seed=seed, dropout_rate=dropout_rate, variadic_args=variadic_args,
      mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length,
  )

  # reshape to original shape
  grads[0] = jnp.reshape(grads[0], original_query_shape)
  grads[1] = jnp.reshape(grads[1], original_key_shape)
  grads[2] = jnp.reshape(grads[2], original_value_shape)
  if has_dbias:
    assert has_bias
    if variadic_args[1]:
      grads[3] = jnp.reshape(grads[3], original_bias_shape)
    else:
      grads.append(jnp.zeros(original_bias_shape, bias.dtype))
    out_bdims += (batch_dims[3],)
  return grads, out_bdims

# custom partitioning
def _get_padded_spec(arg_info):
  spec = None if arg_info.sharding is None else arg_info.sharding.spec
  ndim = arg_info.ndim
  if spec is None:
    return (None,) * ndim
  assert len(spec) <= ndim
  return spec + (None,) * (ndim - len(spec))

def _check_qkv_bias_mask_spec(
    query_spec, key_spec, value_spec, bias_spec, layout):
  # check qkv spec
  if not query_spec == key_spec == value_spec:
    raise ValueError(""Query, key and value should have same sharding."")
  if layout == AttentionLayout.BNTH.value:
    *batch_spec, num_head_spec, q_seq_spec, head_spec = query_spec
  else:
    *batch_spec, q_seq_spec, num_head_spec, head_spec = query_spec
  if q_seq_spec is not None:
    raise ValueError(""Sharding on sequence dim is not allowed."")
  if head_spec is not None:
    raise ValueError(""Sharding on head dim is not allowed."")
  # check bias spec
  if bias_spec:
    *bias_batch_spec, bias_num_head_spec, bias_q_seq_spec, bias_kv_seq_spec = bias_spec
    if any(bias_batch_spec) and bias_batch_spec != batch_spec or \
      bias_num_head_spec is not None and bias_num_head_spec != num_head_spec:
      raise ValueError(
        ""Query and bias should have same sharding on batch and num_head dim."")
    if bias_q_seq_spec is not None or bias_kv_seq_spec is not None:
      raise ValueError(""Sharding on bias sequence dim is not allowed."")


# fwd custom partition
def _infer_fwd_output_sharding(mesh, arg_shapes, variadic_args,is_training, layout):
  # only sharding on batch and num_head dim is allowed
  # (*batch, q_seq, num_head, head)
  query_spec = _get_padded_spec(arg_shapes[0])
  # (*batch, kv_seq, num_head, head)
  key_spec = _get_padded_spec(arg_shapes[1])
  value_spec = _get_padded_spec(arg_shapes[2])
  has_bias, _ = variadic_args
  bias_spec = _get_padded_spec(arg_shapes[3]) if has_bias else None

  _check_qkv_bias_mask_spec(
    query_spec, key_spec, value_spec, bias_spec, layout)
  # keep out sharding same as query sharding since they have same shape
  out_sharding = NamedSharding(mesh, PartitionSpec(*query_spec))
  if is_training:
    # activation sharding
    *batch_spec, q_seq_spec, num_head_spec, _ = query_spec
    activation_sharding = NamedSharding(
      mesh, PartitionSpec(*batch_spec, num_head_spec, q_seq_spec, None))
    return [out_sharding, activation_sharding]
  return [out_sharding]

_dot_product_attention_fwd_lower = custom_partitioning(
    _dot_product_attention_fwd_impl, static_argnums=(10, 11, 12, 13, 14, 15, 16, 17))

def _dot_product_attention_fwd_infer_sharding_from_operands(
    scale, seed, dropout_rate, variadic_args, mask_type, layout, sliding_window_length,
    is_training, mesh, arg_shapes, result_shape):
  return _infer_fwd_output_sharding(mesh, arg_shapes, variadic_args, is_training, layout)

def _dot_product_attention_fwd_partition(
    scale, seed, dropout_rate, variadic_args, mask_type, layout, sliding_window_length,
    is_training, mesh, arg_shapes, result_shape):
  # args sharding
  arg_shardings = tuple(arg_i.sharding for arg_i in arg_shapes)
  out_shardings = _infer_fwd_output_sharding(
    mesh, arg_shapes, variadic_args, is_training, layout)
  impl = functools.partial(
      _dot_product_attention_fwd_impl,
      scale=scale,
      seed=seed,
      dropout_rate=dropout_rate,
      variadic_args=variadic_args,
      mask_type=mask_type,
      layout=layout,
      sliding_window_length=sliding_window_length,
      is_training=is_training,
  )
  return mesh, impl, out_shardings, arg_shardings

# bwd custom partition
def _infer_bwd_output_sharding(mesh, arg_shapes, layout, variadic_args):
  # (*batch, q_seq, num_head, head)
  query_spec = _get_padded_spec(arg_shapes[0])
  # (*batch, kv_seq, num_head, head)
  key_spec = _get_padded_spec(arg_shapes[1])
  value_spec = _get_padded_spec(arg_shapes[2])
  has_bias, has_dbias = variadic_args
  bias_spec = _get_padded_spec(arg_shapes[3]) if has_bias else None
  _check_qkv_bias_mask_spec(
    query_spec, key_spec, value_spec, bias_spec, layout)
  # keep grad query sharding same as query sharding
  grad_query_sharding = NamedSharding(mesh, PartitionSpec(*query_spec))
  grad_key_sharding = NamedSharding(mesh, PartitionSpec(*key_spec))
  grad_value_sharding = NamedSharding(mesh, PartitionSpec(*key_spec))
  out_shardings = [grad_query_sharding, grad_key_sharding, grad_value_sharding]
  if has_dbias:
    grad_bias_sharding = NamedSharding(mesh, PartitionSpec(*bias_spec))
    out_shardings = out_shardings + [grad_bias_sharding]
  return out_shardings

_dot_product_attention_bwd_lower = custom_partitioning(
    _dot_product_attention_bwd_impl, static_argnums=(13, 14, 15, 16, 17, 18, 19)
)

def _dot_product_attention_bwd_infer_sharding_from_operands(
    scale, seed, dropout_rate, variadic_args, mask_type, layout,
    sliding_window_length, mesh, arg_shapes, result_shape):
  return _infer_bwd_output_sharding(mesh, arg_shapes, layout, variadic_args)

def _dot_product_attention_bwd_partition(
    scale, seed, dropout_rate, variadic_args, mask_type, layout,
    sliding_window_length, mesh, arg_shapes, result_shape):
  out_shardings = _infer_bwd_output_sharding(mesh, arg_shapes, layout, variadic_args)
  # args sharding
  arg_shardings = tuple(arg_i.sharding for arg_i in arg_shapes)
  def sharded_impl(*args):
    impl = functools.partial(
      _dot_product_attention_bwd_impl,
      scale=scale,
      seed=seed,
      dropout_rate=dropout_rate,
      variadic_args=variadic_args,
      mask_type=mask_type,
      layout=layout,
      sliding_window_length=sliding_window_length,
    )
    grads = impl(*args)
    _, has_dbias = variadic_args
    if has_dbias:
      query_spec = arg_shardings[0].spec
      batch_spec = query_spec[0]
      local_dbias = grads[3]
      global_dbias = jax.lax.psum(local_dbias, batch_spec)
      grads = grads[:3] + [global_dbias]
    return grads
  return mesh, sharded_impl, out_shardings, arg_shardings

# Create dot_product_attention_fwd_p for forward operation.
_dot_product_attention_fwd_p = core.Primitive(""dot_product_attention_fwd"")
_dot_product_attention_fwd_p.multiple_results = True
_dot_product_attention_fwd_p.def_impl(
    functools.partial(dispatch.apply_primitive, _dot_product_attention_fwd_p)
)
_dot_product_attention_fwd_p.def_abstract_eval(
    _dot_product_attention_fwd_abstract
)

mlir.register_lowering(
  _dot_product_attention_fwd_p,
  _dot_product_attention_fwd_cuda_lowering,
  platform=""cuda"",
)

_dot_product_attention_fwd_p_wrapper = core.Primitive(
    ""dot_product_attention_fwd_wrapper""
)
_dot_product_attention_fwd_p_wrapper.multiple_results = True
_dot_product_attention_fwd_p_wrapper.def_impl(_dot_product_attention_fwd_impl)
_dot_product_attention_fwd_p_wrapper.def_abstract_eval(
    _dot_product_attention_fwd_abstract
)

# Create dot_product_attention_bwd_p for backward operation.
_dot_product_attention_bwd_p = core.Primitive(""dot_product_attention_bwd"")
_dot_product_attention_bwd_p.multiple_results = True
_dot_product_attention_bwd_p.def_impl(
    functools.partial(dispatch.apply_primitive, _dot_product_attention_bwd_p)
)
_dot_product_attention_bwd_p.def_abstract_eval(
    _dot_product_attention_bwd_abstract
)

mlir.register_lowering(
  _dot_product_attention_bwd_p,
  _dot_product_attention_bwd_cuda_lowering,
  platform=""cuda"",
)

_dot_product_attention_bwd_p_wrapper = core.Primitive(
    ""dot_product_attention_bwd_wrapper""
)
_dot_product_attention_bwd_p_wrapper.multiple_results = True
_dot_product_attention_bwd_p_wrapper.def_impl(_dot_product_attention_bwd_impl)
_dot_product_attention_bwd_p_wrapper.def_abstract_eval(
    _dot_product_attention_bwd_abstract
)

batching.primitive_batchers[
    _dot_product_attention_fwd_p_wrapper
] = _dot_product_attention_fwd_batcher
batching.primitive_batchers[
    _dot_product_attention_bwd_p_wrapper
] = _dot_product_attention_bwd_batcher

def not_implemented_sharding_rule(*args, **kwargs):
  return NotImplementedError(""Sharding rule not implemented."")

_dot_product_attention_fwd_lower.def_partition(
  infer_sharding_from_operands=_dot_product_attention_fwd_infer_sharding_from_operands,
  partition=_dot_product_attention_fwd_partition,
  sharding_rule=not_implemented_sharding_rule)

mlir.register_lowering(_dot_product_attention_fwd_p_wrapper,
                        mlir.lower_fun(_dot_product_attention_fwd_lower, multiple_results=True))

_dot_product_attention_bwd_lower.def_partition(
  infer_sharding_from_operands=_dot_product_attention_bwd_infer_sharding_from_operands,
  partition=_dot_product_attention_bwd_partition,
  sharding_rule=not_implemented_sharding_rule)

mlir.register_lowering(_dot_product_attention_bwd_p_wrapper,
                        mlir.lower_fun(_dot_product_attention_bwd_lower, multiple_results=True))

dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fwd_p
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fwd_p_wrapper
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_bwd_p
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_bwd_p_wrapper
)

@functools.partial(jax.custom_vjp, nondiff_argnums=(10, 11, 12, 13, 14, 15, 16, 17, 18))
def _dot_product_attention(query: Array,
                           key: Array,
                           value: Array,
                           bias: Array,
                           q_seqlen: Array,
                           kv_seqlen: Array,
                           q_offsets: Array,
                           kv_offsets: Array,
                           page_table_k: Array,
                           page_table_v: Array,
                           scale: float,
                           seed: int,
                           dropout_rate: float,
                           variadic_args: tuple[bool, ...],
                           mask_type: bool,
                           layout: int,
                           sliding_window_length: int | None,
                           cudnn_version: int,
                           return_residual: bool):
  output = _dot_product_attention_fwd(
      query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      page_table_k, page_table_v, scale=scale, seed=seed, dropout_rate=dropout_rate,
      variadic_args=variadic_args, mask_type=mask_type, layout=layout,
      sliding_window_length=sliding_window_length,
      cudnn_version=cudnn_version, return_residual=return_residual)
  return output

_dot_product_attention.defvjp(
    _dot_product_attention_fwd_rule, _dot_product_attention_bwd_rule
)

fp8_params_keys = [
    'amax_dQ', 'amax_dK', 'amax_dV', 'amax_dP', # place holder for bwd output
    'descale_q', 'descale_k', 'descale_v', 'descale_s',
    'scale_s', 'scale_o', 'descale_o', 'descale_dO',
    'descale_dP', 'scale_dQ', 'scale_dK', 'scale_dV',
    'scale_dP'
]

fp8_params_keys_fwd = [
    'descale_q', 'descale_k', 'descale_v', 'descale_s', 'scale_s', 'scale_o'
]
fp8_params_keys_bwd = [
    'descale_q', 'descale_k', 'descale_v', 'descale_o', 'descale_dO', 'descale_s',
    'descale_dP', 'scale_s', 'scale_dQ', 'scale_dK', 'scale_dV', 'scale_dP',
]
params_from_keys = lambda params, keys: [params[key] for key in keys]

def check_fp8_params(params):
  # Check if all required keys are present
  missing_keys = set(fp8_params_keys) - set(params)
  if missing_keys:
    raise ValueError(f""The following keys are missing from fp8_params: {', '.join(missing_keys)}"")

check_is_flash_attention_fp8 = functools.partial(
    check_is_flash_attention,
    has_bias=False,
    is_fp8=True
)

def _dot_product_attention_fp8_fwd(
    query, key, value,
    fp8_params_fwd,
    scale, use_causal_mask, layout, cudnn_version):
  check_is_flash_attention_fp8(
      query, key, layout, cudnn_version, is_training=False)
  descale_q, descale_k, descale_v, descale_s, scale_s, scale_o = fp8_params_fwd
  outputs = _dot_product_attention_fp8_fwd_p_wrapper.bind(
      query, key, value,
      descale_q, descale_k, descale_v, descale_s,
      scale_s, scale_o,
      scale=scale, use_causal_mask=use_causal_mask, layout=layout, is_training=False)
  return outputs

def _dot_product_attention_fp8_fwd_rule(
    query, key, value,
    fp8_params,
    scale, use_causal_mask, layout, cudnn_version):
  check_is_flash_attention_fp8(
      query, key, layout, cudnn_version, is_training=True)

  outputs = _dot_product_attention_fp8_fwd_p_wrapper.bind(
      query, key, value, *params_from_keys(fp8_params, fp8_params_keys_fwd),
      scale=scale, use_causal_mask=use_causal_mask, layout=layout, is_training=True)
  res = (query, key, value, outputs[3], outputs[0], params_from_keys(fp8_params, fp8_params_keys_bwd))
  return (outputs[0], outputs[1], outputs[2]), res

def _dot_product_attention_fp8_bwd_rule(
    scale, use_causal_mask, layout, cudnn_version, res, g):
  (query, key, value, activation, fwd_output, aux_params) = res
  grad_output = g[0]
  grads = _dot_product_attention_fp8_bwd_p_wrapper.bind(
    query,
    key,
    value,
    fwd_output,
    grad_output,
    activation,
    *aux_params,
    scale=scale,
    use_causal_mask=use_causal_mask,
    layout=layout,
    )

  fp8_params_grads = dict.fromkeys(fp8_params_keys)
  keys_to_grad_indices = ['amax_dQ', 'amax_dK', 'amax_dV', 'amax_dP']
  # grads structure: (dQ, dK, dV, amax_dq, amax_dk, amax_dv, amax_dp)
  for i, key in enumerate(keys_to_grad_indices, start=3):
    fp8_params_grads[key] = grads[i]

  return (grads[0], grads[1], grads[2], fp8_params_grads)

def _dot_product_attention_fp8_fwd_impl(
    query, key, value,
    descale_q, descale_k, descale_v, descale_s, scale_s, scale_o,
    scale, use_causal_mask, layout, is_training):
  outputs = _dot_product_attention_fp8_fwd_p.bind(
      query,
      key,
      value,
      descale_q,
      descale_k,
      descale_v,
      descale_s,
      scale_s,
      scale_o,
      scale=scale,
      use_causal_mask=use_causal_mask,
      layout=layout,
      is_training=is_training,
  )
  return outputs

def _dot_product_attention_fp8_bwd_impl(
    query, key, value, fwd_output, grad_output, activation,
    descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s,
    descale_dP, scale_s, scale_dQ, scale_dK, scale_dV, scale_dP,
    scale, use_causal_mask, layout):
  grads = _dot_product_attention_fp8_bwd_p.bind(
      query, key, value, fwd_output, grad_output, activation,
      descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s,
      descale_dP, scale_s, scale_dQ, scale_dK, scale_dV, scale_dP,
      scale=scale, use_causal_mask=use_causal_mask, layout=layout)
  return grads


def _dot_product_attention_fp8_fwd_abstract(
    query, key, value,
    descale_q, descale_k, descale_v, descale_s, scale_s, scale_o,
    scale, use_causal_mask, layout, is_training):
  query_dtype = dtypes.canonicalize_dtype(query.dtype)
  if layout == AttentionLayout.BNTH.value:
    B, N, T, _ = query.shape
    _, _, S, _ = key.shape
  else:
    B, T, N, _ = query.shape
    _, S, _, _ = key.shape
  output_shape = query.shape
  softmax_stat_shape = (B, N, T)

  # output, amax_s, amax_o[, softmax_stat]
  if is_training:
    return (
      core.ShapedArray(output_shape, query_dtype),
      core.ShapedArray((1,1,1,1), jnp.float32),
      core.ShapedArray((1,1,1,1), jnp.float32),
      core.ShapedArray(softmax_stat_shape, jnp.float32),
    )
  else:
    return (
      core.ShapedArray(output_shape, query_dtype),
      core.ShapedArray((1,1,1,1), jnp.float32),
      core.ShapedArray((1,1,1,1), jnp.float32),
    )

def _dot_product_attention_fp8_bwd_abstract(
    query, key, value, fwd_output, grad_output, activation,
    descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s,
    descale_dP, scale_s, scale_dQ, scale_dK, scale_dV, scale_dP,
    scale, use_causal_mask, layout):
  query_dtype = dtypes.canonicalize_dtype(query.dtype)
  key_dtype = dtypes.canonicalize_dtype(key.dtype)
  value_dtype = dtypes.canonicalize_dtype(value.dtype)

  amax_shape = (1,1,1,1)

  return (
    core.ShapedArray(query.shape, query_dtype),
    core.ShapedArray(key.shape, key_dtype),
    core.ShapedArray(value.shape, value_dtype),
    core.ShapedArray(amax_shape, jnp.float32),
    core.ShapedArray(amax_shape, jnp.float32),
    core.ShapedArray(amax_shape, jnp.float32),
    core.ShapedArray(amax_shape, jnp.float32),
  )

def _dot_product_attention_fp8_fwd_cuda_lowering(
    ctx, query, key, value,
    descale_q, descale_k, descale_v, descale_s, scale_s, scale_o,
    scale, use_causal_mask, layout, is_training):
  query_type = ir.RankedTensorType(query.type)
  query_shape = query_type.shape
  key_type = ir.RankedTensorType(key.type)
  key_shape = key_type.shape

  if layout == AttentionLayout.BNTH.value:
    B, N, T, H = query_shape
    _, _, S, _ = key_shape
    output_layout = (3, 2, 1, 0)
    output_transpose_perm = mlir.dense_int_array((0, 1, 2, 3))
  else:
    B, T, N, H = query_shape
    _, S, _, _ = key_shape
    output_layout = (3, 1, 2, 0)
    output_transpose_perm = mlir.dense_int_array((0, 2, 1, 3))

  output_shape = (B, N, T, H)
  softmax_stat_shape = (B, N, T)
  workspace_shape = (0,)
  amax_shape = (1,1,1,1)
  workspace_type = ir.IntegerType.get_unsigned(8)
  mask_type = MaskType.CAUSAL if use_causal_mask else MaskType.NO_MASK
  backend_config = create_dot_product_attention_fp8_backend_config(
      B, N, T, S, ir.BF16Type.get(),  # query_type.element_type,
      scale, mask_type, layout, is_bwd=False,
  )

  operands = [query, key, value, descale_q, descale_k, descale_v, descale_s, scale_s, scale_o]
  custom_call_name = get_fp8_custom_call_name(is_bwd=False)

  if is_training:
    result_types = [
      ir.RankedTensorType.get(output_shape, query_type.element_type),
      ir.RankedTensorType.get((1,1,1,1), ir.F32Type.get()),
      ir.RankedTensorType.get((1,1,1,1), ir.F32Type.get()),
      ir.RankedTensorType.get(softmax_stat_shape, ir.F32Type.get()),
      ir.RankedTensorType.get(workspace_shape, workspace_type),
    ]
    result_layouts = [output_layout] + default_layouts(amax_shape, amax_shape, softmax_stat_shape, workspace_shape)
  else:
    result_types = [
      ir.RankedTensorType.get(output_shape, query_type.element_type),
      ir.RankedTensorType.get((1,1,1,1), ir.F32Type.get()),
      ir.RankedTensorType.get((1,1,1,1), ir.F32Type.get()),
      ir.RankedTensorType.get(workspace_shape, workspace_type)
    ]
    result_layouts = [output_layout] + default_layouts(amax_shape, amax_shape, workspace_shape)

  operand_shapes = [ir.RankedTensorType(operand.type).shape for operand in operands[:3]]
  operand_shapes += [[1, 1, 1, 1]] * 6
  operand_layouts = default_layouts(*operand_shapes)
  out = mlir.custom_call(
    custom_call_name,
    result_types=result_types,
    operands=operands,
    backend_config=backend_config,
    operand_layouts=operand_layouts,
    result_layouts=result_layouts,
  )

  if is_training:
    return [hlo.transpose(out.results[0], output_transpose_perm), out.results[1], out.results[2], out.results[3]]
  else:
    return [hlo.transpose(out.results[0], output_transpose_perm), out.results[1], out.results[2]]



def _dot_product_attention_fp8_bwd_cuda_lowering(
    ctx, query, key, value, fwd_output, grad_output, activation,
    descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s,
    descale_dP, scale_s, scale_dQ, scale_dK, scale_dV, scale_dP, scale,
    use_causal_mask, layout):
  query_type = ir.RankedTensorType(query.type)
  query_shape = query_type.shape
  key_type = ir.RankedTensorType(key.type)
  key_shape = key_type.shape
  value_type = ir.RankedTensorType(value.type)

  if layout == AttentionLayout.BNTH.value:
    B, q_N, T, H = query_shape
    _, k_N, S, _ = key_shape
    grad_layout = (3, 2, 1, 0)
    grad_transpose_perm = mlir.dense_int_array((0, 1, 2, 3))
  else:
    B, T, q_N, H = query_shape
    _, S, k_N, _ = key_shape
    grad_layout = (3, 1, 2, 0)
    grad_transpose_perm = mlir.dense_int_array((0, 2, 1, 3))

  workspace_shape = (0,)
  workspace_type = ir.IntegerType.get_unsigned(8)
  amax_shape = (1,1,1,1)

  grad_query_shape = (B, q_N, T, H)
  grad_key_shape = (B, k_N, S, H)
  grad_value_shape = (B, k_N, S, H)
  mask_type = MaskType.CAUSAL if use_causal_mask else MaskType.NO_MASK

  backend_config = create_dot_product_attention_fp8_backend_config(
      B, q_N, T, S, ir.BF16Type.get(),
      scale, mask_type, layout, is_bwd=True,
  )

  operands = [
    query,
    key,
    value,
    fwd_output,
    grad_output,
    activation,
    descale_q,
    descale_k,
    descale_v,
    descale_o,
    descale_dO,
    descale_s,
    descale_dP,
    scale_s,
    scale_dQ,
    scale_dK,
    scale_dV,
    scale_dP,
  ]

  custom_call_name = get_fp8_custom_call_name(is_bwd=True)

  result_types = [
    ir.RankedTensorType.get(grad_query_shape, query_type.element_type),
    ir.RankedTensorType.get(grad_key_shape, key_type.element_type),
    ir.RankedTensorType.get(grad_value_shape, value_type.element_type),
    ir.RankedTensorType.get(amax_shape, ir.F32Type.get()),
    ir.RankedTensorType.get(amax_shape, ir.F32Type.get()),
    ir.RankedTensorType.get(amax_shape, ir.F32Type.get()),
    ir.RankedTensorType.get(amax_shape, ir.F32Type.get()),
  ]
  result_layouts = [grad_layout, grad_layout, grad_layout] + default_layouts(amax_shape, amax_shape, amax_shape, amax_shape)

  result_types.append(ir.RankedTensorType.get(workspace_shape, workspace_type))
  result_layouts = result_layouts + default_layouts(workspace_shape)
  out = mlir.custom_call(
    custom_call_name,
    result_types=result_types,
    operands=operands,
    backend_config=backend_config,
    operand_layouts=default_layouts(
      *[ir.RankedTensorType(operand.type).shape for operand in operands]),
    result_layouts=result_layouts,
  )
  dqkv_amaxs = (hlo.transpose(out.results[0], grad_transpose_perm),
          hlo.transpose(out.results[1], grad_transpose_perm),
          hlo.transpose(out.results[2], grad_transpose_perm),
          out.results[3], out.results[4], out.results[5], out.results[6])
  # Only keep dQ, dK, dV, amax_dQ, amax_dK, amax_dV, amax_dP here
  return dqkv_amaxs

def _dot_product_attention_fp8_fwd_batcher(
    batched_args, batch_dims, *, scale, use_causal_mask, layout, is_training):
  _check_valid_batch_dims(batch_dims)
  query, key, value,\
    descale_q, descale_k, descale_v, descale_s, scale_s, scale_o, = batched_args
  query_bdim = batch_dims[0]
  if is_training:
    out_bdims = query_bdim, query_bdim
  else:
    out_bdims = (query_bdim,)

  if layout == AttentionLayout.BNTH.value:
    *Bs, N, T, _ = query.shape
    *_, _, S, _ = key.shape
  else:
    *Bs, T, N, _ = query.shape
    *_, S, _, _ = key.shape
  B = math.prod(Bs)

  # reshape to 4D shape
  query = jnp.reshape(query, (B,) + query.shape[-3:])
  key = jnp.reshape(key, (B,) + key.shape[-3:])
  value = jnp.reshape(value, (B,) + key.shape[-3:])

  outputs = _dot_product_attention_fp8_fwd_p_wrapper.bind(
      query, key, value, descale_q, descale_k, descale_v, descale_s, scale_s, scale_o,
      scale=scale, use_causal_mask=use_causal_mask, layout=layout, is_training=is_training)

  # reshape to original shape
  output, amax_s, amax_o = outputs[0], outputs[1], outputs[2]
  output = jnp.reshape(output, query.shape)
  if is_training:
    activation = outputs[3]
    activation = jnp.reshape(activation, (*Bs, N, T))
    return (output, amax_s, amax_o, activation), out_bdims
  else:
    return (output, amax_s, amax_o), out_bdims

def _dot_product_attention_fp8_bwd_batcher(
    batched_args, batch_dims, *, scale, use_causal_mask, layout):
  _check_valid_batch_dims(batch_dims)
  query, key, value, fwd_output, grad_output, activation,\
    descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s, descale_dP,\
    scale_s, scale_dQ, scale_dK, scale_dV, scale_dP = batched_args
  query_bdim = batch_dims[0]
  out_bdims = query_bdim, query_bdim, query_bdim

  if layout == AttentionLayout.BNTH.value:
    *Bs, N, T, _ = query.shape
    *_, _, S, _ = key.shape
  else:
    *Bs, T, N, _ = query.shape
    *_, S, _, _ = key.shape
  B = math.prod(Bs)

  # reshape to 4D shape
  query = jnp.reshape(query, (B,) + query.shape[-3:])
  key = jnp.reshape(key, (B,) + key.shape[-3:])
  value = jnp.reshape(value, (B,) + key.shape[-3:])

  activation = jnp.reshape(activation, (B, N, T))
  fwd_output = jnp.reshape(fwd_output, (B,) + query.shape[-3:])
  grad_output = jnp.reshape(grad_output, (B,) + query.shape[-3:])

  grads = _dot_product_attention_fp8_bwd_p_wrapper.bind(
      query, key, value, fwd_output, grad_output, activation,
      descale_q, descale_k, descale_v, descale_o, descale_dO, descale_s, descale_dP, scale_s, scale_dQ, scale_dK, scale_dV, scale_dP,
      scale=scale, use_causal_mask=use_causal_mask, layout=layout,
  )

  grad_query, grad_key, grad_value = grads[:3]
  # reshape to original shape
  grad_query = jnp.reshape(grad_query, query.shape)
  grad_key = jnp.reshape(grad_key, key.shape)
  grad_value = jnp.reshape(grad_value, value.shape)

  return grads, out_bdims

def _infer_fp8_fwd_output_sharding(mesh, arg_shapes, is_training, layout):
  # Prepare variadic_args for the original function
  has_bias = False  # Adjust as needed
  variadic_args = (has_bias, None)  # Dummy value, adjust as necessary

  # Call the original function with the required parameters
  output_sharding = _infer_fwd_output_sharding(mesh, arg_shapes, variadic_args, is_training, layout)
  amax_sharding = NamedSharding(mesh, PartitionSpec())
  if is_training:
    out_sharding, activation_sharding = output_sharding[0], output_sharding[1]
    return [out_sharding, amax_sharding, amax_sharding, activation_sharding]
  return output_sharding + [amax_sharding, amax_sharding]

_dot_product_attention_fp8_fwd_lower = custom_partitioning(
    _dot_product_attention_fp8_fwd_impl, static_argnums=(9, 10, 11, 12))

def _dot_product_attention_fp8_fwd_infer_sharding_from_operands(
    scale, use_causal_mask, layout, is_training,
    mesh, arg_shapes, result_shape):
  return _infer_fp8_fwd_output_sharding(mesh, arg_shapes, is_training, layout)

def _dot_product_attention_fp8_fwd_partition(
    scale, use_causal_mask, layout, is_training,
    mesh, arg_shapes, result_shape):
  # args sharding
  arg_shardings = tuple(arg_i.sharding for arg_i in arg_shapes)
  out_shardings = _infer_fp8_fwd_output_sharding(
    mesh, arg_shapes, is_training, layout)
  impl = functools.partial(
      _dot_product_attention_fp8_fwd_impl, scale=scale, use_causal_mask=use_causal_mask,
      layout=layout, is_training=is_training)
  return mesh, impl, out_shardings, arg_shardings

def _infer_fp8_bwd_output_sharding(mesh, arg_shapes, layout):
  # Prepare variadic_args for the original function
  has_bias = False  # Adjust as needed
  has_dbias = False  # Adjust as needed
  variadic_args = (has_bias, has_dbias)  # Dummy value, adjust as necessary

  # Call the original function with the required parameters
  output_shardings = _infer_bwd_output_sharding(mesh, arg_shapes, layout, variadic_args)

  # Prepare amax_sharding
  amax_sharding = NamedSharding(mesh, PartitionSpec())  # Use a default spec or adjust as needed

  # Append amax_sharding for each output sharding
  out_shardings_with_amax = output_shardings + [amax_sharding] * 4

  return out_shardings_with_amax

_dot_product_attention_fp8_bwd_lower = custom_partitioning(
    _dot_product_attention_fp8_bwd_impl, static_argnums=(18,19,20)
)

def _dot_product_attention_fp8_bwd_infer_sharding_from_operands(
    scale, use_causal_mask, layout, mesh,
    arg_shapes, result_shape):
  return _infer_fp8_bwd_output_sharding(mesh, arg_shapes, layout)

def _dot_product_attention_fp8_bwd_partition(
    scale, use_causal_mask, layout, mesh,
    arg_shapes, result_shape):
  out_shardings = _infer_fp8_bwd_output_sharding(mesh, arg_shapes, layout)
  # args sharding
  arg_shardings = tuple(arg_i.sharding for arg_i in arg_shapes)
  impl = functools.partial(
      _dot_product_attention_fp8_bwd_impl, scale=scale,
      use_causal_mask=use_causal_mask, layout=layout
  )
  return mesh, impl, out_shardings, arg_shardings

# Create dot_product_attention_fp8_fwd_p for forward operation.
_dot_product_attention_fp8_fwd_p = core.Primitive(""dot_product_attention_fp8_fwd"")
_dot_product_attention_fp8_fwd_p.multiple_results = True
_dot_product_attention_fp8_fwd_p.def_impl(
    functools.partial(dispatch.apply_primitive, _dot_product_attention_fp8_fwd_p)
)
_dot_product_attention_fp8_fwd_p.def_abstract_eval(
    _dot_product_attention_fp8_fwd_abstract
)

mlir.register_lowering(
  _dot_product_attention_fp8_fwd_p,
  _dot_product_attention_fp8_fwd_cuda_lowering,
  platform=""cuda"",
)

_dot_product_attention_fp8_fwd_p_wrapper = core.Primitive(
    ""dot_product_attention_fp8_fwd_wrapper""
)
_dot_product_attention_fp8_fwd_p_wrapper.multiple_results = True
_dot_product_attention_fp8_fwd_p_wrapper.def_impl(_dot_product_attention_fp8_fwd_impl)
_dot_product_attention_fp8_fwd_p_wrapper.def_abstract_eval(
    _dot_product_attention_fp8_fwd_abstract
)

# Create dot_product_attention_bwd_p for backward operation.
_dot_product_attention_fp8_bwd_p = core.Primitive(""dot_product_attention_fp8_bwd"")
_dot_product_attention_fp8_bwd_p.multiple_results = True
_dot_product_attention_fp8_bwd_p.def_impl(
    functools.partial(dispatch.apply_primitive, _dot_product_attention_fp8_bwd_p)
)
_dot_product_attention_fp8_bwd_p.def_abstract_eval(
    _dot_product_attention_fp8_bwd_abstract
)

mlir.register_lowering(
  _dot_product_attention_fp8_bwd_p,
  _dot_product_attention_fp8_bwd_cuda_lowering,
  platform=""cuda"",
)

_dot_product_attention_fp8_bwd_p_wrapper = core.Primitive(
    ""dot_product_attention_fp8_bwd_wrapper""
)
_dot_product_attention_fp8_bwd_p_wrapper.multiple_results = True
_dot_product_attention_fp8_bwd_p_wrapper.def_impl(_dot_product_attention_fp8_bwd_impl)
_dot_product_attention_fp8_bwd_p_wrapper.def_abstract_eval(
    _dot_product_attention_fp8_bwd_abstract
)

batching.primitive_batchers[
    _dot_product_attention_fp8_fwd_p_wrapper
] = _dot_product_attention_fp8_fwd_batcher
batching.primitive_batchers[
    _dot_product_attention_fp8_bwd_p_wrapper
] = _dot_product_attention_fp8_bwd_batcher

_dot_product_attention_fp8_fwd_lower.def_partition(
  infer_sharding_from_operands=_dot_product_attention_fp8_fwd_infer_sharding_from_operands,
  partition=_dot_product_attention_fp8_fwd_partition)

mlir.register_lowering(_dot_product_attention_fp8_fwd_p_wrapper,
                        mlir.lower_fun(_dot_product_attention_fp8_fwd_lower, multiple_results=True))

_dot_product_attention_fp8_bwd_lower.def_partition(
  infer_sharding_from_operands=_dot_product_attention_fp8_bwd_infer_sharding_from_operands,
  partition=_dot_product_attention_fp8_bwd_partition)

mlir.register_lowering(_dot_product_attention_fp8_bwd_p_wrapper,
                        mlir.lower_fun(_dot_product_attention_fp8_bwd_lower, multiple_results=True))

dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fp8_fwd_p
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fp8_fwd_p_wrapper
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fp8_bwd_p
)
dispatch.prim_requires_devices_during_lowering.add(
  _dot_product_attention_fp8_bwd_p_wrapper
)

@functools.partial(jax.custom_vjp, nondiff_argnums=(4, 5, 6, 7))
def _dot_product_attention_fp8(query: Array,
                               key: Array,
                               value: Array,
                               fp8_params: dict[str, Array],
                               scale: float,
                               use_causal_mask: bool,
                               layout: int,
                               cudnn_version: int):
  output, amax_s, amax_o = _dot_product_attention_fp8_fwd(
      query, key, value, params_from_keys(fp8_params, fp8_params_keys_fwd),
      scale, use_causal_mask, layout, cudnn_version
  )
  return output, amax_s, amax_o

_dot_product_attention_fp8.defvjp(_dot_product_attention_fp8_fwd_rule, _dot_product_attention_fp8_bwd_rule)

def combine_bias_and_mask(bias, mask, dtype):
  if bias is not None:
    # reshape bias to have 4D shape
    bias = bias.reshape((1,) * (4 - len(bias.shape)) + bias.shape)

  if mask is not None:
    if mask.dtype == jnp.bool:
      large_negative_number = get_large_negative_number(dtype)
      mask = jnp.where(mask, jnp.asarray(0, dtype), large_negative_number)
    # reshape mask to have 4D shape
    mask = mask.reshape((1,) * (4 - len(mask.shape)) + mask.shape)  # type: ignore[union-attr]

  # combine bias and mask
  if bias is None:
    bias = mask
  else:
    if mask is not None:
      # should be broadcast to same shape
      bias = bias + mask
  return bias

# User interface
def paged_attention(
    query: Array,
    key: Array,
    value: Array,
    q_seqlen: Array,
    kv_seqlen: Array,
    page_table_k: Array,
    page_table_v: Array,
    bias: Array | None = None,
    mask: Array | None = None,
    fp8_params: FP8Params | None = None,
    *,
    scale: float = 1.0,
    mask_type: MaskType = MaskType.NO_MASK,
    seed: int = 42,
    dropout_rate: float = 0.,
    qkv_layout: str = ""BTNH"",
    sliding_window_length: int | None = None,
    use_fp8: bool = False,
    return_residual: bool = False
):
  """"""Computes paged attention described in https://arxiv.org/pdf/2309.06180.

    B = batch size
    S = length of the key/value (source)
    T = length of the query (target)
    N = number of attention heads
    H = dimensions of each attention head.

  Args:
    query: Queries for attention calculation with a shape of BTNH or BNTH.
    key: Keys for attention calculation with a shape of
      [num_blocks, block_size, N, H] or [num_blocks, N, block_size, H] where
      num_blocks = B * Ceil(S / block_size).
    value: Values to be used in attention with a shape of
      [num_blocks, block_size, N, H] or [num_blocks, N, block_size, H] where
      num_blocks = B * Ceil(S / block_size).
    q_seqlen: Non padded sequence length of query with a shape of B.
    kv_seqlen: Non padded sequence length of key and value with a shape of B.
    page_table_k: page table for key of shape [B, 1, num_blocks_per_batch, 1]
      where num_blocks_per_batch = Ceil(S / block_size).
    page_table_v: page table for value of shape [B, 1, num_blocks_per_batch, 1]
      where num_blocks_per_batch = Ceil(S / block_size).
    bias: Bias to be added to logits with a shape of BNTS.
    mask: Mask used to filter out logits with a shape of BNTS.
    scale: Scale for the query.
    qkv_layout: Layout string, with supported formats being BTNH, BNTH, BSNH,
      BNSH.
    sliding_window_length: Window size to make attention only attend to each
      token's left local window (pos - sliding_window_length, pos] where `pos`
      is the index of each token. E.g., if sliding_window_length == 3 and the
      sequence is [0, 1, 2, 3, c, 4, 5], token `c` can attend to [4, 5, c].
    use_fp8: Whether to use FP8 attention mechanism.
    return_residual: Whether to return the logsumexp tensor of shape BTN
      or BNT to users. See section 3.1.1 in the FlashAttention-2 paper:
      https://arxiv.org/pdf/2307.08691 to find the definition of logsumexp.
  Returns:
    output: the same shape as the query.
    residual: the logsumexp tensor if return_residual=True. (non fp8)
  """"""
  cudnn_version = check_cudnn_version()
  layout = _normalize_layout(qkv_layout)
  if use_fp8:
    raise ValueError(""Paged attention doesn't support fp8 for now."")
  if has_padding(mask_type) and (q_seqlen is None or kv_seqlen is None):
    raise ValueError(""Require q_seqlen and kv_seqlen to generate padding mask."")
  if sliding_window_length is not None and sliding_window_length <= 0:
    raise ValueError(
      f""Require sliding_window_length > 0, got {sliding_window_length}."")

  bias = combine_bias_and_mask(bias, mask, query.dtype)
  # check if input shape and data type is compatiable
  check_layout(query, key, value, bias, q_seqlen, kv_seqlen, None, None,
    page_table_k, page_table_v, layout)
  has_bias = bias is not None
  has_dbias = has_bias and \
    should_export_dbias(bias.shape, query.shape, layout)  # type: ignore[union-attr]
  variadic_args = (has_bias, has_dbias)

  _not_used = jnp.zeros(0, dtype=query.dtype)
  if bias is None:
    bias = _not_used

  output = _dot_product_attention(
      query, key, value, bias, q_seqlen, kv_seqlen, _not_used, _not_used,
      page_table_k, page_table_v, scale, seed, dropout_rate, variadic_args,
      mask_type, layout.value, sliding_window_length, cudnn_version,
      return_residual)
  return output


def dot_product_attention(
    query: Array,
    key: Array,
    value: Array,
    bias: Array | None = None,
    mask: Array | None = None,
    q_seqlen: Array | None = None,
    kv_seqlen: Array | None = None,
    q_offsets: Array | None = None,
    kv_offsets: Array | None = None,
    fp8_params: FP8Params | None = None,
    *,
    scale: float = 1.0,
    mask_type: MaskType = MaskType.NO_MASK,
    seed: int = 42,
    dropout_rate: float = 0.,
    qkv_layout: str = ""BTNH"",
    sliding_window_length: int | None = None,
    use_fp8: bool = False,
    return_residual: bool = False
):
  """"""Computes dot-product attention given query (Q), key (K), and value (V).

  This function serves as the core operation for applying attention
  mechanisms as described in the paper [https://arxiv.org/abs/1706.03762].
  Initially, it determines the attention weights by processing Q and K,
  subsequently combining the outcomes using K. Throughout this function, we
  utilize the following uppercase letters to represent specific parameters of
  array:

    B = batch size
    S = length of the key/value (source)
    T = length of the query (target)
    N = number of attention heads
    H = dimensions of each attention head.

  The supported layouts for Q, K, V are either BT(S)NH or BNT(S)H, and they must
  adhere to the same layout. The output layout remains consistent with Q,
  defaulting to BT(S)NH.

  Args:
    query: Queries for attention calculation with a shape of BTNH or BNTH.
    key: Keys for attention calculation with a shape of BSNH or BNSH.
    value: Values to be used in attention with a shape of BSNH or BNSH.
    bias: Bias to be added to logits with a shape of BNTS.
    mask: Mask used to filter out logits with a shape of BNTS.
    q_seqlen: Non padded sequence length of query with a shape of B.
      If q_offsets is set, q_seqlen should have shape [B,M] where M is the
      maximum number of segments per batch. For batch that has less segments
      than maximum segments, fill the padded entries with -1.
    kv_seqlen: Non padded sequence length of key and value with a shape of B.
      If kv_offsets is set, kv_seqlen should have shape [B,M] where M is the
      maximum number of segments per batch. For batch that has less segments
      than maximum segments, fill the padded entries with -1.
    q_offsets: offset of each segment packed in query with a shape of [B,M+1]
      where M is the maximum number of segments per batch. For batch that has
      less segments than maximum segments, fill the padded entries with -1.
      E.g, if 2 batches has 3 and 2 segments respectively, each segment has
      size 1, q_offsets = [[0,1,2,-1], [0,1,-1,-1]]. q_seqlen should be set
      to indicate the size of each segment.
    kv_offsets: offset of each segment packed in key with a shape of [B,M+1]
      where M is the maximum number of segments per batch. For batch that has
      less segments than maximum segments, fill the padded entries with -1.
      E.g, if 2 batches has 3 and 2 segments respectively, each segment has
      size 1, kv_offsets = [[0,1,2,-1], [0,1,-1,-1]]. kv_seqlen should be set
      to indicate the size of each segment.
    scale: Scale for the query.
    dropout_rate: Dropout rate.
    qkv_layout: Layout string, with supported formats being BTNH, BNTH, BSNH,
      BNSH.
    sliding_window_length: Window size to make attention only attend to each
      token's left local window (pos - sliding_window_length, pos] where `pos`
      is the index of each token. E.g., if sliding_window_length == 3 and the
      sequence is [0, 1, 2, 3, c, 4, 5], token `c` can attend to [4, 5, c].
    use_fp8: Whether to use FP8 attention mechanism.
    return_residual: Whether to return the logsumexp tensor of shape BTN
      or BNT to users. See section 3.1.1 in the FlashAttention-2 paper:
      https://arxiv.org/pdf/2307.08691 to find the definition of logsumexp.
  Returns:
    output: the same shape as the query.
    residual: the logsumexp tensor if return_residual=True. (non fp8)
    amax_s: amax of state. (fp8 only)
    amax_o: amax of output. (fp8 only)
  """"""
  # TODO(b/380898464): Check the compute capability, e.g., require GPU device,
  # in the kernel implementation (c++) code.
  cudnn_version = check_cudnn_version()
  layout = _normalize_layout(qkv_layout)

  if use_fp8:
    if fp8_params is None:
      raise ValueError(""fp8_params should not be None."")
    if  mask_type not in (MaskType.NO_MASK, MaskType.CAUSAL):
      raise ValueError(""Only NO_MASK or CAUSAL masks are supported for fp8."")
    if not all(x is None for x in [bias, mask, q_seqlen, kv_seqlen]):
      raise ValueError(
          f""Expected 'None' for bias, mask, q_seqlen, and kv_seqlen, ""
          f""but got: bias={bias}, mask={mask}, q_seqlen={q_seqlen}, kv_seqlen={kv_seqlen}""
      )
    check_fp8_params(fp8_params)
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      None, None, layout)
    output, amax_s, amax_o = _dot_product_attention_fp8(
        query, key, value, fp8_params,
        scale, mask_type == MaskType.CAUSAL, layout.value, cudnn_version
    )
    return output, amax_s, amax_o
  else:
    if has_padding(mask_type) and (q_seqlen is None or kv_seqlen is None):
        raise ValueError(""Require q_seqlen and kv_seqlen to generate padding mask"")
    if sliding_window_length is not None and sliding_window_length <= 0:
      raise ValueError(
        f""Require sliding_window_length > 0, got {sliding_window_length}"")
    if q_offsets is not None and (q_seqlen is None or kv_seqlen is None):
      raise ValueError(""Require q_seqlen and kv_seqlen to use packed layout"")

    bias = combine_bias_and_mask(bias, mask, query.dtype)
    # check if input shape and data type is compatiable
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
      None, None, layout)
    has_bias = bias is not None
    has_dbias = has_bias and \
      should_export_dbias(bias.shape, query.shape, layout)  # type: ignore[union-attr]
    variadic_args = (has_bias, has_dbias)

    _not_used = jnp.zeros(0, dtype=query.dtype)
    if bias is None:
      bias = _not_used
    if q_seqlen is None:
      q_seqlen = _not_used
    if kv_seqlen is None:
      kv_seqlen = _not_used
    if q_offsets is None:
      q_offsets = _not_used
    if kv_offsets is None:
      kv_offsets = _not_used

    output = _dot_product_attention(
        query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
        _not_used, _not_used, scale, seed, dropout_rate, variadic_args,
        mask_type, layout.value, sliding_window_length, cudnn_version,
        return_residual)
    return output
",python,tab
4267,13701216,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77888,0,"",python,selection_command
4268,13706867,"TERMINAL",0,0,"b",,terminal_output
4269,13707002,"TERMINAL",0,0,"i",,terminal_output
4270,13707283,"TERMINAL",0,0,"a",,terminal_output
4271,13707371,"TERMINAL",0,0,"s",,terminal_output
4272,13707431,"TERMINAL",0,0,".",,terminal_output
4273,13707710,"TERMINAL",0,0,"sh",,terminal_output
4274,13707900,"TERMINAL",0,0,"ap",,terminal_output
4275,13707992,"TERMINAL",0,0,"e",,terminal_output
4276,13708166,"TERMINAL",0,0,"
(Pdb) *** AttributeError: 'NoneType' object has no attribute 'shape'
",,terminal_output
4277,13710186,"TERMINAL",0,0,"b",,terminal_output
4278,13710269,"TERMINAL",0,0,"i",,terminal_output
4279,13710468,"TERMINAL",0,0,"a",,terminal_output
4280,13710570,"TERMINAL",0,0,"s",,terminal_output
4281,13710943,"TERMINAL",0,0,"_",,terminal_output
4282,13711708,"TERMINAL",0,0,"6",,terminal_output
4283,13712275,"TERMINAL",0,0," ",,terminal_output
4284,13712569,"TERMINAL",0,0,"4",,terminal_output
4285,13712851,"TERMINAL",0,0,"d",,terminal_output
4286,13712987,"TERMINAL",0,0,".",,terminal_output
4287,13713203,"TERMINAL",0,0,"sh",,terminal_output
4288,13713378,"TERMINAL",0,0,"a",,terminal_output
4289,13713477,"TERMINAL",0,0,"pe",,terminal_output
4290,13713755,"TERMINAL",0,0,"
(Pdb) *** AttributeError: 'NoneType' object has no attribute 'shape'
",,terminal_output
4291,13717175,"TERMINAL",0,0,"l",,terminal_output
4292,13717359,"TERMINAL",0,0,"
(Pdb) 547  	
548  	        bias_4d = _pad(_merge_batch_dims(bias)) if bias is not None else None
549  	
550  	        # NOTE: jax.nn.dot_product_attention does not support dropout
551  	        breakpoint()
552  ->	        output_4d = jax.nn.dot_product_attention(
553  	            query=query_BTHD,
554  	            key=key_BTHD,
555  	            value=value_BTHD,
556  	            bias=bias_4d,
557  	            mask=mask_B1QK,
",,terminal_output
4293,13720094,"TERMINAL",0,0,"v",,terminal_output
4294,13720155,"TERMINAL",0,0,"i",,terminal_output
4295,13720399,"TERMINAL",0,0,"as",,terminal_output
4296,13720556,"TERMINAL",0,0,"d",,terminal_output
4297,13720827,"TERMINAL",0,0," ",,terminal_output
4298,13720992,"TERMINAL",0,0," ",,terminal_output
4299,13721143,"TERMINAL",0,0," ",,terminal_output
4300,13721308,"TERMINAL",0,0," ",,terminal_output
4301,13721359,"TERMINAL",0,0,"b",,terminal_output
4302,13721935,"TERMINAL",0,0," ",,terminal_output
4303,13722080,"TERMINAL",0,0," b",,terminal_output
4304,13722211,"TERMINAL",0,0,"i",,terminal_output
4305,13722494,"TERMINAL",0,0,"as",,terminal_output
4306,13723127,"TERMINAL",0,0,"_",,terminal_output
4307,13723665,"TERMINAL",0,0,"6",,terminal_output
4308,13724296,"TERMINAL",0,0," ",,terminal_output
4309,13724614,"TERMINAL",0,0,"4",,terminal_output
4310,13725199,"TERMINAL",0,0,"d",,terminal_output
4311,13725530,"TERMINAL",0,0,".",,terminal_output
4312,13725747,"TERMINAL",0,0,"sh",,terminal_output
4313,13725916,"TERMINAL",0,0,"a",,terminal_output
4314,13726030,"TERMINAL",0,0,"p",,terminal_output
4315,13726084,"TERMINAL",0,0,"e",,terminal_output
4316,13726332,"TERMINAL",0,0,"
(Pdb) *** AttributeError: 'NoneType' object has no attribute 'shape'
",,terminal_output
4317,13732741,"/fast/home/franz.srambical/jafar/utils/nn.py",0,0,"",python,tab
4318,13732741,"/fast/home/franz.srambical/jafar/utils/nn.py",18581,0,"",python,selection_command
4319,13743352,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",0,0,"",python,tab
4320,13743352,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",37837,0,"",python,selection_command
4321,13756072,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"",python,tab
4322,13756073,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77888,0,"",python,selection_command
4323,13761208,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",0,0,"",python,tab
4324,13761208,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",37837,0,"",python,selection_command
4325,13773171,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",34585,0,"",python,selection_mouse
4326,13775279,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",34567,0,"",python,selection_mouse
4327,13779582,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",35801,0,"",python,selection_mouse
4328,13781237,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",35791,0,"",python,selection_mouse
4329,13781685,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",34981,0,"",python,selection_command
4330,13784795,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",35744,0,"",python,selection_mouse
4331,13786001,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",35803,0,"",python,selection_mouse
4332,13800658,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"",python,tab
4333,13800658,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77888,0,"",python,selection_command
4334,13803487,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",12107,0,"",python,selection_command
4335,13839005,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77888,0,"",python,selection_command
4336,13843928,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",72178,0,"",python,selection_mouse
4337,13874546,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",0,0,"",python,tab
4338,13874547,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py",37837,0,"",python,selection_command
4339,13876805,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"",python,tab
4340,13876806,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77888,0,"",python,selection_command
4341,13885948,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",76970,0,"",python,selection_mouse
4342,13891317,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",10018,0,"",python,selection_command
4343,13897463,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",10051,0,"",python,selection_mouse
4344,13901650,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",76970,0,"",python,selection_command
4345,13905040,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77859,0,"",python,selection_mouse
4346,13905940,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77878,0,"",python,selection_mouse
4347,13906847,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77781,0,"",python,selection_mouse
4348,13907220,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77787,0,"",python,selection_mouse
4349,13912526,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77816,0,"",python,selection_mouse
4350,13936546,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77773,0,"
",python,content
4351,13936582,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77774,0,"    ",python,content
4352,13936855,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77778,0,"b",python,content
4353,13936855,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77779,0,"",python,selection_keyboard
4354,13937059,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77779,0,"r",python,content
4355,13937060,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77780,0,"",python,selection_keyboard
4356,13937185,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77780,0,"e",python,content
4357,13937186,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77781,0,"",python,selection_keyboard
4358,13937816,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77781,0,"a",python,content
4359,13937817,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77782,0,"",python,selection_keyboard
4360,13937924,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77782,0,"k",python,content
4361,13937925,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77783,0,"",python,selection_keyboard
4362,13938013,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77783,0,"p",python,content
4363,13938014,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77784,0,"",python,selection_keyboard
4364,13938077,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77784,0,"o",python,content
4365,13938077,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77785,0,"",python,selection_keyboard
4366,13938128,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77785,0,"i",python,content
4367,13938129,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77786,0,"",python,selection_keyboard
4368,13938410,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77786,0,"n",python,content
4369,13938411,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77787,0,"",python,selection_keyboard
4370,13938518,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77787,0,"t",python,content
4371,13938518,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77788,0,"",python,selection_keyboard
4372,13939079,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77788,0,"()",python,content
4373,13939079,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77789,0,"",python,selection_keyboard
4374,13939157,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77789,1,")",python,content
4375,13939157,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77790,0,"",python,selection_keyboard
4376,13939561,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77789,0,"",python,selection_command
4377,13945275,"TERMINAL",0,0,"q",,terminal_output
4378,13945388,"TERMINAL",0,0,"ui",,terminal_output
4379,13945441,"TERMINAL",0,0,"t",,terminal_output
4380,13945794,"TERMINAL",0,0,"(",,terminal_output
4381,13945870,"TERMINAL",0,0,")",,terminal_output
4382,13946195,"TERMINAL",0,0,"
jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 207, in <module>
    action_batch_E = genie.vq_encode(batch, training=False)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 433, in vq_encode
    lam_output = self.lam.vq_encode(video_BTHWC, training=training)
  File ""/fast/home/franz.srambical/jafar/models/lam.py"", line 151, in vq_encode
    z_BTNp1L = self.encoder(padded_patch_BTNp1P)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 225, in __call__
    x_BTNM = block(x_BTNM)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 119, in __call__
    z_BTNM = self.spatial_attention(z_BTNM)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 552, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 552, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 90, in trace_dispatch
    return self.dispatch_line(frame)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 115, in dispatch_line
    if self.quitting: raise BdbQuit
bdb.BdbQuit
",,terminal_output
4383,13946709,"TERMINAL",0,0,"(Pdb) ",,terminal_output
4384,13946957,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4385,13947528,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4386,13948051,"TERMINAL",0,0,"

",,terminal_output
4387,13959370,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4388,13964354,"TERMINAL",0,0,"> /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4389,13977738,"TERMINAL",0,0,"c",,terminal_output
4390,13977904,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4391,13980106,"TERMINAL",0,0,"m",,terminal_output
4392,13980324,"TERMINAL",0,0,"a",,terminal_output
4393,13980426,"TERMINAL",0,0,"sk",,terminal_output
4394,13980997,"TERMINAL",0,0,".",,terminal_output
4395,13981260,"TERMINAL",0,0,"sh",,terminal_output
4396,13981475,"TERMINAL",0,0,"ape",,terminal_output
4397,13981735,"TERMINAL",0,0,"
(Pdb) (4, 1, 64, 64)
",,terminal_output
4398,14016153,"utils/nn.py",0,0,"",python,tab
4399,14018002,"utils/nn.py",17856,0,"",python,selection_mouse
4400,14116664,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"",python,tab
4401,14120486,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77780,0,"",python,selection_mouse
4402,14121396,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77807,0,"",python,selection_mouse
4403,14121514,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",67527,0,"",python,selection_command
4404,14134885,"TERMINAL",0,0,"c",,terminal_output
4405,14135723,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4406,14135976,"TERMINAL",0,0,"b",,terminal_output
4407,14136155,"TERMINAL",0,0,"i",,terminal_output
4408,14136336,"TERMINAL",0,0,"a",,terminal_output
4409,14136417,"TERMINAL",0,0,"s",,terminal_output
4410,14136522,"TERMINAL",0,0,".",,terminal_output
4411,14136842,"TERMINAL",0,0,"sh",,terminal_output
4412,14137003,"TERMINAL",0,0,"ap",,terminal_output
4413,14137099,"TERMINAL",0,0,"e",,terminal_output
4414,14137322,"TERMINAL",0,0,"
(Pdb) *** AttributeError: 'NoneType' object has no attribute 'shape'
",,terminal_output
4415,14158907,"TERMINAL",0,0,"c",,terminal_output
4416,14162294,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4417,14172666,"TERMINAL",0,0,"m",,terminal_output
4418,14172980,"TERMINAL",0,0,"a",,terminal_output
4419,14173034,"TERMINAL",0,0,"s",,terminal_output
4420,14173923,"TERMINAL",0,0,".",,terminal_output
4421,14174420,"TERMINAL",0,0," ",,terminal_output
4422,14174603,"TERMINAL",0,0,"k",,terminal_output
4423,14174768,"TERMINAL",0,0,".",,terminal_output
4424,14175068,"TERMINAL",0,0,"sh",,terminal_output
4425,14175245,"TERMINAL",0,0,"a",,terminal_output
4426,14175629,"TERMINAL",0,0,"p",,terminal_output
4427,14175723,"TERMINAL",0,0,"e",,terminal_output
4428,14175924,"TERMINAL",0,0,"
(Pdb) (61, 1, 4, 4)
",,terminal_output
4429,14176577,"TERMINAL",0,0,"c",,terminal_output
4430,14179163,"TERMINAL",0,0," ",,terminal_output
4431,14197633,"TERMINAL",0,0,"w",,terminal_output
4432,14197912,"TERMINAL",0,0,"
(Pdb)   /fast/home/franz.srambical/jafar/sample.py(207)<module>()
-> action_batch_E = genie.vq_encode(batch, training=False)
  /fast/home/franz.srambical/jafar/genie.py(433)vq_encode()
-> lam_output = self.lam.vq_encode(video_BTHWC, training=training)
  /fast/home/franz.srambical/jafar/models/lam.py(151)vq_encode()
-> z_BTNp1L = self.encoder(padded_patch_BTNp1P)
  /fast/home/franz.srambical/jafar/utils/nn.py(225)__call__()
-> x_BTNM = block(x_BTNM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py(73)resolve_kwargs_wrapper()
-> return f(*args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(153)split_inputs_wrapper()
-> pure_args_out, pure_out = f(*pure_args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/traceback_util.py(182)reraise_with_filtered_traceback()
-> return fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(332)fun_remat()
-> jaxpr, consts, out_tree = _trace_to_jaxpr(fun_, in_tree, tuple(in_avals), debug)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(425)_trace_to_jaxpr()
-> jaxpr, _, consts, () = pe.trace_to_jaxpr_dynamic(flat_fun, in_avals)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/profiler.py(354)wrapper()
-> return func(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/interpreters/partial_eval.py(2292)trace_to_jaxpr_dynamic()
-> ans = fun.call_wrapped(*in_tracers)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(211)call_wrapped()
-> return self.f_transformed(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/api_util.py(73)flatten_fun()
-> ans = f(*py_args, **py_kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(396)_get_result_paths_thunk()
-> ans = _fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(198)merge_inputs_wrapper()
-> out = f(*args)
  /fast/home/franz.srambical/jafar/utils/nn.py(126)__call__()
-> z_BNTM = self.temporal_attention(z_BNTM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py(605)__call__()
-> x = self.attention_fn(
  /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py(1204)dot_product_attention()
-> out = cudnn_dot_product_attention(
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4433,14215939,"TERMINAL",0,0,"c",,terminal_output
4434,14216225,"TERMINAL",0,0,"
",,terminal_output
4435,14219608,"TERMINAL",0,0,"(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4436,14224826,"TERMINAL",0,0,"c",,terminal_output
4437,14228495,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4438,14229931,"TERMINAL",0,0,"w",,terminal_output
4439,14230098,"TERMINAL",0,0,"
(Pdb)   /fast/home/franz.srambical/jafar/sample.py(207)<module>()
-> action_batch_E = genie.vq_encode(batch, training=False)
  /fast/home/franz.srambical/jafar/genie.py(433)vq_encode()
-> lam_output = self.lam.vq_encode(video_BTHWC, training=training)
  /fast/home/franz.srambical/jafar/models/lam.py(151)vq_encode()
-> z_BTNp1L = self.encoder(padded_patch_BTNp1P)
  /fast/home/franz.srambical/jafar/utils/nn.py(225)__call__()
-> x_BTNM = block(x_BTNM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py(73)resolve_kwargs_wrapper()
-> return f(*args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(153)split_inputs_wrapper()
-> pure_args_out, pure_out = f(*pure_args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/traceback_util.py(182)reraise_with_filtered_traceback()
-> return fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(332)fun_remat()
-> jaxpr, consts, out_tree = _trace_to_jaxpr(fun_, in_tree, tuple(in_avals), debug)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(425)_trace_to_jaxpr()
-> jaxpr, _, consts, () = pe.trace_to_jaxpr_dynamic(flat_fun, in_avals)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/profiler.py(354)wrapper()
-> return func(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/interpreters/partial_eval.py(2292)trace_to_jaxpr_dynamic()
-> ans = fun.call_wrapped(*in_tracers)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(211)call_wrapped()
-> return self.f_transformed(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/api_util.py(73)flatten_fun()
-> ans = f(*py_args, **py_kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(396)_get_result_paths_thunk()
-> ans = _fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(198)merge_inputs_wrapper()
-> out = f(*args)
  /fast/home/franz.srambical/jafar/utils/nn.py(119)__call__()
-> z_BTNM = self.spatial_attention(z_BTNM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py(605)__call__()
-> x = self.attention_fn(
  /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py(1204)dot_product_attention()
-> out = cudnn_dot_product_attention(
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4440,14250939,"TERMINAL",0,0,"m",,terminal_output
4441,14251140,"TERMINAL",0,0,"a",,terminal_output
4442,14251225,"TERMINAL",0,0,"sk",,terminal_output
4443,14251332,"TERMINAL",0,0,".",,terminal_output
4444,14251533,"TERMINAL",0,0,"sh",,terminal_output
4445,14251758,"TERMINAL",0,0,"ap",,terminal_output
4446,14251925,"TERMINAL",0,0,"e",,terminal_output
4447,14252097,"TERMINAL",0,0,"
(Pdb) (4, 1, 64, 64)
",,terminal_output
4448,14259930,"TERMINAL",0,0,"c",,terminal_output
4449,14260121,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4450,14261841,"TERMINAL",0,0,"
",,terminal_output
4451,14262978,"TERMINAL",0,0,"c",,terminal_output
4452,14263140,"TERMINAL",0,0,"
(Pdb) (Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4453,14265312,"TERMINAL",0,0,"w",,terminal_output
4454,14265546,"TERMINAL",0,0,"
(Pdb)   /fast/home/franz.srambical/jafar/sample.py(207)<module>()
-> action_batch_E = genie.vq_encode(batch, training=False)
  /fast/home/franz.srambical/jafar/genie.py(433)vq_encode()
-> lam_output = self.lam.vq_encode(video_BTHWC, training=training)
  /fast/home/franz.srambical/jafar/models/lam.py(151)vq_encode()
-> z_BTNp1L = self.encoder(padded_patch_BTNp1P)
  /fast/home/franz.srambical/jafar/utils/nn.py(225)__call__()
-> x_BTNM = block(x_BTNM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py(73)resolve_kwargs_wrapper()
-> return f(*args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(153)split_inputs_wrapper()
-> pure_args_out, pure_out = f(*pure_args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/traceback_util.py(182)reraise_with_filtered_traceback()
-> return fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(332)fun_remat()
-> jaxpr, consts, out_tree = _trace_to_jaxpr(fun_, in_tree, tuple(in_avals), debug)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/ad_checkpoint.py(425)_trace_to_jaxpr()
-> jaxpr, _, consts, () = pe.trace_to_jaxpr_dynamic(flat_fun, in_avals)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/profiler.py(354)wrapper()
-> return func(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/interpreters/partial_eval.py(2292)trace_to_jaxpr_dynamic()
-> ans = fun.call_wrapped(*in_tracers)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(211)call_wrapped()
-> return self.f_transformed(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/api_util.py(73)flatten_fun()
-> ans = f(*py_args, **py_kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/linear_util.py(396)_get_result_paths_thunk()
-> ans = _fun(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(198)merge_inputs_wrapper()
-> out = f(*args)
  /fast/home/franz.srambical/jafar/utils/nn.py(126)__call__()
-> z_BNTM = self.temporal_attention(z_BNTM)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py(605)__call__()
-> x = self.attention_fn(
  /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py(1204)dot_product_attention()
-> out = cudnn_dot_product_attention(
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4455,14267118,"TERMINAL",0,0,"c",,terminal_output
4456,14267639,"TERMINAL",0,0,"
",,terminal_output
4457,14267690,"TERMINAL",0,0,"(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4458,14268365,"TERMINAL",0,0,"c",,terminal_output
4459,14268632,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4460,14269246,"TERMINAL",0,0,"c",,terminal_output
4461,14269392,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4462,14269801,"TERMINAL",0,0,"c",,terminal_output
4463,14270017,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4464,14270346,"TERMINAL",0,0,"c",,terminal_output
4465,14270532,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4466,14270865,"TERMINAL",0,0,"c",,terminal_output
4467,14271050,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4468,14271363,"TERMINAL",0,0,"c",,terminal_output
4469,14271558,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4470,14271835,"TERMINAL",0,0,"c",,terminal_output
4471,14272022,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4472,14272268,"TERMINAL",0,0,"c",,terminal_output
4473,14272447,"TERMINAL",0,0,"
",,terminal_output
4474,14272546,"TERMINAL",0,0,"2025-08-01 13:33:01.288585: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4475,14272749,"TERMINAL",0,0,"c",,terminal_output
4476,14272932,"TERMINAL",0,0,"
",,terminal_output
4477,14276991,"TERMINAL",0,0,"2025-08-01 13:33:05.737080: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4478,14279555,"TERMINAL",0,0,"(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4479,14321733,"TERMINAL",0,0,"s",,terminal_output
4480,14321975,"TERMINAL",0,0,"s",,terminal_output
4481,14322897,"TERMINAL",0,0," ",,terminal_output
4482,14323114,"TERMINAL",0,0," ",,terminal_output
4483,14323189,"TERMINAL",0,0,"c",,terminal_output
4484,14323421,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4485,14324177,"TERMINAL",0,0,"c",,terminal_output
4486,14324346,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4487,14325031,"TERMINAL",0,0,"c",,terminal_output
4488,14325263,"TERMINAL",0,0,"
",,terminal_output
4489,14330437,"TERMINAL",0,0,"(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4490,14331816,"TERMINAL",0,0,"c",,terminal_output
4491,14331963,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4492,14332464,"TERMINAL",0,0,"c",,terminal_output
4493,14332643,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4494,14332981,"TERMINAL",0,0,"c",,terminal_output
4495,14333147,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4496,14333464,"TERMINAL",0,0,"c",,terminal_output
4497,14333648,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4498,14333966,"TERMINAL",0,0,"c",,terminal_output
4499,14334133,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4500,14334425,"TERMINAL",0,0,"c",,terminal_output
4501,14334668,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4502,14335020,"TERMINAL",0,0,"c",,terminal_output
4503,14335173,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4504,14335581,"TERMINAL",0,0,"c",,terminal_output
4505,14335813,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4506,14336153,"TERMINAL",0,0,"c",,terminal_output
4507,14336386,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4508,14336683,"TERMINAL",0,0,"c",,terminal_output
4509,14336910,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4510,14337257,"TERMINAL",0,0,"c",,terminal_output
4511,14337367,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4512,14337588,"TERMINAL",0,0,"c",,terminal_output
4513,14337837,"TERMINAL",0,0,"
",,terminal_output
4514,14338039,"TERMINAL",0,0,"c",,terminal_output
4515,14338286,"TERMINAL",0,0,"2025-08-01 13:34:07.012873: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.

",,terminal_output
4516,14341921,"TERMINAL",0,0,"(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(552)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1980)dot_product_attention()
-> bias = combine_bias_and_mask(bias, mask, query.dtype)
",,terminal_output
4517,14343933,"TERMINAL",0,0,"c",,terminal_output
4518,14344149,"TERMINAL",0,0,"
jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in sample_causal
    final_carry_causal, _ = jax.lax.scan(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 126, in __call__
    out = self.f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 373, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 323, in __call__
    z_F1M = self.spatial_attention(z_F1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 552, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py"", line 1204, in dot_product_attention
    out = cudnn_dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 1982, in dot_product_attention
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 332, in check_layout
    raise ValueError(
ValueError: Bias must have same seq length as QKV, got 4 and 4
",,terminal_output
4519,14344375,"TERMINAL",0,0,"c",,terminal_output
4520,14344543,"TERMINAL",0,0,"
",,terminal_output
4521,14344980,"TERMINAL",0,0,"(Pdb) ",,terminal_output
4522,14345232,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4523,14360232,"/fast/home/franz.srambical/jafar/sample.py",0,0,"from dataclasses import dataclass
import time
import os
import optax
import math

import dm_pix as pix
import einops
import jax
import jax.numpy as jnp
import flax.linen as nn
import numpy as np
import orbax.checkpoint as ocp
from PIL import Image, ImageDraw
import tyro
from flax import nnx

from genie import Genie
from utils.dataloader import get_dataloader


@dataclass
class Args:
    # Experiment
    seed: int = 0
    seq_len: int = 16
    image_channels: int = 3
    image_height: int = 90
    image_width: int = 160
    data_dir: str = ""data/coinrun_episodes""
    checkpoint: str = """"
    # Sampling
    batch_size: int = 1
    maskgit_steps: int = 25
    temperature: float = 1.0
    sample_argmax: bool = True
    start_frame: int = 0
    # Tokenizer checkpoint
    tokenizer_dim: int = 512
    tokenizer_ffn_dim: int = 2048
    latent_patch_dim: int = 32
    num_patch_latents: int = 1024
    patch_size: int = 4
    tokenizer_num_blocks: int = 4
    tokenizer_num_heads: int = 8
    # LAM checkpoint
    lam_dim: int = 512
    lam_ffn_dim: int = 2048
    latent_action_dim: int = 32
    num_latent_actions: int = 6
    lam_patch_size: int = 16
    lam_num_blocks: int = 4
    lam_num_heads: int = 8
    # Dynamics checkpoint
    dyna_type: str = ""maskgit""
    dyna_dim: int = 512
    dyna_ffn_dim: int = 2048
    dyna_num_blocks: int = 6
    dyna_num_heads: int = 8
    param_dtype = jnp.float32
    dtype = jnp.bfloat16
    use_flash_attention: bool = True


args = tyro.cli(Args)

if __name__ == ""__main__"":
    """"""
    Dimension keys:
        B: batch size
        T: number of input (conditioning) frames
        N: number of patches per frame
        S: sequence length
        H: height
        W: width
        E: B * (S - 1)
    """"""
    jax.distributed.initialize()

    rng = jax.random.key(args.seed)

    # --- Load Genie checkpoint ---
    rngs = nnx.Rngs(rng)
    genie = Genie(
        # Tokenizer
        in_dim=args.image_channels,
        tokenizer_dim=args.tokenizer_dim,
        tokenizer_ffn_dim=args.tokenizer_ffn_dim,
        latent_patch_dim=args.latent_patch_dim,
        num_patch_latents=args.num_patch_latents,
        patch_size=args.patch_size,
        tokenizer_num_blocks=args.tokenizer_num_blocks,
        tokenizer_num_heads=args.tokenizer_num_heads,
        # LAM
        lam_dim=args.lam_dim,
        lam_ffn_dim=args.lam_ffn_dim,
        latent_action_dim=args.latent_action_dim,
        num_latent_actions=args.num_latent_actions,
        lam_patch_size=args.lam_patch_size,
        lam_num_blocks=args.lam_num_blocks,
        lam_num_heads=args.lam_num_heads,
        lam_co_train=False,
        # Dynamics
        dyna_type=args.dyna_type,
        dyna_dim=args.dyna_dim,
        dyna_ffn_dim=args.dyna_ffn_dim,
        dyna_num_blocks=args.dyna_num_blocks,
        dyna_num_heads=args.dyna_num_heads,
        param_dtype=args.param_dtype,
        dtype=args.dtype,
        use_flash_attention=args.use_flash_attention,
        decode=True,
        rngs=rngs,
    )

    handler_registry = ocp.handlers.DefaultCheckpointHandlerRegistry()
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeSave, ocp.handlers.PyTreeCheckpointHandler
    )
    handler_registry.add(
        ""model_state"", ocp.args.PyTreeRestore, ocp.handlers.PyTreeCheckpointHandler
    )
    checkpoint_options = ocp.CheckpointManagerOptions(
        step_format_fixed_length=6,
    )
    checkpoint_manager = ocp.CheckpointManager(
        args.checkpoint,
        options=checkpoint_options,
        handler_registry=handler_registry,
    )

    dummy_tx = optax.adamw(
        learning_rate=optax.linear_schedule(0.0001, 0.0001, 10000),
        b1=0.9,
        b2=0.9,
        weight_decay=1e-4,
        mu_dtype=args.dtype,
    )
    dummy_optimizer = nnx.Optimizer(genie, dummy_tx)

    abstract_optimizer = nnx.eval_shape(lambda: dummy_optimizer)
    abstract_optimizer_state = nnx.state(abstract_optimizer)
    restored = checkpoint_manager.restore(
        checkpoint_manager.latest_step(),
        args=ocp.args.Composite(
            model_state=ocp.args.PyTreeRestore(abstract_optimizer_state),  # type: ignore
        ),
    )
    restored_optimizer_state = restored[""model_state""]
    nnx.update(dummy_optimizer, restored_optimizer_state)

    # --- Define sampling function ---
    def _sampling_fn(model: Genie, batch: dict) -> jax.Array:
        """"""Runs Genie.sample with pre-defined generation hyper-parameters.""""""
        if args.dyna_type == ""maskgit"":
            return model.sample(
                batch,
                args.seq_len,
                args.maskgit_steps,
                args.temperature,
                args.sample_argmax,
            )
        elif args.dyna_type == ""causal"":
            return model.sample_causal(
                batch,
                args.seq_len,
                args.temperature,
                args.sample_argmax,
            )
        else:
            raise ValueError(f""Invalid dynamics type: {args.dyna_type}"")

    # --- Define autoregressive sampling loop ---
    # FIXME (f.srambical): why is kv caching not working with nnx.jit?
    #@nnx.jit
    def _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E):
        input_video_BTHWC = video_batch_BSHWC[:, : args.start_frame]
        rng, _rng = jax.random.split(rng)
        batch = dict(videos=input_video_BTHWC, latent_actions=action_batch_E, rng=_rng)
        generated_vid_BSHWC = _sampling_fn(genie, batch)
        return generated_vid_BSHWC

    # --- Get video + latent actions ---
    array_record_files = [
        os.path.join(args.data_dir, x)
        for x in os.listdir(args.data_dir)
        if x.endswith("".array_record"")
    ]
    dataloader = get_dataloader(
        array_record_files,
        args.seq_len,
        args.batch_size,
        args.image_height,
        args.image_width,
        args.image_channels,
        # We don't use workers in order to avoid grain shutdown issues (https://github.com/google/grain/issues/398)
        num_workers=0,
        prefetch_buffer_size=1,
        seed=args.seed,
    )
    dataloader = iter(dataloader)
    video_batch_BSHWC = next(dataloader)
    gt_video = jnp.asarray(video_batch_BSHWC, dtype=jnp.float32) / 255.0
    video_batch_BSHWC = gt_video.astype(args.dtype)
    # Get latent actions for all videos in the batch
    batch = dict(videos=video_batch_BSHWC)
    action_batch_E = genie.vq_encode(batch, training=False)

    # --- Sample + evaluate video ---
    # B, S, H, W, _ = video_batch_BSHWC.shape
    # N = math.ceil(H / args.patch_size) * math.ceil(W / args.patch_size)
    # for block in genie.dynamics.transformer.blocks:
    #     block.spatial_attention.init_cache((B * S, 1, args.dyna_dim), dtype=args.dtype)
    #     block.temporal_attention.init_cache((B * (N + 1), 1, args.dyna_dim), dtype=args.dtype)

    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
    recon_video_BSHWC = recon_video_BSHWC.astype(jnp.float32)
    gt = gt_video[:, : recon_video_BSHWC.shape[1]].clip(0, 1).reshape(-1, *gt_video.shape[2:])
    recon = recon_video_BSHWC.clip(0, 1).reshape(-1, *recon_video_BSHWC.shape[2:])
    ssim = jnp.asarray(
        pix.ssim(gt[:, args.start_frame + 1 :], recon[:, args.start_frame + 1 :])
    ).mean()
    print(f""SSIM: {ssim}"")

    # --- Construct video ---
    true_videos = (gt_video * 255).astype(np.uint8)
    pred_videos = (recon_video_BSHWC * 255).astype(np.uint8)
    video_comparison = np.zeros((2, *recon_video_BSHWC.shape), dtype=np.uint8)
    video_comparison[0] = true_videos[:, : args.seq_len]
    video_comparison[1] = pred_videos
    frames = einops.rearrange(video_comparison, ""n b t h w c -> t (b h) (n w) c"")

    # --- Save video ---
    imgs = [Image.fromarray(img) for img in frames]
    # Write actions on each frame, on each row (i.e., for each video in the batch, on the GT row)
    B, S, _, _, _ = video_batch_BSHWC.shape
    action_batch_BSm11 = jnp.reshape(action_batch_E, (B, S-1, 1))
    for t, img in enumerate(imgs[1:]):
        d = ImageDraw.Draw(img)
        for row in range(action_batch_BSm11.shape[0]):
            action = action_batch_BSm11[row, t, 0]
            y_offset = row * video_batch_BSHWC.shape[2] + 2
            d.text((2, y_offset), f""{action}"", fill=255)
    imgs[0].save(
        f""generation_{time.time()}.gif"",
        save_all=True,
        append_images=imgs[1:],
        duration=250,
        loop=0,
    )
",python,tab
4524,14360232,"/fast/home/franz.srambical/jafar/sample.py",5421,0,"",python,selection_command
4525,14361830,"/fast/home/franz.srambical/jafar/sample.py",4750,0,"",python,selection_command
4526,14363777,"/fast/home/franz.srambical/jafar/genie.py",0,0,"",python,tab
4527,14363777,"/fast/home/franz.srambical/jafar/genie.py",17094,0,"",python,selection_command
4528,14363879,"/fast/home/franz.srambical/jafar/genie.py",14275,2358,"            block.spatial_attention.init_cache((B * seq_len, (N + 1), self.dyna_dim), dtype=self.dtype)
            block.temporal_attention.init_cache((B * (N + 1), seq_len, self.dyna_dim), dtype=self.dtype)

        @nnx.jit
        def causal_step_fn(
            carry: tuple[jax.Array, jax.Array, jax.Array, jax.Array], step_n: jax.Array
        ) -> tuple[tuple[jax.Array, jax.Array, jax.Array, jax.Array], None]:
            rng, token_idxs_BSN, action_tokens_EL, step_t = carry
            S, N = token_idxs_BSN.shape[1:]
            L = action_tokens_EL.shape[-1]

            # --- Construct + encode video ---
            vid_embed_BSNM = dynamics_causal.patch_embed(token_idxs_BSN)

            # --- Predict transition ---
            action_tokens_BSm1L = jnp.reshape(action_tokens_EL, (B, S - 1, L))
            act_embed_BSm1M = dynamics_causal.action_up(action_tokens_BSm1L)
            act_embed_BSM = jnp.pad(act_embed_BSm1M, ((0, 0), (1, 0), (0, 0)))
            act_embed_BS1M = jnp.reshape(act_embed_BSM, (B, S, 1, act_embed_BSM.shape[-1]))
            vid_embed_BSNp1M = jnp.concatenate([act_embed_BS1M, vid_embed_BSNM], axis=2)
            final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
            final_logits_BV = final_logits_BTNp1V[:, step_t, step_n, :]

            # --- Sample new tokens for final frame ---
            if sample_argmax:
                sampled_token_idxs_B = jnp.argmax(final_logits_BV, axis=-1)
            else:
                rng, _rng = jax.random.split(rng)
                sampled_token_idxs_B = jax.random.categorical(_rng, final_logits_BV)
            # Update next tokens only
            token_idxs_BSN = token_idxs_BSN.at[:, step_t, step_n].set(sampled_token_idxs_B)
            step_t += 1

            new_carry = (rng, token_idxs_BSN, action_tokens_EL, step_t)
            return new_carry, None

        # --- Run the autoregressive generation using a for loop ---
        rng = batch[""rng""]
        current_token_idxs_BSN = token_idxs_BSN
        
        for step_t in range(T, seq_len):
            rng, step_rng = jax.random.split(rng)

            # --- Reset spatial KV caches before each frame ---
            # for block in dynamics_causal.transformer.blocks:
            #     block.spatial_attention.init_cache((B * seq_len, 1, self.dyna_dim), dtype=self.dtype)
            #breakpoint()
",python,content
4529,14380547,"experiments/sample.sh",0,0,"",shellscript,tab
4530,14381311,"experiments/sample.sh",266,0,"",shellscript,selection_command
4531,14381504,"experiments/sample.sh",248,0,"",shellscript,selection_command
4532,14382190,"experiments/sample.sh",248,1,"2",shellscript,content
4533,14384314,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4534,14384607,"TERMINAL",0,0,"

",,terminal_output
4535,14386135,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.46 task 0: running
",,terminal_output
4536,14386228,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.46
srun: forcing job termination
Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 13, in <module>
    import orbax.checkpoint as ocp
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/__init__.py"", line 22, in <module>
    from orbax.checkpoint.experimental import v1
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/experimental/v1/__init__.py"", line 24, in <module>
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
    from orbax.checkpoint.experimental.v1 import handlers
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/experimental/v1/handlers.py"", line 19, in <module>
[2025-08-01T13:34:54.949] error: *** STEP 14571.46 ON hai005 CANCELLED AT 2025-08-01T13:34:54 DUE to SIGNAL Killed ***
    import orbax.checkpoint.experimental.v1._src.handlers.global_registration
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/global_registration.py"", line 25, in <module>
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4537,14386425,"TERMINAL",0,0,"^C


[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4538,14388087,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",0,0,"",python,tab
4539,14390467,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77778,0,"",python,selection_command
4540,14391042,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77774,17,"",python,content
4541,14391072,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py",77778,0,"",python,selection_command
4542,14393182,"utils/nn.py",0,0,"",python,tab
4543,14399978,"utils/nn.py",18589,0,"",python,selection_command
4544,14400956,"utils/nn.py",18581,21,"",python,content
4545,14400980,"utils/nn.py",18589,0,"",python,selection_command
4546,14401165,"utils/nn.py",18590,0,"",python,selection_command
4547,14401777,"utils/nn.py",18589,0,"",python,selection_command
4548,14405027,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4549,14405331,"TERMINAL",0,0,"

",,terminal_output
4550,14408079,"experiments/sample.sh",0,0,"",shellscript,tab
4551,14416479,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4552,14425504,"TERMINAL",0,0,"2025-08-01 13:35:34.249104: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4553,14429787,"TERMINAL",0,0,"2025-08-01 13:35:38.529676: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4554,14437935,"TERMINAL",0,0,"2025-08-01 13:35:46.678973: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4555,14438528,"experiments/sample.sh",249,0,"",shellscript,selection_command
4556,14441723,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in sample_causal
    final_carry_causal, _ = jax.lax.scan(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 126, in __call__
    out = self.f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 373, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 323, in __call__
    z_F1M = self.spatial_attention(z_F1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 551, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py"", line 1204, in dot_product_attention
    out = cudnn_dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 1981, in dot_product_attention
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 332, in check_layout
    raise ValueError(
ValueError: Bias must have same seq length as QKV, got 4 and 4
",,terminal_output
4557,14442808,"TERMINAL",0,0,"srun: error: hai005: task 0: Exited with exit code 1
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4558,14471882,"/fast/home/franz.srambical/jafar/utils/nn.py",0,0,"",python,tab
4559,14471882,"/fast/home/franz.srambical/jafar/utils/nn.py",18581,0,"",python,selection_command
4560,14471974,"/fast/home/franz.srambical/jafar/utils/nn.py",18581,21,"",python,content
4561,14476723,"/fast/home/franz.srambical/jafar/utils/nn.py",10044,0,"",python,selection_command
4562,14504558,"/fast/home/franz.srambical/jafar/utils/nn.py",10043,0,"
            ",python,content
4563,14505571,"/fast/home/franz.srambical/jafar/utils/nn.py",10056,0,"p",python,content
4564,14505572,"/fast/home/franz.srambical/jafar/utils/nn.py",10057,0,"",python,selection_keyboard
4565,14506345,"/fast/home/franz.srambical/jafar/utils/nn.py",10057,0,"r",python,content
4566,14506346,"/fast/home/franz.srambical/jafar/utils/nn.py",10058,0,"",python,selection_keyboard
4567,14506442,"/fast/home/franz.srambical/jafar/utils/nn.py",10058,0,"i",python,content
4568,14506442,"/fast/home/franz.srambical/jafar/utils/nn.py",10059,0,"",python,selection_keyboard
4569,14506627,"/fast/home/franz.srambical/jafar/utils/nn.py",10059,0,"n",python,content
4570,14506627,"/fast/home/franz.srambical/jafar/utils/nn.py",10060,0,"",python,selection_keyboard
4571,14506657,"/fast/home/franz.srambical/jafar/utils/nn.py",10060,0,"t",python,content
4572,14506657,"/fast/home/franz.srambical/jafar/utils/nn.py",10061,0,"",python,selection_keyboard
4573,14507138,"/fast/home/franz.srambical/jafar/utils/nn.py",10061,0,"()",python,content
4574,14507138,"/fast/home/franz.srambical/jafar/utils/nn.py",10062,0,"",python,selection_keyboard
4575,14507571,"/fast/home/franz.srambical/jafar/utils/nn.py",10062,0,"z_F1M.shape",python,content
4576,14507834,"/fast/home/franz.srambical/jafar/utils/nn.py",10072,0,"",python,selection_command
4577,14509553,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4578,14509845,"TERMINAL",0,0,"

",,terminal_output
4579,14521068,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4580,14529980,"TERMINAL",0,0,"2025-08-01 13:37:18.656239: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4581,14534160,"TERMINAL",0,0,"2025-08-01 13:37:22.906272: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4582,14542214,"TERMINAL",0,0,"2025-08-01 13:37:30.959975: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4583,14544994,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.48 task 0: running
",,terminal_output
4584,14545093,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.48
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T13:37:33.840] error: *** STEP 14571.48 ON hai005 CANCELLED AT 2025-08-01T13:37:33 DUE to SIGNAL Killed ***
",,terminal_output
4585,14545470,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.48
srun: job abort in progress
^Csrun: sending Ctrl-C to StepId=14571.48
^Csrun: sending Ctrl-C to StepId=14571.48
^Csrun: sending Ctrl-C to StepId=14571.48
[franz.srambical@hai005.haicore.berlin:~/jafar] $ ^C


[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4586,14549484,"/fast/home/franz.srambical/jafar/utils/nn.py",10044,31,"",python,content
4587,14549500,"/fast/home/franz.srambical/jafar/utils/nn.py",10056,0,"",python,selection_command
4588,14554442,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4589,14555145,"TERMINAL",0,0,"",,terminal_output
4590,14555752,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4591,14558616,"experiments/sample.sh",0,0,"",shellscript,tab
4592,14559397,"experiments/sample.sh",248,0,"",shellscript,selection_command
4593,14560006,"experiments/sample.sh",227,0,"",shellscript,selection_command
4594,14560160,"experiments/sample.sh",200,0,"",shellscript,selection_command
4595,14560337,"experiments/sample.sh",176,0,"",shellscript,selection_command
4596,14560423,"experiments/sample.sh",174,0,"",shellscript,selection_command
4597,14560616,"experiments/sample.sh",167,0,"",shellscript,selection_command
4598,14560773,"experiments/sample.sh",162,0,"",shellscript,selection_command
4599,14561226,"experiments/sample.sh",167,0,"",shellscript,selection_command
4600,14563921,"experiments/sample.sh",172,0,"",shellscript,selection_command
4601,14564880,"experiments/sample.sh",173,0,"",shellscript,selection_command
4602,14565061,"experiments/sample.sh",173,0," ",shellscript,content
4603,14565061,"experiments/sample.sh",174,0,"",shellscript,selection_keyboard
4604,14566233,"experiments/sample.sh",174,0,"-",shellscript,content
4605,14566233,"experiments/sample.sh",175,0,"",shellscript,selection_keyboard
4606,14566747,"experiments/sample.sh",175,0,"m",shellscript,content
4607,14566748,"experiments/sample.sh",176,0,"",shellscript,selection_keyboard
4608,14566987,"experiments/sample.sh",176,0," ",shellscript,content
4609,14566987,"experiments/sample.sh",177,0,"",shellscript,selection_keyboard
4610,14567834,"experiments/sample.sh",177,0,"p",shellscript,content
4611,14567834,"experiments/sample.sh",178,0,"",shellscript,selection_keyboard
4612,14567990,"experiments/sample.sh",178,0,"d",shellscript,content
4613,14567990,"experiments/sample.sh",179,0,"",shellscript,selection_keyboard
4614,14568079,"experiments/sample.sh",179,0,"b",shellscript,content
4615,14568079,"experiments/sample.sh",180,0,"",shellscript,selection_keyboard
4616,14570366,"experiments/sample.sh",180,0," ",shellscript,content
4617,14570367,"experiments/sample.sh",181,0,"",shellscript,selection_keyboard
4618,14570844,"experiments/sample.sh",181,0,"0",shellscript,content
4619,14570844,"experiments/sample.sh",182,0,"",shellscript,selection_keyboard
4620,14570964,"experiments/sample.sh",182,0,"0",shellscript,content
4621,14570964,"experiments/sample.sh",183,0,"",shellscript,selection_keyboard
4622,14571442,"experiments/sample.sh",182,1,"",shellscript,content
4623,14571706,"experiments/sample.sh",182,0,"-",shellscript,content
4624,14571706,"experiments/sample.sh",183,0,"",shellscript,selection_keyboard
4625,14572057,"experiments/sample.sh",182,1,"",shellscript,content
4626,14572197,"experiments/sample.sh",181,1,"",shellscript,content
4627,14572489,"experiments/sample.sh",181,0,"-",shellscript,content
4628,14572489,"experiments/sample.sh",182,0,"",shellscript,selection_keyboard
4629,14572699,"experiments/sample.sh",182,0,"c",shellscript,content
4630,14572699,"experiments/sample.sh",183,0,"",shellscript,selection_keyboard
4631,14573153,"experiments/sample.sh",183,0," ",shellscript,content
4632,14573153,"experiments/sample.sh",184,0,"",shellscript,selection_keyboard
4633,14573359,"experiments/sample.sh",184,0,"d",shellscript,content
4634,14573359,"experiments/sample.sh",185,0,"",shellscript,selection_keyboard
4635,14573627,"experiments/sample.sh",184,1,"",shellscript,content
4636,14573797,"experiments/sample.sh",183,1,"",shellscript,content
4637,14573853,"experiments/sample.sh",183,0,"c",shellscript,content
4638,14573853,"experiments/sample.sh",184,0,"",shellscript,selection_keyboard
4639,14574432,"experiments/sample.sh",184,0," ",shellscript,content
4640,14574433,"experiments/sample.sh",185,0,"",shellscript,selection_keyboard
4641,14574497,"experiments/sample.sh",184,1,"",shellscript,content
4642,14574682,"experiments/sample.sh",183,1,"",shellscript,content
4643,14574855,"experiments/sample.sh",183,0," ",shellscript,content
4644,14574855,"experiments/sample.sh",184,0,"",shellscript,selection_keyboard
4645,14574857,"experiments/sample.sh",184,0,"c",shellscript,content
4646,14574857,"experiments/sample.sh",185,0,"",shellscript,selection_keyboard
4647,14575053,"experiments/sample.sh",184,1,"",shellscript,content
4648,14575492,"experiments/sample.sh",184,0,"c",shellscript,content
4649,14575492,"experiments/sample.sh",185,0,"",shellscript,selection_keyboard
4650,14575621,"experiments/sample.sh",185,0,"o",shellscript,content
4651,14575621,"experiments/sample.sh",186,0,"",shellscript,selection_keyboard
4652,14575717,"experiments/sample.sh",186,0,"n",shellscript,content
4653,14575718,"experiments/sample.sh",187,0,"",shellscript,selection_keyboard
4654,14575818,"experiments/sample.sh",187,0,"t",shellscript,content
4655,14575818,"experiments/sample.sh",188,0,"",shellscript,selection_keyboard
4656,14575940,"experiments/sample.sh",188,0,"i",shellscript,content
4657,14575940,"experiments/sample.sh",189,0,"",shellscript,selection_keyboard
4658,14576014,"experiments/sample.sh",189,0,"n",shellscript,content
4659,14576014,"experiments/sample.sh",190,0,"",shellscript,selection_keyboard
4660,14576235,"experiments/sample.sh",190,0,"u",shellscript,content
4661,14576235,"experiments/sample.sh",191,0,"",shellscript,selection_keyboard
4662,14576350,"experiments/sample.sh",191,0,"e",shellscript,content
4663,14576350,"experiments/sample.sh",192,0,"",shellscript,selection_keyboard
4664,14576656,"experiments/sample.sh",191,0,"",shellscript,selection_command
4665,14579906,"TERMINAL",0,0,"

",,terminal_output
4666,14591203,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4667,14591839,"/fast/home/franz.srambical/jafar/utils/nn.py",0,0,"",python,tab
4668,14592641,"experiments/sample.sh",0,0,"",shellscript,tab
4669,14593451,"utils/nn.py",0,0,"",python,tab
4670,14600085,"TERMINAL",0,0,"2025-08-01 13:38:28.833035: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4671,14604359,"TERMINAL",0,0,"2025-08-01 13:38:33.101441: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4672,14612383,"TERMINAL",0,0,"2025-08-01 13:38:41.130874: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4673,14616070,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Uncaught exception. Entering post mortem debugging
Running 'cont' or 'step' will restart the program
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py"", line 1723, in main
    pdb._runscript(mainpyfile)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py"", line 1583, in _runscript
    self.run(statement)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 598, in run
    exec(cmd, globals, locals)
  File ""<string>"", line 1, in <module>
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in sample_causal
    final_carry_causal, _ = jax.lax.scan(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 126, in __call__
    out = self.f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 373, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 323, in __call__
    z_F1M = self.spatial_attention(z_F1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 551, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py"", line 1204, in dot_product_attention
    out = cudnn_dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 1981, in dot_product_attention
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 332, in check_layout
    raise ValueError(
ValueError: Bias must have same seq length as QKV, got 4 and 4
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(332)check_layout()
-> raise ValueError(
",,terminal_output
4674,14643373,"TERMINAL",0,0,"w",,terminal_output
4675,14643591,"TERMINAL",0,0,"
(Pdb)   /home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py(1723)main()
-> pdb._runscript(mainpyfile)
  /home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py(1583)_runscript()
-> self.run(statement)
  /home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py(598)run()
-> exec(cmd, globals, locals)
  <string>(1)<module>()
  /fast/home/franz.srambical/jafar/sample.py(216)<module>()
-> recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  /fast/home/franz.srambical/jafar/sample.py(180)_autoreg_sample()
-> generated_vid_BSHWC = _sampling_fn(genie, batch)
  /fast/home/franz.srambical/jafar/sample.py(164)_sampling_fn()
-> return model.sample_causal(
  /fast/home/franz.srambical/jafar/genie.py(411)sample_causal()
-> final_carry_causal, _ = jax.lax.scan(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(431)__call__()
-> pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(126)__call__()
-> out = self.f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/genie.py(373)causal_step_fn()
-> final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  /fast/home/franz.srambical/jafar/utils/nn.py(443)__call__()
-> x_BTNM = block(x_BTNM, pos_index)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py(73)resolve_kwargs_wrapper()
-> return f(*args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(153)split_inputs_wrapper()
-> pure_args_out, pure_out = f(*pure_args)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(198)merge_inputs_wrapper()
-> out = f(*args)
  /fast/home/franz.srambical/jafar/utils/nn.py(323)__call__()
-> z_F1M = self.spatial_attention(z_F1M)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py(605)__call__()
-> x = self.attention_fn(
  /fast/home/franz.srambical/jafar/utils/nn.py(551)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py(1204)dot_product_attention()
-> out = cudnn_dot_product_attention(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1981)dot_product_attention()
-> check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(332)check_layout()
-> raise ValueError(
",,terminal_output
4676,14651065,"TERMINAL",0,0,"u",,terminal_output
4677,14651195,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(1981)dot_product_attention()
-> check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
",,terminal_output
4678,14701683,"TERMINAL",0,0,"u",,terminal_output
4679,14701925,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py(1204)dot_product_attention()
-> out = cudnn_dot_product_attention(
",,terminal_output
4680,14702595,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(551)attention_fn()
-> output_4d = jax.nn.dot_product_attention(
",,terminal_output
4681,14704859,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py(605)__call__()
-> x = self.attention_fn(
",,terminal_output
4682,14706311,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(323)__call__()
-> z_F1M = self.spatial_attention(z_F1M)
",,terminal_output
4683,14710611,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(198)merge_inputs_wrapper()
-> out = f(*args)
",,terminal_output
4684,14711448,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py(153)split_inputs_wrapper()
-> pure_args_out, pure_out = f(*pure_args)
",,terminal_output
4685,14711642,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
",,terminal_output
4686,14712337,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py(2051)update_context_manager_wrapper()
-> return f(*args, **kwargs)
",,terminal_output
4687,14712876,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py(73)resolve_kwargs_wrapper()
-> return f(*args)
",,terminal_output
4688,14713394,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/utils/nn.py(443)__call__()
-> x_BTNM = block(x_BTNM, pos_index)
",,terminal_output
4689,14714886,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/genie.py(373)causal_step_fn()
-> final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
",,terminal_output
4690,14718664,"genie.py",0,0,"",python,tab
4691,14724714,"genie.py",14485,0,"",python,selection_command
4692,14724975,"genie.py",14502,0,"",python,selection_command
4693,14724986,"genie.py",14530,0,"",python,selection_command
4694,14725019,"genie.py",14618,0,"",python,selection_command
4695,14725055,"genie.py",14695,0,"",python,selection_command
4696,14725085,"genie.py",14761,0,"",python,selection_command
4697,14725121,"genie.py",14805,0,"",python,selection_command
4698,14725153,"genie.py",14848,0,"",python,selection_command
4699,14725186,"genie.py",14849,0,"",python,selection_command
4700,14725219,"genie.py",14896,0,"",python,selection_command
4701,14725253,"genie.py",14969,0,"",python,selection_command
4702,14725285,"genie.py",14970,0,"",python,selection_command
4703,14725320,"genie.py",15011,0,"",python,selection_command
4704,14725352,"genie.py",15090,0,"",python,selection_command
4705,14725385,"genie.py",15167,0,"",python,selection_command
4706,14725420,"genie.py",15246,0,"",python,selection_command
4707,14725452,"genie.py",15338,0,"",python,selection_command
4708,14725486,"genie.py",15427,0,"",python,selection_command
4709,14725523,"genie.py",15539,0,"",python,selection_command
4710,14725555,"genie.py",15611,0,"",python,selection_command
4711,14725585,"genie.py",15612,0,"",python,selection_command
4712,14725619,"genie.py",15668,0,"",python,selection_command
4713,14725791,"genie.py",15698,0,"",python,selection_command
4714,14725915,"genie.py",15668,0,"",python,selection_command
4715,14726076,"genie.py",15612,0,"",python,selection_command
4716,14726450,"genie.py",15611,0,"",python,selection_command
4717,14726623,"genie.py",15539,0,"",python,selection_command
4718,14726813,"genie.py",15427,0,"",python,selection_command
4719,14727002,"genie.py",15338,0,"",python,selection_command
4720,14727173,"genie.py",15246,0,"",python,selection_command
4721,14727347,"genie.py",15167,0,"",python,selection_command
4722,14727507,"genie.py",15090,0,"",python,selection_command
4723,14727663,"genie.py",15167,0,"",python,selection_command
4724,14727847,"genie.py",15246,0,"",python,selection_command
4725,14728112,"genie.py",15338,0,"",python,selection_command
4726,14728135,"genie.py",15427,0,"",python,selection_command
4727,14728166,"genie.py",15539,0,"",python,selection_command
4728,14728202,"genie.py",15611,0,"",python,selection_command
4729,14728242,"genie.py",15612,0,"",python,selection_command
4730,14728450,"genie.py",15611,0,"",python,selection_command
4731,14728585,"genie.py",15539,0,"",python,selection_command
4732,14728726,"genie.py",15427,0,"",python,selection_command
4733,14728944,"genie.py",15338,0,"",python,selection_command
4734,14729233,"genie.py",15427,0,"",python,selection_command
4735,14734265,"TERMINAL",0,0,"u",,terminal_output
4736,14734525,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(126)__call__()
-> out = self.f(*args, **kwargs)
",,terminal_output
4737,14736062,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(431)__call__()
-> pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
",,terminal_output
4738,14739223,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/genie.py(411)sample_causal()
-> final_carry_causal, _ = jax.lax.scan(
",,terminal_output
4739,14749608,"genie.py",17057,0,"",python,selection_mouse
4740,14752918,"TERMINAL",0,0,"steo",,terminal_output
4741,14753363,"TERMINAL",0,0,"h",,terminal_output
4742,14754065,"TERMINAL",0,0," ",,terminal_output
4743,14754251,"TERMINAL",0,0," ",,terminal_output
4744,14754670,"TERMINAL",0,0,"p",,terminal_output
4745,14754804,"TERMINAL",0,0,"t",,terminal_output
4746,14755169,"TERMINAL",0,0,"}",,terminal_output
4747,14755552,"TERMINAL",0,0," ",,terminal_output
4748,14755735,"TERMINAL",0,0," ",,terminal_output
4749,14756183,"TERMINAL",0,0,"-",,terminal_output
4750,14756635,"TERMINAL",0,0," ",,terminal_output
4751,14756964,"TERMINAL",0,0,"_",,terminal_output
4752,14757174,"TERMINAL",0,0,"t",,terminal_output
4753,14757418,"TERMINAL",0,0,"
(Pdb) 1
",,terminal_output
4754,14775416,"TERMINAL",0,0,"T",,terminal_output
4755,14775690,"TERMINAL",0,0,"
(Pdb) 1
",,terminal_output
4756,14786524,"TERMINAL",0,0,"d",,terminal_output
4757,14786619,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(431)__call__()
-> pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
",,terminal_output
4758,14789592,"genie.py",14848,0,"",python,selection_mouse
4759,14794760,"TERMINAL",0,0,"step",,terminal_output
4760,14795271,"TERMINAL",0,0,"_",,terminal_output
4761,14795616,"TERMINAL",0,0,"t",,terminal_output
4762,14795905,"TERMINAL",0,0,"
(Pdb) *** NameError: name 'step_t' is not defined
",,terminal_output
4763,14801198,"TERMINAL",0,0,"d",,terminal_output
4764,14801626,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py(126)__call__()
-> out = self.f(*args, **kwargs)
",,terminal_output
4765,14803092,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/genie.py(373)causal_step_fn()
-> final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
",,terminal_output
4766,14805742,"TERMINAL",0,0,"sp",,terminal_output
4767,14806303,"TERMINAL",0,0,"t",,terminal_output
4768,14806605,"TERMINAL",0,0," ",,terminal_output
4769,14806887,"TERMINAL",0,0," tep",,terminal_output
4770,14807242,"TERMINAL",0,0,"_",,terminal_output
4771,14807458,"TERMINAL",0,0,"t",,terminal_output
4772,14807681,"TERMINAL",0,0,"
(Pdb) Traced<int32[]>with<DynamicJaxprTrace>
",,terminal_output
4773,14951159,"TERMINAL",0,0,"e",,terminal_output
4774,14951711,"TERMINAL",0,0," ",,terminal_output
4775,14951840,"TERMINAL",0,0,"q",,terminal_output
4776,14952030,"TERMINAL",0,0,"uit",,terminal_output
4777,14952339,"TERMINAL",0,0,"(",,terminal_output
4778,14952439,"TERMINAL",0,0,")",,terminal_output
4779,14952756,"TERMINAL",0,0,"
(Pdb) Post mortem debugger finished. The /fast/home/franz.srambical/jafar/sample.py will be restarted
> /fast/home/franz.srambical/jafar/sample.py(1)<module>()
-> from dataclasses import dataclass
",,terminal_output
4780,14955048,"TERMINAL",0,0,"
",,terminal_output
4781,14955870,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.49 task 0: running
(Pdb) ",,terminal_output
4782,14956013,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.49
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T13:44:24.755] error: *** STEP 14571.49 ON hai005 CANCELLED AT 2025-08-01T13:44:24 DUE to SIGNAL Killed ***
",,terminal_output
4783,14956081,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4784,14958954,"genie.py",16411,0,"",python,selection_mouse
4785,14959845,"genie.py",16383,0,"
            ",python,content
4786,14960172,"genie.py",16396,0,"j",python,content
4787,14960172,"genie.py",16397,0,"",python,selection_keyboard
4788,14960189,"genie.py",16397,0,"a",python,content
4789,14960189,"genie.py",16398,0,"",python,selection_keyboard
4790,14960282,"genie.py",16398,0,"x",python,content
4791,14960283,"genie.py",16399,0,"",python,selection_keyboard
4792,14960368,"genie.py",16399,0,".",python,content
4793,14960368,"genie.py",16400,0,"",python,selection_keyboard
4794,14961465,"genie.py",16399,1,"",python,content
4795,14961603,"genie.py",16398,1,"",python,content
4796,14961751,"genie.py",16397,1,"",python,content
4797,14961900,"genie.py",16396,1,"",python,content
4798,14961920,"genie.py",16396,0,"b",python,content
4799,14961921,"genie.py",16397,0,"",python,selection_keyboard
4800,14962138,"genie.py",16397,0,"r",python,content
4801,14962138,"genie.py",16398,0,"",python,selection_keyboard
4802,14962187,"genie.py",16398,0,"e",python,content
4803,14962188,"genie.py",16399,0,"",python,selection_keyboard
4804,14962388,"genie.py",16399,0,"a",python,content
4805,14962389,"genie.py",16400,0,"",python,selection_keyboard
4806,14962419,"genie.py",16400,0,"k",python,content
4807,14962419,"genie.py",16401,0,"",python,selection_keyboard
4808,14962615,"genie.py",16401,0,"point()",python,content
4809,14962934,"genie.py",16407,0,"",python,selection_command
4810,14965299,"TERMINAL",0,0,"

[franz.srambical@hai005.haicore.berlin:~/jafar] $ bash experiments/sample.sh ",,terminal_output
4811,14965871,"TERMINAL",0,0,"

",,terminal_output
4812,14977100,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4813,14986102,"TERMINAL",0,0,"2025-08-01 13:44:54.849870: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4814,14990393,"TERMINAL",0,0,"2025-08-01 13:44:59.135733: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4815,14998728,"TERMINAL",0,0,"2025-08-01 13:45:07.462024: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4816,15002354,"TERMINAL",0,0,"jax.errors.SimplifiedTraceback: For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
Uncaught exception. Entering post mortem debugging
Running 'cont' or 'step' will restart the program
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py"", line 1723, in main
    pdb._runscript(mainpyfile)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/pdb.py"", line 1583, in _runscript
    self.run(statement)
  File ""/home/franz.srambical/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/bdb.py"", line 598, in run
    exec(cmd, globals, locals)
  File ""<string>"", line 1, in <module>
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 216, in <module>
    recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 180, in _autoreg_sample
    generated_vid_BSHWC = _sampling_fn(genie, batch)
  File ""/fast/home/franz.srambical/jafar/sample.py"", line 164, in _sampling_fn
    return model.sample_causal(
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 411, in sample_causal
    # start = time.time()
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 431, in __call__
    pure_args_out, pure_kwargs_out, pure_out = self.jitted_fn(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/compilation.py"", line 126, in __call__
    out = self.f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/genie.py"", line 373, in causal_step_fn
    final_logits_BTNp1V = dynamics_causal.transformer(vid_embed_BSNp1M, (step_t, step_n)) / temperature
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 443, in __call__
    x_BTNM = block(x_BTNM, pos_index)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/transforms.py"", line 73, in resolve_kwargs_wrapper
    return f(*args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/graph.py"", line 2051, in update_context_manager_wrapper
    return f(*args, **kwargs)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 153, in split_inputs_wrapper
    pure_args_out, pure_out = f(*pure_args)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/transforms/general.py"", line 198, in merge_inputs_wrapper
    out = f(*args)
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 323, in __call__
    z_F1M = self.spatial_attention(z_F1M)
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/flax/nnx/nn/attention.py"", line 605, in __call__
    x = self.attention_fn(
  File ""/fast/home/franz.srambical/jafar/utils/nn.py"", line 551, in attention_fn
    output_4d = jax.nn.dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/nn/functions.py"", line 1204, in dot_product_attention
    out = cudnn_dot_product_attention(
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 1981, in dot_product_attention
    check_layout(query, key, value, bias, q_seqlen, kv_seqlen, q_offsets, kv_offsets,
  File ""/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py"", line 332, in check_layout
    raise ValueError(
ValueError: Bias must have same seq length as QKV, got 4 and 4
> /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/jax/_src/cudnn/fused_attention_stablehlo.py(332)check_layout()
-> raise ValueError(
",,terminal_output
4817,15012299,"experiments/sample.sh",0,0,"",shellscript,tab
4818,15013127,"experiments/sample.sh",184,0,"",shellscript,selection_command
4819,15013401,"experiments/sample.sh",182,0,"",shellscript,selection_command
4820,15013864,"experiments/sample.sh",181,0,"",shellscript,selection_command
4821,15014121,"experiments/sample.sh",181,1,"-",shellscript,selection_command
4822,15014166,"experiments/sample.sh",181,2,"-c",shellscript,selection_command
4823,15014364,"experiments/sample.sh",181,11,"-c continue",shellscript,selection_command
4824,15014532,"experiments/sample.sh",181,18,"-c continue sample",shellscript,selection_command
4825,15014848,"experiments/sample.sh",181,17,"-c continue sampl",shellscript,selection_command
4826,15014998,"experiments/sample.sh",181,16,"-c continue samp",shellscript,selection_command
4827,15015304,"experiments/sample.sh",181,15,"-c continue sam",shellscript,selection_command
4828,15015439,"experiments/sample.sh",181,13,"-c continue s",shellscript,selection_command
4829,15015801,"experiments/sample.sh",181,4,"-c c",shellscript,selection_command
4830,15016016,"experiments/sample.sh",181,11,"-c continue",shellscript,selection_command
4831,15016340,"experiments/sample.sh",181,11,"",shellscript,content
4832,15016627,"experiments/sample.sh",181,1,"",shellscript,content
4833,15019686,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.50 task 0: running
^Csrun: sending Ctrl-C to StepId=14571.50
(Pdb) --KeyboardInterrupt--
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T13:45:28.433] error: *** STEP 14571.50 ON hai005 CANCELLED AT 2025-08-01T13:45:28 DUE to SIGNAL Killed ***
",,terminal_output
4834,15019830,"TERMINAL",0,0,"(Pdb) ^Csrun: sending Ctrl-C to StepId=14571.50
srun: job abort in progress
",,terminal_output
4835,15020050,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4836,15021468,"genie.py",0,0,"",python,tab
4837,15802971,"TERMINAL",0,0,"i",,terminal_output
4838,15803081,"TERMINAL",0,0,"p",,terminal_output
4839,15803214,"TERMINAL",0,0,"y",,terminal_output
4840,15803375,"TERMINAL",0,0,"th",,terminal_output
4841,15803475,"TERMINAL",0,0,"o",,terminal_output
4842,15803576,"TERMINAL",0,0,"n",,terminal_output
4843,15803846,"TERMINAL",0,0,"

",,terminal_output
4844,15805125,"TERMINAL",0,0,"Python 3.10.18 (main, Jul  1 2025, 18:35:05) [Clang 20.1.4 ]
Type 'copyright', 'credits' or 'license' for more information
IPython 8.37.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: ",,terminal_output
4845,15805183,"TERMINAL",0,0,"",,terminal_output
4846,15806635,"TERMINAL",0,0,"entity: str = """"",,terminal_output
4847,15806847,"TERMINAL",0,0,"x",,terminal_output
4848,15806932,"TERMINAL",0,0,"i",,terminal_output
4849,15807047,"TERMINAL",0,0,"t",,terminal_output
4850,15807263,"TERMINAL",0,0,"In [1]: exit

",,terminal_output
4851,15807321,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4852,15814074,"sample.py",0,0,"",python,tab
4853,15815677,"genie.py",0,0,"",python,tab
4854,15816356,"experiments/sample.sh",0,0,"",shellscript,tab
4855,15816897,"experiments/sample.sh",177,0,"",shellscript,selection_command
4856,15817062,"experiments/sample.sh",175,0,"",shellscript,selection_command
4857,15817211,"experiments/sample.sh",174,0,"",shellscript,selection_command
4858,15817519,"experiments/sample.sh",167,0,"",shellscript,selection_command
4859,15818018,"experiments/sample.sh",167,0,"i",shellscript,content
4860,15818018,"experiments/sample.sh",168,0,"",shellscript,selection_keyboard
4861,15818330,"experiments/sample.sh",167,0,"",shellscript,selection_command
4862,15818559,"experiments/sample.sh",175,0,"",shellscript,selection_command
4863,15818743,"experiments/sample.sh",176,0,"",shellscript,selection_command
4864,15820162,"experiments/sample.sh",176,1,"m",shellscript,selection_command
4865,15820263,"experiments/sample.sh",176,5,"m pdb",shellscript,selection_command
4866,15820479,"experiments/sample.sh",176,12,"m pdb sample",shellscript,selection_command
4867,15820917,"experiments/sample.sh",176,7,"m pdb s",shellscript,selection_command
4868,15821061,"experiments/sample.sh",176,3,"m p",shellscript,selection_command
4869,15821616,"experiments/sample.sh",176,5,"m pdb",shellscript,selection_command
4870,15823007,"experiments/sample.sh",180,0,"",shellscript,selection_command
4871,15824152,"experiments/sample.sh",178,0,"",shellscript,selection_command
4872,15825066,"experiments/sample.sh",182,0,"",shellscript,selection_command
4873,15825607,"experiments/sample.sh",178,0,"",shellscript,selection_command
4874,15825753,"experiments/sample.sh",176,0,"",shellscript,selection_command
4875,15826035,"experiments/sample.sh",180,0,"",shellscript,selection_command
4876,15826387,"experiments/sample.sh",180,1,"b",shellscript,selection_command
4877,15826626,"experiments/sample.sh",178,3,"pdb",shellscript,selection_command
4878,15826951,"experiments/sample.sh",176,5,"m pdb",shellscript,selection_command
4879,15827467,"experiments/sample.sh",175,6,"-m pdb",shellscript,selection_command
4880,15827863,"experiments/sample.sh",175,6,"",shellscript,content
4881,15830475,"experiments/sample.sh",174,0,"",shellscript,selection_command
4882,15830641,"experiments/sample.sh",174,1,"",shellscript,content
4883,15831326,"experiments/sample.sh",180,0,"",shellscript,selection_command
4884,15831520,"experiments/sample.sh",181,0,"",shellscript,selection_command
4885,15831893,"experiments/sample.sh",183,0,"",shellscript,selection_command
4886,15840756,"experiments/sample.sh",184,0,"",shellscript,selection_command
4887,15841247,"experiments/sample.sh",184,0," ",shellscript,content
4888,15841247,"experiments/sample.sh",185,0,"",shellscript,selection_keyboard
4889,15841907,"experiments/sample.sh",185,0,"0",shellscript,content
4890,15841907,"experiments/sample.sh",186,0,"",shellscript,selection_keyboard
4891,15842026,"experiments/sample.sh",186,0,"0",shellscript,content
4892,15842026,"experiments/sample.sh",187,0,"",shellscript,selection_keyboard
4893,15842516,"experiments/sample.sh",186,1,"",shellscript,content
4894,15842697,"experiments/sample.sh",185,1,"",shellscript,content
4895,15842936,"experiments/sample.sh",185,0,"-",shellscript,content
4896,15842937,"experiments/sample.sh",186,0,"",shellscript,selection_keyboard
4897,15843037,"experiments/sample.sh",186,0,"-",shellscript,content
4898,15843037,"experiments/sample.sh",187,0,"",shellscript,selection_keyboard
4899,15843494,"experiments/sample.sh",186,0,"",shellscript,selection_command
4900,15844166,"experiments/sample.sh",214,0,"",shellscript,selection_command
4901,15844555,"experiments/sample.sh",213,0,"",shellscript,selection_command
4902,15844703,"experiments/sample.sh",207,0,"",shellscript,selection_command
4903,15844917,"experiments/sample.sh",206,0,"",shellscript,selection_command
4904,15845054,"experiments/sample.sh",196,0,"",shellscript,selection_command
4905,15845178,"experiments/sample.sh",194,0,"",shellscript,selection_command
4906,15847054,"experiments/sample.sh",194,1,"-",shellscript,selection_command
4907,15847477,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4908,15847580,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4909,15847912,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4910,15848042,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4911,15848185,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4912,15848343,"experiments/sample.sh",194,2,"--",shellscript,selection_command
4913,15848722,"experiments/sample.sh",311,2,"",shellscript,content
4914,15848722,"experiments/sample.sh",282,2,"",shellscript,content
4915,15848722,"experiments/sample.sh",260,2,"",shellscript,content
4916,15848722,"experiments/sample.sh",242,2,"",shellscript,content
4917,15848722,"experiments/sample.sh",221,2,"",shellscript,content
4918,15848722,"experiments/sample.sh",194,2,"",shellscript,content
4919,15848724,"experiments/sample.sh",194,0,"",shellscript,selection_command
4920,15857629,"experiments/sample.sh",301,0,"--",shellscript,content
4921,15857635,"experiments/sample.sh",274,0,"--",shellscript,content
4922,15857635,"experiments/sample.sh",254,0,"--",shellscript,content
4923,15857635,"experiments/sample.sh",238,0,"--",shellscript,content
4924,15857641,"experiments/sample.sh",219,0,"--",shellscript,content
4925,15857642,"experiments/sample.sh",194,0,"--",shellscript,content
4926,15857643,"experiments/sample.sh",194,0,"",shellscript,selection_command
4927,15859563,"TERMINAL",0,0,"ipython",,terminal_output
4928,15860012,"TERMINAL",0,0,"

",,terminal_output
4929,15860984,"TERMINAL",0,0,"Python 3.10.18 (main, Jul  1 2025, 18:35:05) [Clang 20.1.4 ]
Type 'copyright', 'credits' or 'license' for more information
IPython 8.37.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: entity: str = """"",,terminal_output
4930,15861175,"TERMINAL",0,0,"x",,terminal_output
4931,15861264,"TERMINAL",0,0,"i",,terminal_output
4932,15861376,"TERMINAL",0,0,"t",,terminal_output
4933,15861584,"TERMINAL",0,0,"In [1]: exit

",,terminal_output
4934,15861647,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4935,15862128,"TERMINAL",0,0,"ipythonbash experiments/sample.sh ",,terminal_output
4936,15862818,"TERMINAL",0,0,"

",,terminal_output
4937,15874339,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4938,15878200,"TERMINAL",0,0,"^Csrun: interrupt (one more within 1 sec to abort)
srun: StepId=14571.51 task 0: running
",,terminal_output
4939,15878278,"TERMINAL",0,0,"^Csrun: sending Ctrl-C to StepId=14571.51
srun: forcing job termination
srun: Job step aborted: Waiting up to 32 seconds for job step to finish.
[2025-08-01T13:59:47.026] error: *** STEP 14571.51 ON hai005 CANCELLED AT 2025-08-01T13:59:47 DUE to SIGNAL Killed ***
",,terminal_output
4940,15879306,"TERMINAL",0,0,"[franz.srambical@hai005.haicore.berlin:~/jafar] $ ",,terminal_output
4941,15879549,"experiments/sample.sh",166,0,"",shellscript,selection_command
4942,15879677,"experiments/sample.sh",173,0,"",shellscript,selection_command
4943,15880702,"experiments/sample.sh",174,0,"",shellscript,selection_command
4944,15880920,"experiments/sample.sh",174,0," ",shellscript,content
4945,15880921,"experiments/sample.sh",175,0,"",shellscript,selection_keyboard
4946,15881220,"experiments/sample.sh",175,0,"-",shellscript,content
4947,15881220,"experiments/sample.sh",176,0,"",shellscript,selection_keyboard
4948,15881376,"experiments/sample.sh",176,0,"-",shellscript,content
4949,15881376,"experiments/sample.sh",177,0,"",shellscript,selection_keyboard
4950,15882025,"experiments/sample.sh",177,0,"p",shellscript,content
4951,15882025,"experiments/sample.sh",178,0,"",shellscript,selection_keyboard
4952,15882181,"experiments/sample.sh",178,0,"d",shellscript,content
4953,15882182,"experiments/sample.sh",179,0,"",shellscript,selection_keyboard
4954,15882281,"experiments/sample.sh",179,0,"b",shellscript,content
4955,15882281,"experiments/sample.sh",180,0,"",shellscript,selection_keyboard
4956,15882573,"experiments/sample.sh",179,0,"",shellscript,selection_command
4957,15884326,"TERMINAL",0,0,"bash experiments/sample.sh ",,terminal_output
4958,15884552,"TERMINAL",0,0,"

",,terminal_output
4959,15895942,"TERMINAL",0,0,"/fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py:1256: UserWarning: Sharding info not provided when restoring. Populating sharding info from sharding file. Please note restoration time will be slightly increased due to reading from file. Note also that this option is unsafe when restoring on a different topology than the checkpoint was saved with.
  warnings.warn(
",,terminal_output
4960,15905013,"TERMINAL",0,0,"2025-08-01 14:00:13.754927: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4961,15909333,"TERMINAL",0,0,"2025-08-01 14:00:18.077159: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4962,15917460,"TERMINAL",0,0,"2025-08-01 14:00:26.196420: W external/xla/xla/service/gpu/autotuning/dot_search_space.cc:200] All configs were filtered out because none of them sufficiently match the hints. Maybe the hints set does not contain a good representative set of valid configs?Working around this by using the full hints set instead.
",,terminal_output
4963,15921033,"TERMINAL",0,0,"> /fast/home/franz.srambical/jafar/genie.py(395)sample_causal()
-> rng, step_rng = jax.random.split(rng)
",,terminal_output
4964,15938108,"TERMINAL",0,0,"w",,terminal_output
4965,15938292,"TERMINAL",0,0,"
(Pdb)   /fast/home/franz.srambical/jafar/.venv/bin/ipython(10)<module>()
-> sys.exit(start_ipython())
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/__init__.py(130)start_ipython()
-> return launch_new_instance(argv=argv, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/traitlets/config/application.py(1074)launch_instance()
-> app.initialize(argv)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/traitlets/config/application.py(118)inner()
-> return method(app, *args, **kwargs)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/terminal/ipapp.py(284)initialize()
-> self.init_code()
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/core/shellapp.py(353)init_code()
-> self._run_cmd_line_code()
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/core/shellapp.py(478)_run_cmd_line_code()
-> self._exec_file(fname, shell_futures=True)
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/core/shellapp.py(403)_exec_file()
-> self.shell.safe_execfile(full_filename,
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/core/interactiveshell.py(2934)safe_execfile()
-> py3compat.execfile(
  /fast/home/franz.srambical/jafar/.venv/lib/python3.10/site-packages/IPython/utils/py3compat.py(55)execfile()
-> exec(compiler(f.read(), fname, ""exec""), glob, loc)
  /fast/home/franz.srambical/jafar/sample.py(216)<module>()
-> recon_video_BSHWC = _autoreg_sample(genie, rng, video_batch_BSHWC, action_batch_E)
  /fast/home/franz.srambical/jafar/sample.py(180)_autoreg_sample()
-> generated_vid_BSHWC = _sampling_fn(genie, batch)
  /fast/home/franz.srambical/jafar/sample.py(164)_sampling_fn()
-> return model.sample_causal(
> /fast/home/franz.srambical/jafar/genie.py(395)sample_causal()
-> rng, step_rng = jax.random.split(rng)
",,terminal_output
4966,16010651,"TERMINAL",0,0,"u",,terminal_output
4967,16010893,"TERMINAL",0,0,"
(Pdb) > /fast/home/franz.srambical/jafar/sample.py(164)_sampling_fn()
-> return model.sample_causal(
",,terminal_output
4968,16014053,"TERMINAL",0,0,"m",,terminal_output
4969,16014588,"TERMINAL",0,0," ",,terminal_output
4970,16014900,"TERMINAL",0,0,"n",,terminal_output
4971,16015022,"TERMINAL",0,0,"n",,terminal_output
4972,16015101,"TERMINAL",0,0,"x",,terminal_output
4973,16015220,"TERMINAL",0,0,".",,terminal_output
4974,16016074,"TERMINAL",0,0,"	",,terminal_output
4975,16016801,"TERMINAL",0,0,"",,terminal_output
4976,16016968,"TERMINAL",0,0," ",,terminal_output
4977,16017171,"TERMINAL",0,0," ",,terminal_output
4978,16017323,"TERMINAL",0,0," ",,terminal_output
4979,16017474,"TERMINAL",0,0," ",,terminal_output
4980,16017787,"TERMINAL",0,0,"n",,terminal_output
4981,16017973,"TERMINAL",0,0,"n",,terminal_output
4982,16018103,"TERMINAL",0,0,"x",,terminal_output
4983,16018237,"TERMINAL",0,0,".,",,terminal_output
4984,16018831,"TERMINAL",0,0," ",,terminal_output
4985,16019784,"TERMINAL",0,0,"di",,terminal_output
4986,16019959,"TERMINAL",0,0,"sp",,terminal_output
4987,16020260,"TERMINAL",0,0,"l",,terminal_output
4988,16020366,"TERMINAL",0,0,"ay",,terminal_output
4989,16020802,"TERMINAL",0,0,"(",,terminal_output
4990,16021262,"TERMINAL",0,0,"m",,terminal_output
4991,16021362,"TERMINAL",0,0,"o",,terminal_output
4992,16021527,"TERMINAL",0,0,"d",,terminal_output
4993,16021592,"TERMINAL",0,0,"e",,terminal_output
4994,16021729,"TERMINAL",0,0,"l",,terminal_output
4995,16022211,"TERMINAL",0,0,"0",,terminal_output
4996,16023116,"TERMINAL",0,0," ",,terminal_output
4997,16023244,"TERMINAL",0,0,")",,terminal_output
4998,16023723,"TERMINAL",0,0,"
",,terminal_output
4999,16025483,"TERMINAL",0,0,"(Pdb) <IPython.core.display.HTML object>
",,terminal_output
5000,16036839,"TERMINAL",0,0,"<IPython.core.display.HTML object>
",,terminal_output
5001,16237376,"experiments/sample.sh",0,0,"",shellscript,tab
5002,16342448,"test.ipynb",0,0,"nnx.display(nnx.eval_shape(lambda: restored))",python,tab
5003,16345160,"experiments/sample.sh",0,0,"",shellscript,tab
5004,16356870,"test.ipynb",0,0,"",python,tab
5005,16362706,"test.ipynb",0,0,"from flax import nnx

class MyModel(nnx.Module):
  """"""A simple model containing a Linear layer and dropout.""""""

  def __init__(self, use_bias: bool, dropout: float, *, rngs: nnx.Rngs):
    self.linear = nnx.Linear(10, 20, use_bias=use_bias, rngs=rngs)
    self.dropout = nnx.Dropout(dropout, rngs=rngs)

  def __call__(self, x):
    x = self.linear(x)
    x = self.dropout(x)
    return x",python,tab
5006,16362706,"test.ipynb",303,0,"",python,selection_mouse
5007,16363660,"test.ipynb",0,0,"import jax 
import jax.numpy as jnp
import orbax.checkpoint as ocp
from flax import nnx
from genie import Genie
import optax

from utils.lr_utils import get_lr_schedule


num_steps: int = 200_000
seed: int = 0
seq_len: int = 16
image_channels: int = 3
image_height: int = 90
image_width: int = 160
data_dir: str = """"
save_ckpt: bool = False
restore_ckpt: bool = False
# Optimization
batch_size: int = 36
init_lr: float = 0.0
max_lr: float = 3e-5
decay_end: float = 0.0
wsd_decay_steps: int = (
    10000  # NOTE: wsd_decay_steps will only be used when using a wsd-schedule
)
warmup_steps: int = 5000
lr_schedulee: str = ""wsd""  # supported options: wsd, cos
# Tokenizer
tokenizer_dim: int = 512
tokenizer_ffn_dim: int = 2048
latent_patch_dim: int = 32
num_patch_latents: int = 1024
patch_size: int = 4
tokenizer_num_blocks: int = 4
tokenizer_num_heads: int = 8
tokenizer_checkpoint: str = """"
# LAM
lam_dim: int = 512
lam_ffn_dim: int = 2048
latent_action_dim: int = 32
num_latent_actions: int = 6
lam_patch_size: int = 16
lam_num_blocks: int = 4
lam_num_heads: int = 8
lam_checkpoint: str = """"
# Dynamics
dyna_dim: int = 512
dyna_ffn_dim: int = 2048
dyna_num_blocks: int = 6
dyna_num_heads: int = 8
dropout: float = 0.0
mask_limit: float = 0.5
param_dtype = jnp.float32
dtype = jnp.bfloat16
use_flash_attention: bool = True
# Logging
log: bool = False
entity: str = """"
project: str = """"
name: str = ""train_dynamics""
log_interval: int = 5
log_image_interval: int = 250
ckpt_dir: str = ""/home/franz.srambical/jafar/checkpoints/dynamics_openai_grain_tok_lam_restore/000030/model_state""
log_checkpoint_interval: int = 25000
log_checkpoint_keep_period: int = 20000
log_gradients: bool = False
wandb_id: str = """"

num_devices = jax.device_count()
if num_devices == 0:
    raise ValueError(""No JAX devices found."")
print(f""Running on {num_devices} devices."")

rng = jax.random.PRNGKey(0)

# --- Initialize model ---
rng, _rng = jax.random.split(rng)
rngs = nnx.Rngs(_rng)
genie = Genie(
    # Tokenizer
    in_dim=image_channels,
    tokenizer_dim=tokenizer_dim,
    tokenizer_ffn_dim=tokenizer_ffn_dim,
    latent_patch_dim=latent_patch_dim,
    num_patch_latents=num_patch_latents,
    patch_size=patch_size,
    tokenizer_num_blocks=tokenizer_num_blocks,
    tokenizer_num_heads=tokenizer_num_heads,
    # LAM
    lam_dim=lam_dim,
    lam_ffn_dim=lam_ffn_dim,
    latent_action_dim=latent_action_dim,
    num_latent_actions=num_latent_actions,
    lam_patch_size=lam_patch_size,
    lam_num_blocks=lam_num_blocks,
    lam_num_heads=lam_num_heads,
    lam_co_train=not lam_checkpoint,
    # Dynamics
    dyna_dim=dyna_dim,
    dyna_ffn_dim=dyna_ffn_dim,
    dyna_num_blocks=dyna_num_blocks,
    dyna_num_heads=dyna_num_heads,
    dropout=dropout,
    mask_limit=mask_limit,
    param_dtype=param_dtype,
    dtype=dtype,
    use_flash_attention=use_flash_attention,
    rngs=rngs,
)
lr_schedule = get_lr_schedule(
    lr_schedulee,
    init_lr,
    max_lr,
    decay_end,
    num_steps,
    warmup_steps,
    wsd_decay_steps,
)
tx = optax.adamw(
    learning_rate=lr_schedule,
    b1=0.9,
    b2=0.9,
    weight_decay=1e-4,
    mu_dtype=dtype,
)
optimizer = nnx.Optimizer(genie, tx)

# optimizer.model.tokenizer.vq.drop = nnx.Dropout(0.1, rngs=rngs)
abstract_optimizer = nnx.eval_shape(lambda: optimizer)
abstract_optimizer_state = nnx.state(abstract_optimizer)

checkpointer = ocp.PyTreeCheckpointer()
restored = checkpointer.restore(ckpt_dir)",python,tab
5008,16364245,"test.ipynb",0,0,"",python,tab
5009,16364944,"test.ipynb",0,0,"# nnx.state(nnx.eval_shape(lambda: optimizer.model.tokenizer.vq)) # <- works
# nnx.state(nnx.eval_shape(lambda: optimizer.model.tokenizer)).vq) # <- works
# nnx.state(nnx.eval_shape(lambda: optimizer.model)).tokenizer.vq)
# nnx.state(nnx.eval_shape(lambda: optimizer)).model.tokenizer.vq)

# jax.tree.map(ocp.utils.to_shape_dtype_struct, nnx.state(optimizer.model.tokenizer.vq)) # <- works
# jax.tree.map(ocp.utils.to_shape_dtype_struct, nnx.state(optimizer.model.tokenizer)).vq # <- works
# jax.tree.map(ocp.utils.to_shape_dtype_struct, nnx.state(optimizer.model)).tokenizer.vq
# jax.tree.map(ocp.utils.to_shape_dtype_struct, nnx.state(optimizer)).model.tokenizer.vq",python,tab