MilesCranmer / pysr_scaling_lawsLinks
You should use PySR to find scaling laws. Here's an example.
☆33Updated 2 years ago
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:
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- ☆60Updated 3 years ago
- Automated SMC with Probabilistic Program Proposals, for the Gen PPL.☆23Updated last year
- Visualize neural networks using TikZ in Julia☆15Updated 8 months ago
- ☆28Updated 3 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Bayesian inference for a logistic regression model in various languages☆42Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- Bibtex for various Python science and machine learning software☆33Updated 2 years ago
- Code for paper https://arxiv.org/abs/2306.07961☆53Updated 3 weeks ago
- ☆16Updated 2 years ago
- ☆36Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- lecture materials of the ML for Physics course 2021 in Perimeter Institute☆21Updated 4 years ago
- ☆44Updated 2 months ago
- ☆15Updated 4 years ago
- Exponential families for JAX☆75Updated 2 weeks ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 8 months ago
- Tidy autoregressive inference in JAX☆14Updated last month
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated 3 weeks ago
- Parallel random matrix tools and complexity for deep learning☆33Updated last year
- Graph neural networks in JAX.☆67Updated last year
- Because we don't want a jupyter notebook mess...☆61Updated 3 months ago
- Julia implementation of flash-attention operation for neural networks.☆11Updated 2 years ago
- Jax SSM Library☆49Updated 2 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago