---
license: apache-2.0
tags:
  - molecule-generation
  - chemistry
  - smiles
  - drug-discovery
---

# RiemannGen checkpoints

Trained checkpoints for the RiemannGen project: two from-scratch
MolMIM-style latent sequence models.

- `atom/final-atom.pt` + `atom/vocab_new.txt` -- plain-SMILES model.
- `fragment/final.pt` + `fragment/vocab_new.txt` -- SAFE-fragment-encoded
  model, for scaffold growth / local fragment editing.

Licensed under Apache License 2.0 -- free to use, modify, and redistribute.

## Citation

If you use these checkpoints or the RiemannGen codebase, please cite:

```
@misc{riemanngen2026,
  title  = {RiemannGen: latent-space molecule generation with atom and fragment backends},
  author = {XeonChem},
  year   = {2026},
  url    = {https://huggingface.co/XeonChem/riemanngen}
}
```