An open API service providing repository metadata for many open source software ecosystems.

GitHub topics: jax

ndarwin314/ninjax

(hopefully) fast pokemon simulator written in JAX to be run on GPU

Language: Python - Size: 249 KB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 1 - Forks: 0

rainx0r/metaworld-algorithms

Implementations of Multi-Task and Meta-Learning baselines for the Metaworld benchmark

Language: Python - Size: 9.15 MB - Last synced at: about 2 hours ago - Pushed at: about 3 hours ago - Stars: 4 - Forks: 0

apple/ml-aim

This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.

Language: Python - Size: 797 KB - Last synced at: 6 days ago - Pushed at: 17 days ago - Stars: 1,275 - Forks: 59

exoplanet-dev/jaxoplanet

Astronomical time series analysis with JAX

Language: Python - Size: 5.91 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 59 - Forks: 14

probml/dynamax

A Python package for probabilistic state space modeling with JAX

Language: Python - Size: 241 MB - Last synced at: 7 days ago - Pushed at: 19 days ago - Stars: 813 - Forks: 94

sail-sg/autofd

Automatic Functional Differentiation in JAX

Language: Python - Size: 286 KB - Last synced at: 6 days ago - Pushed at: about 2 months ago - Stars: 72 - Forks: 1

kazewong/flowMC

Normalizing-flow enhanced sampling package for probabilistic inference in Jax

Language: Python - Size: 30.9 MB - Last synced at: 7 days ago - Pushed at: 11 days ago - Stars: 233 - Forks: 27

luchris429/purejaxrl

Really Fast End-to-End Jax RL Implementations

Language: Python - Size: 1.46 MB - Last synced at: 8 days ago - Pushed at: 8 months ago - Stars: 866 - Forks: 70

GalacticDynamics/diffraxtra

Extras for Diffrax: OOP and vectorization

Language: Python - Size: 113 KB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 2 - Forks: 1

jeremiecoullon/jax-tqdm

Add a tqdm progress bar to your JAX scans and loops.

Language: Python - Size: 94.7 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 108 - Forks: 7

alpa-projects/alpa 📦

Training and serving large-scale neural networks with auto parallelization.

Language: Python - Size: 7.11 MB - Last synced at: 5 days ago - Pushed at: over 1 year ago - Stars: 3,129 - Forks: 359

XanaduAI/GradDFT

GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.

Language: Python - Size: 181 MB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 100 - Forks: 8

jaxleyverse/jaxley

Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.

Language: Python - Size: 19.4 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 92 - Forks: 16

lv416e/archetypax

ArchetypAX: Hardware-accelerated Archetypal Analysis implementation using JAX

Language: Python - Size: 519 KB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 1 - Forks: 0

lanl/scico

Scientific Computational Imaging COde

Language: Python - Size: 2.26 MB - Last synced at: 7 days ago - Pushed at: 17 days ago - Stars: 124 - Forks: 18

AshishKumar4/FlaxDiff

A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplified mathematical explanations. Made as part of my journey for learning and experimenting with generative AI.

Language: Jupyter Notebook - Size: 238 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 25 - Forks: 0

kosaokis/LLaMA-Factory

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

Language: Python - Size: 40.5 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 0 - Forks: 0

GalacticDynamics/quaxed

Pre-quaxed libraries for use in multiple dispatch over abstract array types in JAX.

Language: Python - Size: 422 KB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 5 - Forks: 1

GalacticDynamics/quax-blocks

Blocks for constructing `quax` classes

Language: Python - Size: 130 KB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 1 - Forks: 1

wcxve/elisa

Efficient library for spectral analysis in high-energy astrophysics.

Language: Python - Size: 13.5 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 14 - Forks: 4

GalacticDynamics/coordinax

Coordinates in JAX

Language: Python - Size: 1.67 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 17 - Forks: 4

EMI-Group/evorl

EvoRL is a fully GPU-accelerated framework for Evolutionary Reinforcement Learning, implemented with JAX. It supports Reinforcement Learning (RL), Evolutionary Computation (EC), Evolution-guided Reinforcement Learning (ERL), AutoRL, and seamless integration with GPU-optimized simulation environments.

Language: Python - Size: 2.55 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 75 - Forks: 8

