- (DONE) dataloaders
- (DONE) supportive utils
- loss functions
- basic models
- NEW: mamba
- NEW: model initializers
- train_one_batch
- eval_one_batch
- training wrapper
- eval wrapper
- NEW: CLI + JSON interface
annabellarge / aav-regression-in-jax Goto Github PK
View Code? Open in Web Editor NEWPrediction of AAV capsid fitness from VP1 protein sequence; written in Jax with PyTorch dataloaders