SamsungSAILMontreal / AnyMolGenCritic
☆21Updated 4 months ago
Alternatives and similar repositories for AnyMolGenCritic:
Users that are interested in AnyMolGenCritic are comparing it to the libraries listed below
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 6 months ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆32Updated 4 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- This is a port of Mistral-7B model in JAX☆31Updated 7 months ago
- Code for https://arxiv.org/abs/2406.04329☆58Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆127Updated last year
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆52Updated last year
- Latent Diffusion Language Models☆68Updated last year
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆151Updated 3 weeks ago
- ☆47Updated 2 months ago
- Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)☆64Updated 6 months ago
- Graph neural networks in JAX.☆67Updated 8 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
- TLDRs for ML in Drug Discovery papers☆71Updated last year
- ☆23Updated this week
- σ-GPT: A New Approach to Autoregressive Models☆61Updated 6 months ago
- ☆36Updated 2 months ago
- Implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆54Updated 2 weeks ago
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago
- This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…☆41Updated 8 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated last month
- Building the cognitive-core to solve ARC-AGI-2☆17Updated 2 weeks ago
- ☆98Updated 9 months ago
- Replication attempt for the Protein Folding Model described in https://www.biorxiv.org/content/10.1101/2021.08.02.454840v1☆37Updated 2 years ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 8 months ago
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆45Updated last week
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆72Updated 3 years ago
- Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…☆56Updated 2 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆103Updated 2 months ago