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
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• cgarciae / einop
  ☆59Updated 3 years ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆21Updated last year
• cgarciae / jax-gravity
  ☆24Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 3 years ago
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆32Updated 2 years ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 2 months ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆42Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 2 months ago
• gaurav-arya / differentiable_mh
  Code for paper https://arxiv.org/abs/2306.07961
  ☆53Updated 8 months ago
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• google-deepmind / flows_for_atomic_solids
  ☆49Updated 2 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆63Updated this week
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• rgiordan / paragami
  "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.
  ☆14Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆44Updated 2 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆47Updated 2 weeks ago
• srush / drop7
  ☆18Updated 10 months ago
• maxxxzdn / jax-geometric
  Implementation of various equivariant models in JAX
  ☆12Updated 11 months 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 11 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated 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
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆28Updated last year