google-deepmind / understanding_deep_generative_models_with_generalized_empirical_likelihood

Related projects: ⓘ

• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated last year
• lucidrains / flash-genomics-model
  My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…
  ☆51Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆56Updated 10 months ago
• lucidrains / hyena-pytorch
  ☆23Updated this week
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• lucidrains / equiformer-diffusion
  Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…
  ☆56Updated last year
• facebookresearch / LAWT
  Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)
  ☆61Updated last month
• lucidrains / triangle-multiplicative-module
  Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …
  ☆29Updated 3 years ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆29Updated last year
• lucidrains / evolutionary-design-molecules
  Implementation of the algorithm detailed in paper "Evolutionary design of molecules based on deep learning and a genetic algorithm"
  ☆21Updated 9 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆33Updated last week
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆27Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆28Updated 4 months ago
• sooheon / perceiver-jax
  Perceiver (transformer variant) implemented in JAX and Flax
  ☆11Updated 3 years ago
• google-deepmind / nuclease_design
  ML-guided enzyme engineering
  ☆34Updated last week
• lucidrains / spatial-rmsnorm
  ☆11Updated this week
• gle-bellier / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆24Updated 3 weeks ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated this week
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆66Updated 3 months ago
• lucidrains / gvp-transformer
  ☆15Updated this week
• HannesStark / dirichlet-flow-matching
  ☆75Updated 4 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• nshepperd / jaxtorch
  A JAX nn library
  ☆20Updated 6 months ago
• lucidrains / omegafold-pytorch
  ☆23Updated this week
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆30Updated last year
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆60Updated 2 years ago
• OpenBioML / rl-oed
  ☆11Updated last year
• lucidrains / genetic-algorithm-pytorch
  Toy genetic algorithm in Pytorch
  ☆28Updated 6 months ago
• OpenBioML / project-proposal-template
  The project proposal template for OpenBioML community projects.
  ☆18Updated last year