objax
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
772 stars
25 watching
76 forks
Language: Python
last commit: 12 months ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 roberttlange/gymnax | A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration | 669 |
| google/evojax | A toolkit for accelerating neuroevolution with neural networks on accelerators | 856 |
 asappresearch/flambe | An ML framework for accelerating research and its integration into production workflows | 264 |
 google-research/dex-lang | A research language for array processing, exploring type systems and program transformations. | 1,596 |
| google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 461 |
| google-research/noisystudent | A semi-supervised learning method to improve the accuracy of machine learning models by using noisy teacher models and student models. | 755 |
| google/praxis | A library designed to be usable by other JAX-based ML projects, providing optimized layers for machine learning at scale. | 178 |
| google-research/deep_ope | Provides benchmarking policies and datasets for offline reinforcement learning | 85 |
 zk-ml/research | Research on integrating machine learning with emergent runtimes to improve performance and security. | 22 |
 gomlx/gomlx | An accelerated machine learning framework for Go that provides tools and libraries for training, fine-tuning, and modifying machine learning models. | 553 |
 google-deepmind/distrax | A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks. | 538 |
 jimjam-slam/talk-runapp-quarto-2022 | An introduction to Quarto as a tool for creating beautiful documents with R, Python, Julia, and Observable | 20 |
| google-deepmind/functa | A repository containing code for a meta-learning experiment on image datasets | 150 |
 orgexyz/blockagi | An AI research agent built on top of LangChain and OpenAI, designed to conduct iterative, domain-specific research and provide detailed narrative reports. | 296 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |