sooheon / perceiver-jax

Related projects: ⓘ

• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• neel-dey / equivariant-gans
  [ICLR 2021] Group Equivariant Generative Adversarial Networks.
  ☆13Updated 3 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆40Updated last year
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆23Updated 6 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆19Updated 10 months ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 3 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆33Updated 2 years ago
• danielegrattarola / GINR
  Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).
  ☆67Updated last year
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆12Updated 2 years ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆38Updated 7 months ago
• cscarv / riemannian-metric-learning-ot
  ☆15Updated last year
• gallego-posada / eman
  Equivariant Mesh Attention Networks
  ☆21Updated 2 years ago
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆27Updated last year
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• merlresearch / partial_gcnn
  Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"
  ☆10Updated last year
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆40Updated 2 years ago
• necludov / wl-mechanics
  ☆30Updated 3 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆19Updated last month
• ngruver / lie-deriv
  ☆35Updated last year
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆26Updated last year
• maxxxzdn / clifford-group-equivariant-cnns
  Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"
  ☆39Updated last month
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆46Updated 6 months ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 2 years ago
• DiffEqML / kairos
  Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure
  ☆20Updated last year
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆63Updated 3 years ago