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
  ☆98Updated 5 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆195Updated 9 months ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆25Updated 10 months ago
• apple / ml-pararnn
  ☆161Updated 2 months ago
• kmkolasinski / nano-umap
  Simplified implementation of UMAP like dimensionality reduction algorithm
  ☆53Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 7 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated last year
• nikhilvyas / SOAP
  ☆237Updated last year
• cgarciae / einop
  ☆60Updated 3 years 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
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆191Updated last week
• BlackHC / neural_net_checklist
  ☆214Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• probabilists / azula
  Diffusion models in PyTorch
  ☆123Updated 2 weeks ago
• yobibyte / compressor
  Because we don't have enough time to read everything
  ☆89Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆83Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆118Updated last year
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆198Updated last year
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated last year
• Artur-Galstyan / statedict2pytree
  ☆44Updated 2 months ago
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆156Updated 2 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆173Updated last week
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆45Updated last year
• okarthikb / state-space-models
  ☆27Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆75Updated 6 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆298Updated last year