alexshtf / torchcurvesLinks
Parametric differentiable curves with PyTorch for KANs, continuous embeddings, or shape-restricted models
☆36Updated 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
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 9 months ago
- Getting crystal-like representations with harmonic loss☆192Updated 6 months ago
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆96Updated 3 months ago
- A State-Space Model with Rational Transfer Function Representation.☆82Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 9 months ago
- Simplified implementation of UMAP like dimensionality reduction algorithm☆53Updated 11 months ago
- ☆60Updated 3 years ago
- Neural Networks for JAX☆84Updated last year
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆71Updated 5 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last week
- Diffusion models in PyTorch☆111Updated this week
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆102Updated 10 months ago
- 🧱 Modula software package☆291Updated 2 months ago
- Lightning-like training API for JAX with Flax☆44Updated 10 months ago
- ☆56Updated 11 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆147Updated 6 months ago
- This is a port of Mistral-7B model in JAX☆32Updated last year
- Because we don't want a jupyter notebook mess...☆61Updated 4 months ago
- Running Jax in PyTorch Lightning☆113Updated 10 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆74Updated 3 months ago
- ☆230Updated last week
- ☆218Updated 10 months ago
- Graph neural networks in JAX.☆68Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆295Updated last year
- Jax like function transformation engine but micro, microjax☆33Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- ☆58Updated last year
- ☆81Updated last year