yobibyte / compressor
Because we don't have enough time to read everything
โ83Updated 2 weeks ago
Related projects: โ
- Graph neural networks in JAX.โ66Updated 3 months ago
- Run PyTorch in JAX. ๐คโ187Updated 10 months ago
- A functional training loops library for JAXโ84Updated 7 months ago
- Pytorch-like dataloaders in JAX.โ52Updated last month
- A simple library for scaling up JAX programsโ116Updated last month
- JAX Arrays for human consumptionโ87Updated last year
- โ40Updated 2 months ago
- Neural Networks for JAXโ82Updated 2 months ago
- โ104Updated last week
- โ56Updated 2 years ago
- Flow-matching algorithms in JAXโ62Updated last month
- Visualizations of the theory behind diffusion models.โ63Updated 5 months ago
- A Python package of computer vision models for the Equinox ecosystem.โ100Updated 2 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/โ26Updated last year
- Running Jax in PyTorch Lightningโ76Updated last year
- WandB sweeps integration with Hydra sweeperโ45Updated 7 months ago
- Named Tensors for Legible Deep Learning in JAXโ147Updated this week
- Agustinus' very opiniated publication-ready plotting libraryโ51Updated 2 weeks ago
- diffusionjax is a simple and accessible diffusion models package in JAXโ32Updated 3 weeks ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.โ34Updated last month
- A MAD laboratory to improve AI architecture designs ๐งชโ84Updated 4 months ago
- TopoBenchmarkX is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learningโ65Updated last week
- โ137Updated last month
- Multiple dispatch over abstract array types in JAX.โ100Updated last month
- Lightning-like training API for JAX with Flaxโ28Updated 4 months ago
- โ180Updated 2 months ago
- Use Jax functions in Pytorchโ224Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.โ37Updated last year
- Turn jitted jax functions back into python source codeโ20Updated 2 months ago
- โ191Updated 4 months ago