jax-unirep
Protein featurizer
Reimplementation of a protein featurization model using the JAX framework.
Reimplementation of the UniRep protein featurization model.
104 stars
5 watching
31 forks
Language: TeX
last commit: 3 months ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 ikostrikov/jaxrl | Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces. | 630 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 hicai-zju/promptprotein | An implementation of a protein language model that uses prompts to learn from multi-level structural information in proteins. | 32 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 132 |
 jeremiecoullon/jax-tqdm | Library that enhances JAX with support for dynamic progress bars in scans and loops | 96 |
 dfm/extending-jax | This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing | 378 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
 ibalazevic/tucker | A PyTorch implementation of tensor factorization for knowledge graph completion | 350 |
 uclanlp/elmo-c | Efficient Contextual Representation Learning Model with Continuous Outputs | 4 |
 pku-yuangroup/video-bench | Evaluates and benchmarks large language models' video understanding capabilities | 117 |
 ykamikawa/tf-keras-segnet | An implementation of SegNet architecture for semantic segmentation using tensorflow and keras. | 178 |
 l0sg/relational-rnn-pytorch | An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 244 |
 tiger-ai-lab/uniir | Trains and evaluates a universal multimodal retrieval model to perform various information retrieval tasks. | 110 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 461 |
 kaixhin/grokking-pytorch | An introduction to using PyTorch for deep learning tasks | 1,194 |