MilesCranmer / pysr_scaling_lawsLinks
You should use PySR to find scaling laws. Here's an example.
☆33Updated last year
Alternatives and similar repositories for pysr_scaling_laws
Users that are interested in pysr_scaling_laws are comparing it to the libraries listed below
Sorting:
- ☆60Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- Automated SMC with Probabilistic Program Proposals, for the Gen PPL.☆23Updated last year
- Visualize neural networks using TikZ in Julia☆15Updated 7 months ago
- Bayesian inference for a logistic regression model in various languages☆42Updated 2 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- ☆15Updated 4 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Code for paper https://arxiv.org/abs/2306.07961☆53Updated last year
- ☆28Updated 3 years ago
- ☆43Updated last month
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- lecture materials of the ML for Physics course 2021 in Perimeter Institute☆21Updated 4 years ago
- Jax SSM Library☆49Updated 2 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- Tidy autoregressive inference in JAX☆14Updated 2 weeks ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 8 months ago
- Einsum-like high-level array sharding API for JAX☆35Updated last year
- Bibtex for various Python science and machine learning software☆33Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated last week
- Quantification of Uncertainty with Adversarial Models☆29Updated 2 years ago
- ☆36Updated 2 years ago
- Exponential families for JAX☆73Updated last week
- A Wadler--Lindig pretty printer for Python☆41Updated 2 months ago
- Because we don't want a jupyter notebook mess...☆61Updated 3 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 4 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Stencil computations in JAX☆71Updated last year