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
• cgarciae / einop
  ☆59Updated 2 years ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆42Updated last year
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 3 years ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• mossr / TikzNeuralNetworks.jl
  Visualize neural networks using TikZ in Julia
  ☆12Updated 3 weeks ago
• webdataset / WebDataset.jl
  A high performance I/O library for deep learning in Julia, based on the PyTorch WebDataset library
  ☆10Updated 2 months ago
• google-deepmind / flows_for_atomic_solids
  ☆48Updated 2 years ago
• gaurav-arya / differentiable_mh
  Code for paper https://arxiv.org/abs/2306.07961
  ☆53Updated 7 months ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆21Updated last year
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 2 months ago
• rgiordan / paragami
  "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.
  ☆13Updated last year
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆12Updated 4 months ago
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆32Updated 2 weeks ago
• cgarciae / jax-gravity
  ☆24Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆43Updated last month
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• darrenjw / logreg
  Bayesian inference for a logistic regression model in various languages
  ☆43Updated last year
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 3 months ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• JiaYaobo / fenbux
  A Simple Statistical Distribution Library in JAX
  ☆16Updated 10 months ago
• tom-pollak / claudette-pydantic
  ☆9Updated 3 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆63Updated last week