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: 6 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. | 640 |
 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. | 134 |
 jeremiecoullon/jax-tqdm | Library that enhances JAX with support for dynamic progress bars in scans and loops | 100 |
 dfm/extending-jax | This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing | 379 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
 ibalazevic/tucker | A PyTorch implementation of tensor factorization for knowledge graph completion | 353 |
 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 | 121 |
 ykamikawa/tf-keras-segnet | An implementation of SegNet architecture for semantic segmentation using tensorflow and keras. | 179 |
 l0sg/relational-rnn-pytorch | An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 245 |
 tiger-ai-lab/uniir | Trains and evaluates a universal multimodal retrieval model to perform various information retrieval tasks. | 114 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
 kaixhin/grokking-pytorch | An introduction to using PyTorch for deep learning tasks | 1,194 |