cambridge-mlg / jolt
☆11Updated last month
Alternatives and similar repositories for jolt:
Users that are interested in jolt are comparing it to the libraries listed below
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated 2 months ago
- 🧮 Algebraic Positional Encodings.☆11Updated 2 months ago
- Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model☆14Updated last year
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 5 months ago
- Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression on tabular data with tree-based diffusion …☆41Updated last month
- ☆55Updated last week
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 2 months ago
- Code for minimum-entropy coupling.☆31Updated 9 months ago
- Quantification of Uncertainty with Adversarial Models☆28Updated last year
- ☆16Updated 11 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- This is the repository for the CONFLARE (CONformal LArge language model REtrieval) Python package.☆18Updated 11 months ago
- Code for☆24Updated 3 months ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated last year
- Neural Graphical models are neural network based graphical models that offer richer representation, faster inference & sampling☆28Updated last year
- ☆14Updated 2 years ago
- Differentiable Euler Characteristic Transform☆17Updated 9 months ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 8 months ago
- ☆11Updated 10 months ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆27Updated 11 months ago
- Implementations of growing and pruning in neural networks☆22Updated last year
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated 3 weeks ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 5 months ago
- A simple example of VAEs with KANs☆12Updated 10 months ago
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- A short introduction to Conformal Prediction methods, with a few examples for classification and regression from the Astrophysical domain…☆12Updated 9 months ago
- Evaluation of neuro-symbolic engines☆35Updated 8 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Updated last year