yiqingchen's picture
Fix tensor_embedding.distance_proj for stock (PyG) matgl: split the fused Warp distance_proj [3*units] into distance_proj1/2/3 [units] (exact reshape). The fused form silently random-initialized on non-Warp matgl (load_state_dict strict=False), breaking the model; the split loads cleanly on PyG matgl and merges back under Warp. Si bulk modulus restored to the DFT reference.
0e4ef64 verified