yobibyte / compressorLinks
Because we don't have enough time to read everything
☆89Updated last year
Alternatives and similar repositories for compressor
Users that are interested in compressor are comparing it to the libraries listed below
Sorting:
- Graph neural networks in JAX.☆68Updated last year
- Running Jax in PyTorch Lightning☆119Updated last year
- ☆60Updated 3 years ago
- Because we don't want a jupyter notebook mess...☆61Updated 7 months ago
- Neural Networks for JAX☆84Updated last year
- ☆45Updated 3 months ago
- A functional training loops library for JAX☆88Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆32Updated 2 years ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- JAX Arrays for human consumption☆110Updated 3 months ago
- Hierarchical Associative Memory User Experience☆106Updated 2 weeks ago
- ☆120Updated this week
- Uncertainty quantification with PyTorch☆379Updated this week
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Automatic gradient descent☆217Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆191Updated 3 weeks ago
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- ☆215Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆69Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆70Updated 8 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated last year
- ☆195Updated 3 weeks ago
- Diffusion models in PyTorch☆124Updated this week
- Pytorch-like dataloaders for JAX.☆98Updated last month
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆77Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆298Updated last year
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 3 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆75Updated 7 months ago