jax-unirep
by ElArkk
Reimplementation of the UniRep protein featurization model.
AI summary
Protein featurizer
Reimplementation of a protein featurization model using the JAX framework.
- stars
- 104
- forks
- 31
- watching
- 5
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Protein Model
An implementation of a protein language model that uses prompts to learn from multi-level structural information in proteins.
Memory optimizer
A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation.
JAX progress bar library
Library that enhances JAX with support for dynamic progress bars in scans and loops
JAX wrapper
This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
Knowledge Graph Completion Library
A PyTorch implementation of tensor factorization for knowledge graph completion
ELMo model variant
Efficient Contextual Representation Learning Model with Continuous Outputs
Video benchmarking toolkit
Evaluates and benchmarks large language models' video understanding capabilities
Segmentation model
An implementation of SegNet architecture for semantic segmentation using tensorflow and keras.
Language model
An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling
Retrieval model trainer
Trains and evaluates a universal multimodal retrieval model to perform various information retrieval tasks.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
PyTorch tutorial
An introduction to using PyTorch for deep learning tasks
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jax-unirep.
