SamsungSAILMontreal / AnyMolGenCritic
☆20Updated last month
Related projects ⓘ
Alternatives and complementary repositories for AnyMolGenCritic
- Simple Scalable Discrete Diffusion for text in PyTorch☆28Updated last month
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆47Updated 3 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆46Updated this week
- TLDRs for ML in Drug Discovery papers☆71Updated last year
- The Forward-Forward Algorithm for Drug Discovery☆34Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆95Updated 6 months ago
- ☆46Updated last month
- This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…☆37Updated 5 months ago
- Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"☆23Updated last week
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆20Updated 9 months ago
- Graph neural networks in JAX.☆67Updated 5 months ago
- ☆36Updated this week
- ☆53Updated 10 months ago
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆69Updated 3 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆64Updated this week
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆123Updated 11 months ago
- ☆34Updated this week
- Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead☆113Updated this week
- ☆33Updated 3 years ago
- This is a port of Mistral-7B model in JAX☆30Updated 4 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…☆93Updated last year
- Resources from the EleutherAI Math Reading Group☆51Updated last month
- [ICLR 2022] Data-Efficient Graph Grammar Learning for Molecular Generation☆93Updated 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
- Fine-tuning Galactica and Gemma to operate on SMILES. Integrates into a molecular optimization algorithm.☆24Updated 2 months ago
- An annotated implementation of the Hyena Hierarchy paper☆31Updated last year
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆72Updated 3 years ago
- Generative Flow Networks - GFlowNet☆191Updated 2 weeks ago
- Latent Diffusion Language Models☆67Updated last year