train_simulate.py File Reference

Go to the source code of this file.

Classes
class	aquin.compute.train_simulate.SimulateRequest
class	aquin.compute.train_simulate.CompareRequest

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.train_simulate

Functions
bool	aquin.compute.train_simulate._is_heavy_sim_model (dict[str, Any] cfg)
int	aquin.compute.train_simulate._sim_batch_limit (dict[str, Any] cfg)
list[torch.Tensor]	aquin.compute.train_simulate._grad_vector (torch.Tensor loss, list[torch.nn.Parameter] params, bool retain_graph=False)
list[torch.Tensor]	aquin.compute.train_simulate._hvp (torch.Tensor loss, list[torch.nn.Parameter] params, list[torch.Tensor] v)
list[torch.Tensor]	aquin.compute.train_simulate._hvp_from_fn (loss_fn, list[torch.nn.Parameter] params, list[torch.Tensor] v)
float	aquin.compute.train_simulate._grad_dot (list[torch.Tensor] ga, list[torch.Tensor] gb)
float	aquin.compute.train_simulate._tensor_list_norm (list[torch.Tensor] ts)
float	aquin.compute.train_simulate._normalized_influence (list[torch.Tensor] ga, list[torch.Tensor] gb)
list[torch.Tensor]	aquin.compute.train_simulate._lissa_inverse_hvp (test_loss_fn, train_loss_fn, int n_train, list[torch.nn.Parameter] params, float scale=25.0, float damping=0.05, int n_iter=10)
list[dict]	aquin.compute.train_simulate._influence_via_grad_dot (test_loss_fn, list[int] train_indices, list[dict] rows, list[torch.nn.Parameter] params, train_loss_fn)
dict[int, float]	aquin.compute.train_simulate._ntk_diagonal (torch.Tensor loss, list[torch.nn.Parameter] params, list[torch.Tensor] grads)
float	aquin.compute.train_simulate._power_iteration_max_eigenvalue (torch.Tensor loss, list[torch.nn.Parameter] params, int n_iter=3)
int	aquin.compute.train_simulate._hidden_state_index (model, int sae_layer)
torch.Tensor|None	aquin.compute.train_simulate._sae_grad_scores_from_batch (model, sae, int sae_layer, torch.Tensor input_ids, torch.Tensor attention_mask, torch.Tensor|None base_acts_mean)
dict[str, Any]	aquin.compute.train_simulate._run_dataset_quality (list[dict] rows)
dict	aquin.compute.train_simulate._detect_ai_generated (list[dict] rows)
None	aquin.compute.train_simulate._run_simulation (SimulateRequest req, asyncio.Queue queue, asyncio.AbstractEventLoop loop)
	aquin.compute.train_simulate.training_simulate (SimulateRequest req)
Any	aquin.compute.train_simulate._sanitize (Any obj)
dict	aquin.compute.train_simulate.compare_simulation_results (dict result_a, dict result_b, str label_a="Run A", str label_b="Run B", *, str run_id_a="", str run_id_b="")
	aquin.compute.train_simulate.compare_simulations (CompareRequest req)

Variables
	aquin.compute.train_simulate.router = APIRouter()
	aquin.compute.train_simulate.DEVICE = resolve_compute_device()
	aquin.compute.train_simulate.DTYPE = default_dtype_for_device(DEVICE)
int	aquin.compute.train_simulate._SIM_MAX_SAMPLES = 64
str	aquin.compute.train_simulate._SIM_FT_CKPT_PATH = None
str	aquin.compute.train_simulate._SIM_FT_MODEL_ID = None
list	aquin.compute.train_simulate._HARMFUL_KEYWORDS
list	aquin.compute.train_simulate._AI_FINGERPRINTS