jax-ml/oryx

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

Language: Python - Size: 292 KB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 257 - Forks: 11

GalacticDynamics/xmmutablemap

Immutable Map, compatible with Jax

Language: Python - Size: 51.8 KB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 4 - Forks: 1

GalacticDynamics/galax

Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)

Language: Python - Size: 5.48 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 38 - Forks: 8

GalacticDynamics/unxt

Unitful Quantities in JAX

Language: Python - Size: 1.57 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 36 - Forks: 4

gordicaleksa/get-started-with-JAX

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.

Language: Jupyter Notebook - Size: 1.78 MB - Last synced at: 5 days ago - Pushed at: over 1 year ago - Stars: 725 - Forks: 106

CQCL/qujax

Simulating quantum circuits with JAX

Language: Python - Size: 4.29 MB - Last synced at: 7 days ago - Pushed at: 2 months ago - Stars: 64 - Forks: 12

gwkokab/gwkokab

An open-source JAX-based gravitational-wave population inference toolkit for parametric models

Language: Python - Size: 27.3 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 13 - Forks: 0

google-deepmind/dm_nevis

NEVIS'22: Benchmarking the next generation of never-ending learners

Language: Python - Size: 390 KB - Last synced at: 3 days ago - Pushed at: over 2 years ago - Stars: 102 - Forks: 6

yui-mhcp/data_processing

Data processing utilities in keras3

Language: Jupyter Notebook - Size: 85.6 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 5 - Forks: 1

srigas/jaxKAN

Kolmogorov-Arnold Networks built on JAX

Language: Python - Size: 16.5 MB - Last synced at: 7 days ago - Pushed at: 30 days ago - Stars: 75 - Forks: 9

mosecorg/mosec

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

Language: Python - Size: 1.14 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 840 - Forks: 62

mansooralodhi/intervalAdjointSigAnalysis

msThesis

Language: Python - Size: 25.3 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 0 - Forks: 0

SSAGESLabs/PySAGES

Python Suite for Advanced General Ensemble Simulations

Language: Python - Size: 59.4 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 79 - Forks: 27

Toni-SM/skrl

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab

Language: Python - Size: 7.52 MB - Last synced at: 9 days ago - Pushed at: 20 days ago - Stars: 744 - Forks: 81

patrick-kidger/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Language: Python - Size: 307 KB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 1,390 - Forks: 74

lebrice/torch_jax_interop

Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!

Language: Python - Size: 1.12 MB - Last synced at: 5 days ago - Pushed at: 9 days ago - Stars: 28 - Forks: 2

rkruegs123/idp-design

Optimizing IDPs directly at the level of simulations via a continuous sequence representation.

Language: Jupyter Notebook - Size: 13.5 MB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 7 - Forks: 1

DifferentiableUniverseInitiative/jaxDecomp

JAX bindings for the NVIDIA cuDecomp library

Language: Python - Size: 33.3 MB - Last synced at: 2 days ago - Pushed at: 5 days ago - Stars: 35 - Forks: 1

innat/medic-ai

AI Toolkit for Healthcare Imaging in Keras 3

Language: Jupyter Notebook - Size: 15.5 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 1 - Forks: 1

pasqal-io/horqrux

Jax-based quantum state vector simulator.

Language: Python - Size: 1.79 MB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 26 - Forks: 2

QuantEcon/lecture-jax

Lectures on Quantitative Economics Using JAX

Language: TeX - Size: 65.6 MB - Last synced at: 3 days ago - Pushed at: 10 days ago - Stars: 37 - Forks: 5

instadeepai/Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

Language: Python - Size: 280 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 792 - Forks: 105

ankandrew/fast-plate-ocr

Lightweight & fast OCR models for license plate text recognition.

Language: Python - Size: 267 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 144 - Forks: 26

sandialabs/pancax

A set of tools for developing new methods and techniques in physics informed neural networks written in jax.

Language: Python - Size: 6.59 MB - Last synced at: 10 days ago - Pushed at: 11 days ago - Stars: 6 - Forks: 3

astro-informatics/s2fft

S2FFT: Differentiable and accelerated spherical transforms

Language: Python - Size: 57.7 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 150 - Forks: 9

