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
- Neural Networks for JAX☆84Updated last year
- ☆44Updated last month
- ☆60Updated 3 years ago
- Running Jax in PyTorch Lightning☆114Updated last year
- A functional training loops library for JAX☆88Updated last year
- Because we don't want a jupyter notebook mess...☆61Updated 6 months ago
- Hierarchical Associative Memory User Experience☆105Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- ☆118Updated last week
- Automatic gradient descent☆215Updated 2 years ago
- Uncertainty quantification with PyTorch☆376Updated 2 months ago
- ☆192Updated 5 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated this week
- ☆210Updated last year
- JAX Arrays for human consumption☆110Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆297Updated last year
- Run PyTorch in JAX. 🤝☆309Updated 2 months ago
- Pytorch-like dataloaders for JAX.☆97Updated this week
- ☆248Updated 5 months ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆70Updated 2 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆31Updated 2 years ago
- Diffusion models in PyTorch☆120Updated 2 weeks ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 10 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆38Updated 3 years ago
- The Energy Transformer block, in JAX☆62Updated 2 years ago
- ☆62Updated last year
- Parameter-Free Optimizers for Pytorch☆130Updated last year