The Unified AI Framework
Fully differentiable RL environments, written in Ivy.
Mechanics functions with end-to-end support for deep learning developers, written in Ivy.
3D Vision functions with end-to-end support for deep learning developers, written in Ivy.
Implementation of https://srush.github.io/annotated-s4
Graph Convolutional Networks in JAX
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Python code for "Probabilistic Machine learning" book by Kevin Murphy