google-deepmind / synjaxLinks
☆247Updated 3 months ago
Alternatives and similar repositories for synjax
Users that are interested in synjax are comparing it to the libraries listed below
Sorting:
- Named Tensors for Legible Deep Learning in JAX☆210Updated last week
- A functional training loops library for JAX☆88Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- JAX Arrays for human consumption☆109Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆280Updated 2 weeks ago
- Neural Networks for JAX☆84Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆157Updated last year
- Tools for JAX☆49Updated last week
- ☆60Updated 3 years ago
- Hierarchical Associative Memory User Experience☆104Updated 3 months ago
- ☆106Updated last year
- ☆115Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆351Updated 6 months ago
- Compositional Linear Algebra☆489Updated 2 months ago
- Running Jax in PyTorch Lightning☆112Updated 10 months ago
- Uncertainty quantification with PyTorch☆375Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆304Updated last week
- Multiple dispatch over abstract array types in JAX.☆131Updated last week
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 10 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last week
- Documentation:☆124Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆98Updated 6 months ago
- Automatic gradient descent☆215Updated 2 years ago
- ☆158Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆295Updated last year
- OpTree: Optimized PyTree Utilities☆196Updated last week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 9 months ago
- Pytorch-like dataloaders for JAX.☆93Updated 4 months ago
- Parameter-Free Optimizers for Pytorch☆131Updated last year
- ☆224Updated last week