nickrb commited on
Commit
d6bc67d
Β·
verified Β·
1 Parent(s): 72f4075

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +91 -0
README.md ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: jax
4
+ tags:
5
+ - protein-structure-prediction
6
+ - alphafold3
7
+ - jax
8
+ - equinox
9
+ - biology
10
+ ---
11
+
12
+ # Protenij β€” JAX/Equinox weights for Protenix
13
+
14
+ This repository hosts JAX/Equinox-converted model weights (and a mirror of the
15
+ original PyTorch `protenix-v2` checkpoint) for use with
16
+ [protenij](https://github.com/escalante-bio/protenij), a JAX/Equinox translation
17
+ of [Protenix](https://github.com/bytedance/Protenix), ByteDance's implementation
18
+ of the AlphaFold 3 architecture.
19
+
20
+ The JAX/Equinox weights are format conversions of the original PyTorch
21
+ checkpoints released by ByteDance β€” the underlying model parameters are
22
+ numerically identical, only the serialization format has changed (PyTorch `.pt`
23
+ β†’ Equinox `.eqx` + pickled skeleton).
24
+
25
+ ## Files
26
+
27
+ | File | Format | Size | Source |
28
+ | --- | --- | --- | --- |
29
+ | `protenix-v2.eqx` / `protenix-v2.skeleton.pkl` | Equinox | 1.86 GB | Converted from `protenix-v2.pt` |
30
+ | `protenix-v2.pt` | PyTorch | 1.86 GB | Mirror of upstream ByteDance release |
31
+ | `protenix_base_default_v1.0.0.eqx` / `.skeleton.pkl` | Equinox | β€” | Converted from upstream |
32
+ | `protenix_base_20250630_v1.0.0.eqx` / `.skeleton.pkl` | Equinox | β€” | Converted from upstream |
33
+ | `protenix_mini_default_v0.5.0.eqx` / `.skeleton.pkl` | Equinox | β€” | Converted from upstream |
34
+ | `protenix_tiny_default_v0.5.0.eqx` / `.skeleton.pkl` | Equinox | β€” | Converted from upstream |
35
+ | `components.v20240608.cif` | Data | β€” | CCD chemical components (upstream) |
36
+ | `components.v20240608.cif.rdkit_mol.pkl` | Data | β€” | CCD rdkit mol cache (upstream) |
37
+ | `clusters-by-entity-40.txt` | Data | β€” | PDB entity-40 clusters (upstream) |
38
+
39
+ ## Usage
40
+
41
+ ```python
42
+ from protenix.backend import load_model
43
+ model = load_model("protenix-v2") # auto-downloads from this repo
44
+ ```
45
+
46
+ ## License and attribution
47
+
48
+ Released under the **Apache License 2.0**, matching the upstream
49
+ [bytedance/Protenix](https://github.com/bytedance/Protenix) project.
50
+
51
+ The upstream Protenix README explicitly states:
52
+
53
+ > "The Protenix project including both code and model parameters is released
54
+ > under the Apache 2.0 License. It is free for both academic research and
55
+ > commercial use."
56
+
57
+ ### Modification notice (Apache 2.0 Β§4(b))
58
+
59
+ The `.eqx` and `.skeleton.pkl` files in this repository are **format
60
+ conversions** of the original PyTorch checkpoints released by ByteDance. The
61
+ PyTorch state dicts were loaded and the tensors re-serialized in Equinox format
62
+ using
63
+ [`protenix/backend.py`](https://github.com/escalante-bio/protenij/blob/main/protenix/backend.py)
64
+ and
65
+ [`translate_models.py`](https://github.com/escalante-bio/protenij/blob/main/translate_models.py).
66
+ No weights were retrained, fine-tuned, or otherwise numerically modified.
67
+
68
+ The `protenix-v2.pt` file in this repository is a bit-for-bit mirror of the
69
+ original PyTorch checkpoint hosted at
70
+ `https://protenix.tos-cn-beijing.volces.com/checkpoint/protenix-v2.pt` (mirrored
71
+ here after the upstream URL became unreachable).
72
+
73
+ ### Copyright notice (Apache 2.0 Β§4(c))
74
+
75
+ Copyright 2024 ByteDance and/or its affiliates. The original Protenix code and
76
+ model parameters were released under Apache License 2.0. See the `LICENSE` file
77
+ in this repository for the full license text.
78
+
79
+ ### Citations
80
+
81
+ If you use these weights, please cite the original Protenix work:
82
+
83
+ - Protenix repository: https://github.com/bytedance/Protenix
84
+ - Protenix technical reports in `docs/` of the upstream repository
85
+
86
+ ## Disclaimer
87
+
88
+ These files are provided as-is. The weights are format conversions only β€” for
89
+ the authoritative source and for training code, model cards, and technical
90
+ reports, refer to the [upstream ByteDance Protenix
91
+ repository](https://github.com/bytedance/Protenix).