Alternatives and similar repositories for torchcurves

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

• BlackHC / neural_net_checklist
  ☆150Updated last year
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 9 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆192Updated 6 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
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆96Updated 3 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆82Updated last year
• 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
• kmkolasinski / nano-umap
  Simplified implementation of UMAP like dimensionality reduction algorithm
  ☆53Updated 11 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆71Updated 5 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated last week
• probabilists / azula
  Diffusion models in PyTorch
  ☆111Updated this week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆102Updated 10 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆291Updated 2 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆44Updated 10 months ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated 11 months ago
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆147Updated 6 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 10 months ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 3 months ago
• google-research / optformer
  ☆230Updated last week
• nikhilvyas / SOAP
  ☆218Updated 10 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆295Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated last year
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆90Updated last year
• shikaiqiu / compute-better-spent
  ☆58Updated last year
• epfml / DenseFormer
  ☆81Updated last year