alexshtf / torchcurvesLinks
Parametric differentiable curves with PyTorch for KANs, continuous embeddings, or shape-restricted models
☆24Updated last week
Alternatives and similar repositories for torchcurves
Users that are interested in torchcurves are comparing it to the libraries listed below
Sorting:
- ☆150Updated last year
- Getting crystal-like representations with harmonic loss☆194Updated 6 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆95Updated 2 months ago
- Modular, scalable library to train ML models☆165Updated this week
- Utilities for PyTorch distributed☆25Updated 7 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆183Updated last week
- ☆67Updated 10 months ago
- Simplified implementation of UMAP like dimensionality reduction algorithm☆52Updated 10 months ago
- ☆60Updated 3 years ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆102Updated 9 months ago
- Portfolio REgret for Confidence SEquences☆20Updated 9 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Updated last year
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated last year
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 8 months ago
- Because we don't want a jupyter notebook mess...☆61Updated 3 months ago
- ☆56Updated 10 months ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆293Updated last year
- ☆28Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- ☆215Updated 10 months ago
- Large multi-modal models (L3M) pre-training.☆170Updated 2 weeks ago
- 🧱 Modula software package☆277Updated last month
- Fast, Modern, and Low Precision PyTorch Optimizers☆113Updated last month
- Jax like function transformation engine but micro, microjax☆32Updated 11 months ago
- ☆82Updated last year
- Dion optimizer algorithm☆360Updated last week
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆70Updated last month
- ☆21Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆46Updated last year