Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for deep learning jax
deep-learning
x
jax
x
120 search results found
Transformers
⭐
124,049
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Keras
⭐
60,949
Deep Learning for humans
D2l En
⭐
20,613
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Best Of Ml Python
⭐
14,990
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
Ivy
⭐
13,904
The Unified AI Framework
Wandb
⭐
8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Trax
⭐
7,818
Trax — Deep Learning with Clear Code and Speed
Einops
⭐
7,548
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Whisper Jax
⭐
3,824
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
Interviews.ai
⭐
3,146
It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.
Alpa
⭐
2,878
Training and serving large-scale neural networks with auto parallelization.
Dm Haiku
⭐
2,807
JAX-based neural network library
Thinc
⭐
2,774
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Scenic
⭐
2,733
Scenic: A Jax Library for Computer Vision Research and Beyond
Deepxde
⭐
2,135
A library for scientific machine learning and physics-informed learning
Easylm
⭐
2,087
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Pennylane
⭐
2,022
PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
Uvadlc_notebooks
⭐
1,860
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
Equinox
⭐
1,614
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Machine_learning_refined
⭐
1,574
Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).
Big_transfer
⭐
1,421
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Jraph
⭐
1,233
A Graph Neural Network Library in Jax
Transformerengine
⭐
1,172
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Awesome Jax
⭐
1,156
JAX - A curated list of resources https://github.com/google/jax
Jraph
⭐
1,141
A Graph Neural Network Library in Jax
Diffrax
⭐
1,131
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Jaxopt
⭐
857
Hardware accelerated, batchable and differentiable optimizers in JAX.
Fortuna
⭐
824
A Library for Uncertainty Quantification.
Mosec
⭐
661
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
Long Range Arena
⭐
635
Long Range Arena for Benchmarking Efficient Transformers
Tensor Sensor
⭐
634
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
Jaxrl
⭐
576
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Pdebench
⭐
522
PDEBench: An Extensive Benchmark for Scientific Machine Learning
Get Started With Jax
⭐
487
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
Netket
⭐
476
Machine learning algorithms for many-body quantum systems
Elegy
⭐
430
A High Level API for Deep Learning in JAX
Neuralcompression
⭐
430
A collection of tools for neural compression enthusiasts.
Dataloader
⭐
390
The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
Tpu Starter
⭐
381
Everything you want to know about Google Cloud TPU
Git Re Basin
⭐
378
Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"
Machine Learning Guide
⭐
366
Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models (LLMs), and Training Models.
Learning Python Physics Informed Machine Learning Pinns Deeponets
⭐
297
Physics Informed Machine Learning Tutorials (Pytorch and Jax)
Squirrel Core
⭐
271
A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way 🌰
Skrl
⭐
263
Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Isaac Orbit and Omniverse Isaac Gym
Gradcache
⭐
254
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
Annotated S4
⭐
231
Implementation of https://srush.github.io/annotated-s4
Treex
⭐
217
A Pytree Module system for Deep Learning in JAX
Jaxfluids
⭐
203
Differentiable Fluid Dynamics Package
L2p
⭐
202
Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ ECCV22
Flaxmodels
⭐
201
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Easy Neural Ode
⭐
186
Code for the paper "Learning Differential Equations that are Easy to Solve"
Optimistix
⭐
185
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Palm Jax
⭐
181
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
Jaxseq
⭐
172
Train very large language models in Jax.
Jaxnet
⭐
171
Concise deep learning for JAX
E3nn Jax
⭐
143
jax library for E3 Equivariant Neural Networks
Tf_jax_tutorials
⭐
135
All about the fundamental blocks of TF and JAX!
Sunstreaker
⭐
126
以jax为后端的类似keras的框架
Flash Attention Jax
⭐
123
Implementation of Flash Attention in Jax
Swanlab
⭐
108
🧐SwanLab: track and visualize all the pieces of your machine learning pipeline. Join our WeChat ⬇️
Autoray
⭐
106
Abstract your array operations.
Bayesde
⭐
101
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
Symjax
⭐
90
Documentation:
Jax Models
⭐
82
Unofficial JAX implementations of deep learning research papers
Jestimator
⭐
77
Amos optimizer with JEstimator lib.
Nnx
⭐
70
Neural Networks for JAX
Vision
⭐
69
3D Vision functions with end-to-end support for deep learning developers, written in Ivy.
Jax Resnet
⭐
67
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Catx
⭐
56
Contextual bandits library for continuous action trees with smoothing in JAX
Ninjax
⭐
55
General Modules for JAX
Gym
⭐
55
Fully differentiable RL environments, written in Ivy.
Imgx Diffseg
⭐
53
A JAX-based deep learning framework for image segmentation using diffusion models.
Jax Rl
⭐
52
JAX implementations of core Deep RL algorithms
Dks
⭐
50
Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
Madam
⭐
45
Pytorch and Jax code for the Madam optimiser.
Graddft
⭐
45
Grad-DFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.
Revisiting_rainbow
⭐
45
Revisiting Rainbow
Sax
⭐
44
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
Mlp Gpt Jax
⭐
42
A GPT, made only of MLPs, in Jax
Kerax
⭐
39
Keras-like APIs for JAX framework
Sebulba
⭐
31
🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
Manyfold
⭐
28
🧬 ManyFold: An efficient and flexible library for training and validating protein folding models
Haiku Geometric
⭐
26
A collection of graph neural networks implementations in JAX
Tensorshare
⭐
25
🤝 Trade any tensors over the network
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Jaxmapp
⭐
23
JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.
Flaxlm
⭐
23
Train large Huggingface models with flexible data/model parallelism in Flax/JAX
Inox
⭐
21
Stainless neural networks in JAX
Jax Dataloader
⭐
20
Pytorch-like dataloaders in JAX.
Connex
⭐
19
Fine-grained, dynamic control of neural network topology in JAX.
Nixos Nvidia Cuda Python Docker Compose
⭐
18
A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host
Haiku Mup
⭐
18
A port of muP to JAX/Haiku
Kauldron
⭐
17
Modular, scalable codebase to train ML models
Jax Experimental
⭐
16
JAX for Graphcore IPU (experimental)
Pax
⭐
15
A stateful pytree library for training neural networks.
Mech
⭐
14
Mechanics functions with end-to-end support for deep learning developers, written in Ivy.
Cosmopower Jax
⭐
13
Differentiable cosmological emulators: the JAX version of cosmopower
Deep Learning Projects
⭐
13
Best Deep Learning Projects for Advanced Learners
Flaim
⭐
12
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
Detect Fake Text
⭐
12
LLM - Detect AI Generated Text || Identify which essay was written by a large language model
Related Searches
Python Deep Learning (13,095)
Jupyter Notebook Deep Learning (9,310)
Deep Learning Neural Network (5,801)
Deep Learning Pytorch (4,652)
Deep Learning Tensorflow (4,442)
Deep Learning Convolutional Neural Networks (3,932)
Deep Learning Computer Vision (3,018)
Deep Learning Artificial Intelligence (2,898)
Deep Learning Keras (2,519)
Deep Learning Natural Language Processing (2,411)
1-100 of 120 search results
Next >
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.