Upload README.md with huggingface_hub
Browse files
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).
|