Skip to content

Prediction of AAV capsid fitness from VP1 protein sequence; written in Jax with PyTorch dataloaders

Notifications You must be signed in to change notification settings

AnnabelLarge/AAV-regression-in-Jax

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

To Do

  • (DONE) dataloaders
    • (DONE) supportive utils
  • (DONE) loss functions
  • basic models
    • NEW: mamba
  • NEW: model initializers
  • train_one_batch
  • eval_one_batch
  • training wrapper
  • eval wrapper
  • NEW: CLI + JSON interface

About

Prediction of AAV capsid fitness from VP1 protein sequence; written in Jax with PyTorch dataloaders

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages