Search results

chore: Fix error on flax 0.12.0

Status: Open.

#228 In p-doom/jasmine;

· emergenz opened on Oct 19, 2025

Remove redunant <code>latent_action_dim</code> Arg from <code>train_dynamics.py</code>

Status: Open.

#226 In p-doom/jasmine;

· emergenz opened on Oct 8, 2025

Conditional mixed precision via Args

Status: Open.

#225 In p-doom/jasmine;

· emergenz opened on Oct 4, 2025

Add <code>restore_steps</code> to training script <code>Args</code>

Status: Open.

#224 In p-doom/jasmine;

· emergenz opened on Oct 4, 2025

Potential redundant slicing

Status: Open.

#222 In p-doom/jasmine;

· emergenz opened on Oct 2, 2025

Check if re-init leads to performance degradation

Status: Open.

#194 In p-doom/jasmine;

· maharajamihir opened on Sep 17, 2025

Refactor sampling to use model forward pass

Status: Open.

#193 In p-doom/jasmine;

· emergenz opened on Sep 16, 2025

Activate KV caching

Status: Open.

#192 In p-doom/jasmine;

· emergenz opened on Sep 16, 2025

Change <code>decode</code> to <code>use_kv_cache</code> in <code>genie.py</code>

Status: Open.

#191 In p-doom/jasmine;

· emergenz opened on Sep 16, 2025

Swich to GPT-2-style weight initialization

Status: Open.

#186 In p-doom/jasmine;

· emergenz opened on Sep 14, 2025

Implement rotation trick STE

Status: Open.

#180 In p-doom/jasmine;

· emergenz opened on Sep 6, 2025

Checkpoint restore for dynamics model fails without cotraining

Status: Open.

#179 In p-doom/jasmine;

· maharajamihir opened on Sep 5, 2025