escalante-bio
/

esmj

Model card Files Files and versions

1.33 GB

Ctrl+K

Ctrl+K

1 contributor

History: 3 commits

nickrb's picture

Regenerate skeleton for jax 0.10.0 (ShapeDtypeStruct.vma)

ad87798 verified 17 days ago

.gitattributes

1.57 kB
Upload folder using huggingface_hub about 2 months ago
esmc_300m.eqx

1.33 GB
xet

Upload folder using huggingface_hub about 2 months ago
esmc_300m.skeleton.pkl
Detected Pickle imports (14)
- "esmj.model.Sequential",
- "esmj.model.MultiHeadAttention",
- "esmj.model.LayerNorm",
- "esmj.model.TransformerStack",
- "esmj.model.UnifiedTransformerBlock",
- "jax._src.nn.functions.gelu",
- "numpy.dtype",
- "jax.ShapeDtypeStruct",
- "esmj.model.swiglu",
- "esmj.model.ESMC",
- "esmj.model.SparseEmbedding",
- "esmj.model.Linear",
- "equinox.nn._embedding.Embedding",
- "esmj.model.RotaryEmbedding"
How to fix it?
2.3 kB
xet

Regenerate skeleton for jax 0.10.0 (ShapeDtypeStruct.vma) 17 days ago