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
  ☆60Updated 3 years ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆22Updated last year
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• Artur-Galstyan / statedict2pytree
  ☆43Updated last month
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• 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
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆32Updated 2 years ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆14Updated last month
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆13Updated 8 months ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 4 months ago
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆27Updated 3 months ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 3 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆38Updated 2 weeks ago
• glouppe / ssi2023
  ☆21Updated last year
• ericmjl / score-models
  In which I learn about score functions and how they can be used to generate data.
  ☆16Updated last year
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 3 months ago
• tom-pollak / claudette-pydantic
  ☆9Updated 6 months ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 5 years ago
• cgarciae / jax-gravity
  ☆24Updated last year
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• gaurav-arya / differentiable_mh
  Code for paper https://arxiv.org/abs/2306.07961
  ☆53Updated 10 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated last month
• NeilGirdhar / efax
  Exponential families for JAX
  ☆67Updated last week
• webdataset / WebDataset.jl
  A high performance I/O library for deep learning in Julia, based on the PyTorch WebDataset library
  ☆12Updated 5 months ago