crowsonkb

"""Implementation of the syre weight decay algorithm from "Remove Symmetries to Control Model
Expressivity and Improve Optimization" (https://arxiv.org/abs/2408.15495)."""

import math

import torch
from torch import optim
import triton
import triton.language as tl

travisdoesmath

let test = new TuplePendulum({N: 4, thetas:[1, 2, 3], omegas:[0.1,0.2,0.3]})

var nPendulums = 5

var pendulums = d3.range(nPendulums).map(x => new TuplePendulum({N: 2**x, thetas: [0.5 * Math.PI]}))

var fadeBackground = false;

var svg = d3.select("svg")
	width = +svg.attr("width"),