jax-unirep Flexible output sizes

Flexible output sizes for jax-unirep

PR: https://github.com/ElArkk/jax-unirep/pull/80

Goals:

  • We want flexible number of dimensions for outputs. This should help with training models on large amounts of data.
  • A slightly more functional API for the fit function.
  • Update docs.
Pages that link here

jax-unirep
Key Information Repo: https://github

jax-unirep

Key Information

Open PRs

Paper

  • Next place to submit to: PeerJ Computer Science. (see: PeerJ CS submission guidelines)
  • Other locations in consideration: JCIM Application Notes
  • Rejected at:
    • Bioinformatics
    • JMLR-MLOSS

The main uses of jax-unirep

The main use of jax-unirep really is in getting "standardized" representations of proteins that can be used for downstream ML applications easily.