sooheon / perceiver-jax
Perceiver (transformer variant) implemented in JAX and Flax
☆11Updated 3 years ago
Alternatives and similar repositories for perceiver-jax:
Users that are interested in perceiver-jax are comparing it to the libraries listed below
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 9 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- ☆17Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Updated 2 years ago
- Official implementation of Deep Momentum Schrödinger Bridge☆22Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated 11 months ago
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- Code for "Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation" @ NeurIPS 2023☆11Updated last year
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- ☆23Updated last year
- Euclidean Wasserstein-2 optimal transportation