awesome-jax
awesome-jax copied to clipboard
n2cholas
JAX - A curated list of resources https://github.com/google/jax
Added the [Jochastic](https://github.com/nestordemeure/jochastic) library to the new libraries list.
Hi everyone! I'm suggesting to add our new library for writing differentiable acoustic simulations in `jax`, in case it is deemed of interest. Thank you again for the great project!
The "New Libraries" section was meant to showcase cool libraries that are up-and-coming, to expose them to new users to help them develop and grow. However, some of the libraries...
This is just an implementation of VGG-16 in Haiku + pretrained imagenet weights. I'm not sure if it's substantial enough to make the list, but I figured I'd make a...
Included NanoDL, a library for building and training diverse transformers in Jax/Flax, with high flexibility.
EasyDeL is a well-documented source library to make your training faster and more Optimized With cool Options for training and serving and supports a wide range of models that have...
Reinforced-lib is a lightweight Python library for the rapid development of RL solutions. It is open-source, provides comprehensive [documentation](https://reinforced-lib.readthedocs.io/en/latest/), and is built on JAX. An [article](https://papers.ssrn.com/sol3/papers.cfm?abstract_id=4584200) about the library is...
I just wanted to share this lightweight library that allows to automatically parse and convert biological network models (written in standard SBML format) into python models written end-to-end in JAX,...