MaxMSun/lqrax

JAX-enabled continuous-time LQR solver

Language: Python - Size: 2.57 MB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 1 - Forks: 0

debangshu-mukherjee/rheedium

a JAX based package for differentiable RHEED simulations and reconstructions.

Language: Python - Size: 255 KB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 0 - Forks: 0

apax-hub/apax

A flexible and performant framework for training machine learning potentials.

Language: Python - Size: 5.15 MB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 19 - Forks: 3

adtzlr/felupe

:mag: finite element analysis for continuum mechanics of solid bodies

Language: Python - Size: 31.8 MB - Last synced at: 5 days ago - Pushed at: 11 days ago - Stars: 109 - Forks: 13

google-deepmind/dks

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.

Language: Python - Size: 1.23 MB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 70 - Forks: 5

google-research/sofima

Scalable Optical Flow-based Image Montaging and Alignment

Language: Jupyter Notebook - Size: 4.73 MB - Last synced at: 4 days ago - Pushed at: 3 months ago - Stars: 69 - Forks: 16

mmarcinmichal/trax Fork of google/trax

Trax — Deep Learning with Clear Code and Speed

Language: Python - Size: 162 MB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 0 - Forks: 0

evanatyourservice/psgd_jax

Implementation of PSGD optimizer in JAX

Language: Python - Size: 329 KB - Last synced at: 6 days ago - Pushed at: 4 months ago - Stars: 33 - Forks: 2

netket/netket

Machine learning algorithms for many-body quantum systems

Language: Python - Size: 64.3 MB - Last synced at: 12 days ago - Pushed at: 12 days ago - Stars: 592 - Forks: 195

arpastrana/jax_fdm

Auto-differentiable and hardware-accelerated force density method

Language: Python - Size: 114 MB - Last synced at: 3 days ago - Pushed at: 3 months ago - Stars: 88 - Forks: 6

samuela/git-re-basin

Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"

Language: Python - Size: 1.7 MB - Last synced at: 5 days ago - Pushed at: about 2 years ago - Stars: 479 - Forks: 41

LouisDesdoigts/dLux

Differentiable optical models as parameterised neural networks in Jax using Zodiax

Language: Python - Size: 740 MB - Last synced at: 12 days ago - Pushed at: 12 days ago - Stars: 56 - Forks: 8

google-deepmind/PGMax

Loopy belief propagation for factor graphs on discrete variables in JAX

Language: Jupyter Notebook - Size: 14.6 MB - Last synced at: 5 days ago - Pushed at: 7 months ago - Stars: 150 - Forks: 11

pyrddlgym-project/pyRDDLGym-jax

JAX compilation of RDDL description files, and a differentiable planner in JAX.

Language: Python - Size: 12.1 MB - Last synced at: 12 days ago - Pushed at: 12 days ago - Stars: 5 - Forks: 1

FLAIROx/JaxGL

Simple JAX Graphics Library.

Language: Python - Size: 60.5 KB - Last synced at: 3 days ago - Pushed at: 6 months ago - Stars: 36 - Forks: 0

frankroeder/goal_conditioned_rl

Goal-conditioned reinforcement learning like 🔥

Language: Python - Size: 29.3 KB - Last synced at: 4 days ago - Pushed at: over 1 year ago - Stars: 12 - Forks: 0

young-geng/EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Language: Python - Size: 378 KB - Last synced at: 10 days ago - Pushed at: 9 months ago - Stars: 2,474 - Forks: 259

pyro-ppl/numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Language: Python - Size: 95.2 MB - Last synced at: 13 days ago - Pushed at: 13 days ago - Stars: 2,428 - Forks: 256

patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Language: Python - Size: 6.87 MB - Last synced at: 12 days ago - Pushed at: 15 days ago - Stars: 1,607 - Forks: 147

mpi4jax/mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

Language: Python - Size: 5.06 MB - Last synced at: 7 days ago - Pushed at: about 2 months ago - Stars: 478 - Forks: 31

calvinikchen/expanding-ejecta

Pipeline for measuring supernova morphology and the Hubble constant through the expanding ejecta method using intensity interferometer.

Language: Jupyter Notebook - Size: 7.07 MB - Last synced at: 13 days ago - Pushed at: 13 days ago - Stars: 1 - Forks: 1

