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:
- ☆60Updated 3 years ago
- Graph neural networks in JAX.☆68Updated last year
- ☆44Updated 2 months ago
- Because we don't want a jupyter notebook mess...☆61Updated 7 months ago
- Neural Networks for JAX☆84Updated last year
- A functional training loops library for JAX☆88Updated last year
- Running Jax in PyTorch Lightning☆118Updated last year
- Code for minimum-entropy coupling.☆32Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Hierarchical Associative Memory User Experience☆106Updated 2 months ago
- ☆118Updated last month
- JAX Arrays for human consumption☆110Updated 2 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 11 months ago
- Run PyTorch in JAX. 🤝☆309Updated 2 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated 2 weeks ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- Uncertainty quantification with PyTorch☆379Updated this week
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- ☆250Updated 6 months ago
- ☆192Updated 6 months ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- ☆214Updated last year
- Pytorch-like dataloaders for JAX.☆97Updated 3 weeks ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆77Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Automatic gradient descent☆216Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆297Updated last year
- Compositional Linear Algebra☆504Updated 5 months ago