aradha / agop_feature_learningLinks
☆14Updated last year
Alternatives and similar repositories for agop_feature_learning
Users that are interested in agop_feature_learning are comparing it to the libraries listed below
Sorting:
- Code for verifying deep neural feature ansatz☆19Updated 2 years ago
- ☆55Updated 2 months ago
- DISCS: The code base for the Benchmark for Discrete Sampling☆56Updated 10 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆24Updated last year
- An implementation of ESM2 in Equinox+JAX☆29Updated 3 weeks ago
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).☆23Updated 8 months ago
- Code for lin-RFM used for sparse recovery tasks☆14Updated 3 months ago
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- DiBS: Differentiable Bayesian Structure Learning, NeurIPS 2021☆50Updated last year
- Implementation of Action Matching☆44Updated 2 years ago
- In which I learn about score functions and how they can be used to generate data.☆16Updated last year
- Notebooks for 6.S088 IAP 2023☆14Updated 10 months ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆19Updated 10 months ago
- Pytorch-like dataloaders for JAX.☆87Updated last month
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆21Updated 11 months ago
- Multiple dispatch over abstract array types in JAX.☆125Updated last week
- Contrastive neighbor embeddings☆54Updated 3 months ago
- ☆68Updated 6 months ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆55Updated last year
- pyGPCCA - python GPCCA: Generalized Perron Cluster Cluster Analysis package to coarse-grain reversible and non-reversible Markov state mo…☆21Updated 2 months ago
- An annotated implementation of the Hyena Hierarchy paper☆33Updated 2 years ago
- PhD Thesis☆34Updated last year
- ☆43Updated 2 weeks ago
- Riemannian Optimization Using JAX☆49Updated last year
- TARDIS: Topological Algorithms for Robust DIscovery of Singularities☆42Updated last year
- Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)☆68Updated 10 months ago
- Research Project Template Repository☆32Updated 2 weeks ago
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆78Updated 2 years ago
- Exact OU processes with JAX☆48Updated 3 months ago