Explore Projects
Discover 54 open source projects
Showing 41-54 of 54 projects
patrick-kidger/jaxtyping
A Python library that provides type annotations and runtime checking for JAX/NumPy/PyTorch arrays.
google-research/big_transfer
Official repository for the 'Big Transfer (BiT): General Visual Representation Learning' paper, focused on transfer learning for computer vision.
google-deepmind/jraph
A Graph Neural Network Library in Jax for building AI-powered graph-based applications.
pytorch/functorch
functorch is a PyTorch library that provides JAX-like composable function transforms for gradient, hessian, and other computations.
apple/ml-aim
This repository provides code and models for large-scale computer vision research projects using AI tools.
jax-md/jax-md
A differentiable, hardware-accelerated molecular dynamics library for scientific computing.
keras-team/keras-core
A multi-backend Keras API implementation with support for TensorFlow, JAX, and PyTorch.
vojtechhabarta/typescript-generator
Generates TypeScript from Java JSON declarations and REST service clients
DBraun/DawDreamer
A powerful Digital Audio Workstation (DAW) built with Python, VST instruments/effects, and AI/ML tools like FAUST, JAX, and JUCE.
ashishpatel26/Treasure-of-Transformers
A comprehensive collection of Transformer models for natural language processing tasks
pdebench/PDEBench
An extensive benchmark for scientific machine learning, focused on physics-informed neural networks and partial differential equations.
waymo-research/waymax
A JAX-based simulator for autonomous driving research, focused on developing AI-powered self-driving systems.
luchris429/purejaxrl
Fast and efficient Jax-based reinforcement learning implementations for developers building AI-powered applications.
google/jaxopt
JAX-based, hardware accelerated, batchable and differentiable optimizers for machine learning.
Stay in the loop
Get weekly updates on trending AI coding tools and projects.