instadeepai/matrax

A collection of matrix games in JAX

Language: Python - Size: 326 KB - Last synced at: 13 days ago - Pushed at: 5 months ago - Stars: 11 - Forks: 3

EmptyJackson/unifloral

Unified Implementations of Offline Reinforcement Learning Algorithms

Language: Python - Size: 47.9 KB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 58 - Forks: 3

google/paxml

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.

Language: Python - Size: 4.14 MB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 489 - Forks: 69

TolgaOk/jaxdp

A Dynamic Programming package for discrete MDPs implemented in JAX

Language: Python - Size: 549 KB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 5 - Forks: 1

PennyLaneAI/pennylane-cirq

The PennyLane-Cirq plugin integrates Google's Cirq software library with with PennyLane's quantum machine learning capabilities.

Language: Python - Size: 2.07 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 56 - Forks: 18

narendasan/VibeRL

VibeRL is a toolkit for reinforcement learning, designed to facilitate the use of standalone RL implementations such as CleanRL and ReJAX in experiments primarily in JAX

Language: Python - Size: 922 KB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 0 - Forks: 0

mancusolab/jaxqtl

Flexible, scalable, QTL analyses using generalized linear models implemented in JAX

Language: Python - Size: 225 MB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 4 - Forks: 0

JeyRunner/flaxfit

Fitting jax flax models made simple.

Language: Python - Size: 138 KB - Last synced at: 15 days ago - Pushed at: 15 days ago - Stars: 0 - Forks: 0

ZhengYinan-AIR/FISOR

[ICLR 2024] The official implementation of "Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model"

Language: Python - Size: 13.1 MB - Last synced at: 13 days ago - Pushed at: 3 months ago - Stars: 97 - Forks: 7

AaltoML/BayesNewton

Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.

Language: Python - Size: 1.34 MB - Last synced at: about 16 hours ago - Pushed at: over 1 year ago - Stars: 231 - Forks: 29

DarshanDeshpande/jax-models

Unofficial JAX implementations of deep learning research papers

Language: Python - Size: 201 KB - Last synced at: 12 days ago - Pushed at: almost 3 years ago - Stars: 156 - Forks: 9

cbg-ethz/Jnotype

Probabilistic modeling of high-dimensional binary data in JAX

Language: Python - Size: 276 KB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 3 - Forks: 0

NeuralQXLab/nqxpack

Save/Load files from NetKet, flax and other scientific ML libraries

Language: Python - Size: 52.7 KB - Last synced at: 15 days ago - Pushed at: 15 days ago - Stars: 1 - Forks: 0

tianjuxue/jax-am

Additive manufacturing simulation with JAX.

Language: Jupyter Notebook - Size: 57.7 MB - Last synced at: 14 days ago - Pushed at: 8 months ago - Stars: 296 - Forks: 58

JPGoodale/hippox

High-order Polynomial Projection Operators for JAX

Language: Python - Size: 113 KB - Last synced at: 4 days ago - Pushed at: about 2 years ago - Stars: 7 - Forks: 0

brentyi/jax_dataclasses

Pytrees + dataclasses ❤️

Language: Python - Size: 71.3 KB - Last synced at: 1 day ago - Pushed at: 15 days ago - Stars: 62 - Forks: 6

bethgelab/foolbox

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

Language: Python - Size: 10.7 MB - Last synced at: 7 days ago - Pushed at: about 1 year ago - Stars: 2,852 - Forks: 432

brentyi/jaxlie

Rigid transforms + Lie groups for JAX

Language: Python - Size: 13.3 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 253 - Forks: 16

jax-ml/scaling-book

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

Language: HTML - Size: 54.6 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 247 - Forks: 37

gerdm/bayes

Neat Bayesian machine learning examples

Language: Jupyter Notebook - Size: 45.9 MB - Last synced at: 3 days ago - Pushed at: 4 months ago - Stars: 56 - Forks: 8

google-deepmind/dm_pix

PIX is an image processing library in JAX, for JAX.

Language: Python - Size: 761 KB - Last synced at: 8 days ago - Pushed at: 2 months ago - Stars: 415 - Forks: 29

