Alternatives and similar repositories for pysr_scaling_laws

Users that are interested in pysr_scaling_laws are comparing it to the libraries listed below

• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• Artur-Galstyan / statedict2pytree
  ☆43Updated this week
• cgarciae / einop
  ☆60Updated 3 years ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆22Updated last year
• christopher-beckham / coms-are-energy-models
  Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model
  ☆14Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 5 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 4 months ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• mossr / TikzNeuralNetworks.jl
  Visualize neural networks using TikZ in Julia
  ☆13Updated 4 months ago
• JiaYaobo / fenbux
  A Simple Statistical Distribution Library in JAX
  ☆16Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• gaurav-arya / differentiable_mh
  Code for paper https://arxiv.org/abs/2306.07961
  ☆53Updated 11 months ago
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 5 months ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• QuantEcon / lecture-jax
  Lectures on Quantitative Economics Using JAX
  ☆37Updated this week
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆15Updated 2 months ago
• cgarciae / jax-gravity
  ☆24Updated last year
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆13Updated 8 months ago
• darrenjw / logreg
  Bayesian inference for a logistic regression model in various languages
  ☆42Updated last year
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆39Updated 2 weeks ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆43Updated last year
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆47Updated this week
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• webdataset / WebDataset.jl
  A high performance I/O library for deep learning in Julia, based on the PyTorch WebDataset library
  ☆12Updated 6 months ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 4 months ago
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆33Updated 2 years ago