jax-unirep
A performant JAX reimplementation of the UniRep model for generating protein sequence representations.
What is jax-unirep?
jax-unirep is a Python library that reimplements the UniRep model, a deep learning model for generating numerical representations (embeddings) of protein sequences. It solves the problem of efficiently computing these representations for downstream tasks in protein engineering and analysis, using JAX for accelerated performance.
Computational biologists, bioinformaticians, and researchers in protein engineering who need high-performance tools for protein sequence featurization and machine learning workflows.
Developers choose jax-unirep for its performance gains from JAX, its self-contained and easily customizable design, and its utility APIs tailored for protein engineering, offering a modern alternative to the original implementation.
Overview
Reimplementation of the UniRep protein featurization model.
Use Cases
Best For
- Generating protein sequence embeddings for machine learning models
- Accelerating protein engineering research with GPU support
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