rezaakb/pinns-jax

PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.

Language: Python - Size: 137 KB - Last synced at: 15 days ago - Pushed at: 8 months ago - Stars: 47 - Forks: 6

mia-jinns/jinns

Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs with JAX 📓 Check out our various notebooks to get started ⚠️ Mirror repository of jinns (development happens on Gitlab)

Language: Jupyter Notebook - Size: 110 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 29 - Forks: 7

kubeflow/katib

Automated Machine Learning on Kubernetes

Language: Python - Size: 100 MB - Last synced at: 15 days ago - Pushed at: 22 days ago - Stars: 1,571 - Forks: 473

alonfnt/bayex

Minimal Implementation of Bayesian Optimization in JAX

Language: Python - Size: 370 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 94 - Forks: 3

zombie-einstein/jaxpr-viz

Jaxpr Visualisation Tool

Language: Python - Size: 423 KB - Last synced at: 16 days ago - Pushed at: 5 months ago - Stars: 24 - Forks: 1

ergodicio/tsadar-app

Streamlit application for TSADAR - AD-based Thomson Scattering Analysis software. It is hosted on AWS for simple browser-based access and runs on GPUs for rapid analysis

Language: Python - Size: 411 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 1 - Forks: 0

awslabs/fortuna 📦

A Library for Uncertainty Quantification.

Language: Python - Size: 4.56 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 913 - Forks: 48

BirkhoffG/jax-dataloader

Pytorch-like dataloaders for JAX.

Language: Jupyter Notebook - Size: 697 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 80 - Forks: 3

DBraun/audiotree

Audio data loading and augmentations in JAX

Language: Python - Size: 265 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 3 - Forks: 0

VishwamAI/ProtienFlex

ProteinFlex is a comprehensive platform for protein structure analysis and drug discovery, leveraging advanced AI and machine learning techniques. The platform combines state-of-the-art protein structure prediction with interactive visualization and sophisticated drug discovery tools.

Language: Python - Size: 739 KB - Last synced at: 1 day ago - Pushed at: 17 days ago - Stars: 0 - Forks: 0

ziatdinovmax/NeuroBayes

Fully and Partially Bayesian Neural Nets

Language: Python - Size: 116 MB - Last synced at: 12 days ago - Pushed at: about 1 month ago - Stars: 68 - Forks: 8

Related Keywords
jax 1,202 deep-learning 256 machine-learning 224 python 218 flax 177 pytorch 170 tensorflow 116 reinforcement-learning 109 numpy 64 neural-networks 53 neural-network 44 transformer 44 keras 41 transformers 41 optimization 40 optax 40 automatic-differentiation 40 ai 37 computer-vision 34 nlp 33 haiku 31 gpu 30 llm 28 equinox 27 deep-reinforcement-learning 23 bayesian-inference 23 tpu 23 diffusion-models 21 differentiable-programming 21 dm-haiku 21 python3 20 torch 20 artificial-intelligence 18 physics 18 probabilistic-programming 17 simulation 17 autodiff 16 cuda 16 huggingface 16 xla 16 ml 16 autograd 15 gaussian-processes 14 numpyro 14 gpt 14 research 13 graph-neural-networks 13 astrophysics 13 large-language-models 13 ppo 13 natural-language-processing 13 data-science 13 astronomy 12 scientific-computing 12 tensorflow2 12 vision-transformer 12 mxnet 11 variational-inference 11 quantum-computing 11 language-model 11 deep-neural-networks 11 rl 11 gym 11 keras3 11 reinforcement-learning-algorithms 11 attention 11 robotics 10 gpu-acceleration 10 mujoco 10 diffusion 10 brax 10 quantum-chemistry 10 tutorial 10 vae 10 scipy 9 inverse-problems 9 visualization 9 mlops 9 jit 9 sac 9 scikit-learn 9 physics-informed-neural-networks 9 pde 9 generative-model 9 optics 9 quantum 9 physics-simulation 8 neuroevolution 8 hpc 8 optimization-algorithms 8 cupy 8 pytorch-lightning 8 rnn 8 differential-equations 8 meta-learning 8 cnn 8 dynamical-systems 8 td3 8 score-based-generative-modeling 8 dqn 8