SamsungSAILMontreal / AnyMolGenCriticLinks
☆21Updated 2 months ago
Alternatives and similar repositories for AnyMolGenCritic
Users that are interested in AnyMolGenCritic are comparing it to the libraries listed below
Sorting:
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 5 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 10 months ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆33Updated 8 months ago
- Latent Diffusion Language Models☆68Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆62Updated last month
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆130Updated last year
- ☆54Updated 6 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 4 months ago
- This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…☆47Updated last year
- Official implementation of "BERTs are Generative In-Context Learners"☆28Updated 2 months ago
- ☆80Updated last year
- ESM2 protein language models in JAX/Flax☆17Updated 2 years ago
- Graph neural networks in JAX.☆67Updated 11 months ago
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆180Updated 4 months ago
- Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…☆56Updated 2 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆45Updated 2 months ago
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 7 months ago
- Code for the paper https://arxiv.org/abs/2402.04997☆78Updated last year
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆93Updated 2 weeks ago
- TLDRs for ML in Drug Discovery papers☆70Updated 2 years ago
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆53Updated last year
- Making folding experiments more accessible .☆36Updated last month
- ☆109Updated last year
- ☆53Updated last year
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 11 months ago
- Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)☆68Updated 9 months ago
- Utilities for PyTorch distributed☆24Updated 3 months ago
- RITA is a family of autoregressive protein models, developed by LightOn in collaboration with the OATML group at Oxford and the Debora Ma…☆96Updated 2 years ago
- supporting pytorch FSDP for optimizers☆79Updated 5 months ago