MilesCranmer / pysr_scaling_laws

Related projects ⓘ

Alternatives and complementary repositories for pysr_scaling_laws

• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• cgarciae / jax-gravity
  ☆24Updated last year
• ejmichaud / precision-ml
  ☆13Updated last year
• google-deepmind / flows_for_atomic_solids
  ☆48Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆30Updated 4 months ago
• glouppe / flowing-with-jax
  ☆15Updated 3 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• mossr / TikzNeuralNetworks.jl
  Visualize neural networks using TikZ in Julia
  ☆12Updated 2 years ago
• srush / drop7
  ☆18Updated 6 months ago
• MilesCranmer / python_citations
  Bibtex for various Python science and machine learning software
  ☆31Updated last year
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆19Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated last year
• normal-computing / thermox
  Exact OU processes with JAX
  ☆34Updated last month
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• maxxxzdn / jax-geometric
  Implementation of various equivariant models in JAX
  ☆12Updated 6 months ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆12Updated 3 weeks ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆14Updated 9 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• bazingagin / npc_lv
  ☆8Updated last year
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆41Updated last year
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆12Updated 3 months 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
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 5 months ago
• ericmjl / score-models
  In which I learn about score functions and how they can be used to generate data.
  ☆15Updated 7 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆15Updated 3 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆102Updated 3 weeks ago