yobibyte / compressorLinks
Because we don't have enough time to read everything
☆89Updated 10 months ago
Alternatives and similar repositories for compressor
Users that are interested in compressor are comparing it to the libraries listed below
Sorting:
- Because we don't want a jupyter notebook mess...☆61Updated last month
- Graph neural networks in JAX.☆67Updated last year
- Neural Networks for JAX☆84Updated 10 months ago
- Running Jax in PyTorch Lightning☆109Updated 7 months ago
- ☆43Updated 2 months ago
- ☆60Updated 3 years ago
- ☆115Updated last week
- A functional training loops library for JAX☆88Updated last year
- JAX Arrays for human consumption☆106Updated last month
- ☆150Updated 11 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆172Updated 2 years ago
- Hierarchical Associative Memory User Experience☆103Updated 3 weeks ago
- Run PyTorch in JAX. 🤝☆268Updated this week
- Pytorch-like dataloaders for JAX.☆94Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated last year
- Minimal, lightweight JAX implementations of popular models.☆79Updated this week
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated last week
- minGPT in JAX☆48Updated 3 years ago
- ☆247Updated last month
- Uncertainty quantification with PyTorch☆367Updated 3 months ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated 2 years ago
- Automatic gradient descent☆208Updated 2 years ago
- A simple library for scaling up JAX programs☆140Updated 9 months ago
- Diffusion models in PyTorch☆107Updated last month
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 6 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 7 months ago
- LoRA for arbitrary JAX models and functions☆140Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 11 months ago
- Use Jax functions in Pytorch☆248Updated 2 years ago