All Projects → n2cholas → Awesome Jax

n2cholas / Awesome Jax

Licence: cc0-1.0
JAX - A curated list of resources https://github.com/google/jax

Projects that are alternatives of or similar to Awesome Jax

Pyresample
Geospatial image resampling in Python
Stars: ✭ 188 (-12.15%)
Mutual labels:  numpy
Alyn
Detect and fix skew in images containing text
Stars: ✭ 202 (-5.61%)
Mutual labels:  numpy
Bohrium
Automatic parallelization of Python/NumPy, C, and C++ codes on Linux and MacOSX
Stars: ✭ 209 (-2.34%)
Mutual labels:  numpy
Fashion Recommendation
A clothing retrieval and visual recommendation model for fashion images.
Stars: ✭ 193 (-9.81%)
Mutual labels:  numpy
Data Science Projects With Python
A Case Study Approach to Successful Data Science Projects Using Python, Pandas, and Scikit-Learn
Stars: ✭ 198 (-7.48%)
Mutual labels:  numpy
Nptyping
💡 Type hints for Numpy
Stars: ✭ 204 (-4.67%)
Mutual labels:  numpy
Tftb
A Python module for time-frequency analysis
Stars: ✭ 185 (-13.55%)
Mutual labels:  numpy
Awkward 0.x
Manipulate arrays of complex data structures as easily as Numpy.
Stars: ✭ 216 (+0.93%)
Mutual labels:  numpy
Python Novice Inflammation
Programming with Python
Stars: ✭ 199 (-7.01%)
Mutual labels:  numpy
Tsalib
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
Stars: ✭ 209 (-2.34%)
Mutual labels:  numpy
Pybotics
The Python Toolbox for Robotics
Stars: ✭ 192 (-10.28%)
Mutual labels:  numpy
Gluon Nlp
NLP made easy
Stars: ✭ 2,344 (+995.33%)
Mutual labels:  numpy
Cheatsheets Ai
Essential Cheat Sheets for deep learning and machine learning researchers https://medium.com/@kailashahirwar/essential-cheat-sheets-for-machine-learning-and-deep-learning-researchers-efb6a8ebd2e5
Stars: ✭ 14,095 (+6486.45%)
Mutual labels:  numpy
Xtensor
C++ tensors with broadcasting and lazy computing
Stars: ✭ 2,453 (+1046.26%)
Mutual labels:  numpy
3dreconstruction
3D reconstruction, sfm with Python3
Stars: ✭ 213 (-0.47%)
Mutual labels:  numpy
Rootpy
A pythonic interface for the ROOT libraries on top of the PyROOT bindings.
Stars: ✭ 186 (-13.08%)
Mutual labels:  numpy
Awkward 1.0
Manipulate JSON-like data with NumPy-like idioms.
Stars: ✭ 203 (-5.14%)
Mutual labels:  numpy
Python Wechat Itchat
微信机器人,基于Python itchat接口功能实例展示:01-itchat获取微信好友或者微信群分享文章、02-itchat获取微信公众号文章、03-itchat监听微信公众号发送的文章、04 itchat监听微信群或好友撤回的消息、05 itchat获得微信好友信息以及表图对比、06 python打印出微信被删除好友、07 itchat自动回复好友、08 itchat微信好友个性签名词云图、09 itchat微信好友性别比例、10 微信群或微信好友撤回消息拦截、11 itchat微信群或好友之间转发消息
Stars: ✭ 216 (+0.93%)
Mutual labels:  numpy
Pysr
Simple, fast, and parallelized symbolic regression in Python/Julia via regularized evolution and simulated annealing
Stars: ✭ 213 (-0.47%)
Mutual labels:  numpy
Windrose
A Python Matplotlib, Numpy library to manage wind data, draw windrose (also known as a polar rose plot), draw probability density function and fit Weibull distribution
Stars: ✭ 208 (-2.8%)
Mutual labels:  numpy

Awesome JAX AwesomeJAX Logo

JAX brings automatic differentiation and the XLA compiler together through a NumPy-like API for high performance machine learning research on accelerators like GPUs and TPUs.

This is a curated list of awesome JAX libraries, projects, and other resources. Contributions are welcome!

Contents

Libraries

  • Neural Network Libraries
    • Flax - Centered on flexibility and clarity.
    • Haiku - Focused on simplicity, created by the authors of Sonnet at DeepMind.
    • Objax - Has an object oriented design similar to PyTorch.
    • Elegy - A framework-agnostic Trainer interface for the Jax ecosystem. Supports Flax, Haiku, and Optax.
    • Trax - "Batteries included" deep learning library focused on providing solutions for common workloads.
    • Jraph - Lightweight graph neural network library.
    • Neural Tangents - High-level API for specifying neural networks of both finite and infinite width.
  • NumPyro - Probabilistic programming based on the Pyro library.
  • Chex - Utilities to write and test reliable JAX code.
  • Optax - Gradient processing and optimization library.
  • RLax - Library for implementing reinforcement learning agents.
  • JAX, M.D. - Accelerated, differential molecular dynamics.
  • Coax - Turn RL papers into code, the easy way.
  • SymJAX - Symbolic CPU/GPU/TPU programming.
  • mcx - Express & compile probabilistic programs for performant inference.

New Libraries

This section contains libraries that are well-made and useful, but have not necessarily been battle-tested by a large userbase yet.

  • Neural Network Libraries
    • FedJAX - Federated learning in JAX, built on Optax and Haiku.
  • jax-unirep - Library implementing the UniRep model for protein machine learning applications.
  • jax-flows - Normalizing flows in JAX.
  • sklearn-jax-kernels - scikit-learn kernel matrices using JAX.
  • jax-cosmo - Differentiable cosmology library.
  • efax - Exponential Families in JAX.
  • mpi4jax - Combine MPI operations with your Jax code on CPUs and GPUs.
  • imax - Image augmentations and transformations.
  • FlaxVision - Flax version of TorchVision.
  • Oryx - Probabilistic programming language based on program transformations.

Models and Projects

Videos

Papers

This section contains papers focused on JAX (e.g. JAX-based library whitepapers, research on JAX, etc). Papers implemented in JAX are listed in the Models/Projects section.

Tutorials and Blog Posts

Community

Contributing

Contributions welcome! Read the contribution guidelines first.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].