sjmikler / pytorch-symbolicLinks
Symbolic API for model creation in PyTorch.
☆66Updated 3 months ago
Alternatives and similar repositories for pytorch-symbolic
Users that are interested in pytorch-symbolic are comparing it to the libraries listed below
Sorting:
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Memory mapped numpy arrays of varying shapes☆298Updated last year
- Create Numpy .npy files by appending on the growth axis☆63Updated 3 weeks ago
- NumPy arrays, ready for human consumption☆69Updated this week
- Neural Networks for JAX☆84Updated 9 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated last month
- Running Jax in PyTorch Lightning☆102Updated 6 months ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆177Updated 2 weeks ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- My implementation of DeepMind's Perceiver☆63Updated 4 years ago
- ☆74Updated 2 years ago
- Check if you have training samples in your test set☆64Updated 3 years ago
- Lightweight Hyperparameter Optimization 🚂☆147Updated 9 months ago
- Image augmentation library for Jax☆39Updated last year
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated 11 months ago
- ☆61Updated last year
- Convert scikit-learn models to PyTorch modules☆163Updated last year
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Mathematical operations for JAX pytrees☆199Updated 6 months ago
- An alternative to convolution in neural networks☆255Updated last year
- ☆114Updated 3 weeks ago
- Notebooks for the "Deep Learning with JAX" book☆149Updated 2 weeks ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Module 0 - Fundamentals☆102Updated 9 months ago
- This is a port of Mistral-7B model in JAX☆32Updated 11 months ago