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 3 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated last month
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 9 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆114Updated this week
• BlackHC / neural_net_checklist
  ☆150Updated last year
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆192Updated 7 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated 10 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated 11 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆114Updated 11 months ago
• kmkolasinski / nano-umap
  Simplified implementation of UMAP like dimensionality reduction algorithm
  ☆53Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆44Updated 11 months ago
• NVlabs / Forecasting-Model-Search
  A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
  ☆25Updated last year
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 10 months ago
• apple / ml-pararnn
  ☆140Updated 2 weeks ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆83Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆18Updated last year
• apple / ml-ademamix
  ☆68Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 4 months ago
• nikhilvyas / SOAP
  ☆223Updated 11 months ago
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆105Updated 3 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆197Updated 11 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆303Updated 3 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 5 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆18Updated 3 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆74Updated last week
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆275Updated last week
• shikaiqiu / compute-better-spent
  ☆61Updated last year