maedoc

function sa()
{
    name=$1
    if [[ ! -d ~/env/$name ]]
    then
        mkdir -p ~/env
        uv venv ~/env/$name
    fi
    source ~/env/$name/bin/activate
}

maedoc

import numpy as np, os, pickle, tqdm, jax, jax.numpy as jp

# setup
key = jax.random.PRNGKey(42)
batch_size = 1
nl, n_head, vocab_size = 2, 2, 65
n_embd = n_head * 8
nh, hs = n_head, n_embd//n_head
B, C = batch_size, n_embd

maedoc

maedoc

// iChannel0 is buffer A
// iChannel1 is some texture to use for weights

float rand()
{
    return 0.0;
}

void mainImage( out vec4 fragColor, in vec2 fragCoord )
{

maedoc

#!/bin/bash

set -eu

# allow scripting password, pass as env var MDP
MDP=${MDP:-""}

# setup some vars
export u=$(whoami)
export kt=$HOME/.my-keytab

maedoc

maedoc

import os
import time
import s3fs

fs = s3fs.S3FileSystem(
    endpoint_url="http://10.67.123.1:9000",
    key="adni-user",
    secret="12345679"
)

maedoc

bistable:
        movaps  xmm3, xmm0
        movaps  xmm1, xmm0
        mulss   xmm3, xmm0
        mulss   xmm1, xmm3
        divss   xmm1, DWORD PTR .LC0[rip]
        addss   xmm1, xmm0
        movaps  xmm0, xmm1
        mulss   xmm0, xmm1
        addss   xmm0, xmm3

maedoc

#include<stdbool.h>
#include<stdio.h>

struct sim {
    const int rng_seed;

    const int num_item;
    const int num_node;
    const int num_svar;
    const int num_time;

maedoc

def heun(y, t, dt):
    d1 = dfun(y, t)
    d2 = dfun(y + dt*d1, t + dt)
    err = np.mean(np.abs((d1 - d2)))#/(1e-9+d2)))
    return y + dt/2*(d1 + d2), err

def solve_adapt1(y0, ts, tol):
    ys = [y0]
    max_dt = dt = ts[1] - ts[0]
    t0 = ts[0]