Alternatives and similar repositories for torchcurves

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

• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆97Updated 5 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 2 weeks ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 11 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆83Updated last year
• probabilists / azula
  Diffusion models in PyTorch
  ☆120Updated last week
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• apple / ml-pararnn
  ☆156Updated 2 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated this week
• kmkolasinski / nano-umap
  Simplified implementation of UMAP like dimensionality reduction algorithm
  ☆53Updated last year
• ESHyperscale / HyperscaleES
  Jax Codebase for Evolutionary Strategies at the Hyperscale
  ☆205Updated last week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆194Updated 8 months ago
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 6 months ago
• BlackHC / neural_net_checklist
  ☆212Updated last year
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆182Updated last week
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆310Updated 2 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated 2 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 6 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆117Updated last year
• google-deepmind / tf2jax
  ☆118Updated 3 weeks ago
• Artur-Galstyan / statedict2pytree
  ☆44Updated last month
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆172Updated last week
• nikhilvyas / SOAP
  ☆230Updated last year
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆153Updated last month
• okarthikb / state-space-models
  ☆28Updated last year
• Think-a-Tron / evolve
  open source alpha evolve
  ☆67Updated 7 months ago